-
Notifications
You must be signed in to change notification settings - Fork 7
/
Copy pathtype_string.go
87 lines (76 loc) · 1.7 KB
/
type_string.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
package generic
import (
"database/sql/driver"
"encoding/json"
)
// String is generic string type structure
type String struct {
ValidFlag
string string
}
// MarshalString return generic.String converting of request data
func MarshalString(x interface{}) (String, error) {
v := String{}
err := v.Scan(x)
return v, err
}
// MustString return generic.String converting of request data
func MustString(x interface{}) String {
v, err := MarshalString(x)
if err != nil {
panic(err)
}
return v
}
// Value implements the driver Valuer interface.
func (v String) Value() (driver.Value, error) {
if !v.Valid() {
return nil, nil
}
return v.string, nil
}
// Scan implements the sql.Scanner interface.
func (v *String) Scan(x interface{}) (err error) {
v.string, v.ValidFlag, err = asString(x)
if err != nil {
v.ValidFlag = false
return err
}
return
}
// Weak returns string, but if String.ValidFlag is false, returns nil.
func (v String) Weak() interface{} {
i, _ := v.Value()
return i
}
// Set sets a specified value.
func (v *String) Set(x interface{}) (err error) {
return v.Scan(x)
}
// String implements the Stringer interface.
func (v String) String() string {
if !v.Valid() {
return ""
}
return v.string
}
// MarshalJSON implements the json.Marshaler interface.
func (v String) MarshalJSON() ([]byte, error) {
if !v.Valid() {
return nullBytes, nil
}
s := `"` + v.string + `"`
bs := make([]byte, 0, len(s))
return append(bs, s...), nil
}
// UnmarshalJSON implements the json.Unmarshaler interface.
func (v *String) UnmarshalJSON(data []byte) error {
if len(data) == 0 {
return nil
}
var in interface{}
if err := json.Unmarshal(data, &in); err != nil {
return err
}
return v.Scan(in)
}