-
Notifications
You must be signed in to change notification settings - Fork 0
/
string.go
77 lines (60 loc) · 1.32 KB
/
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
package nullable
import (
"bytes"
"database/sql"
"encoding/json"
)
// String is a nullable string.
type String struct {
sql.NullString
}
// NewString returns a new String.
func NewString(s string, valid bool) String {
return String{
sql.NullString{
String: s,
Valid: valid,
},
}
}
// NewStringFromPtr returns a new String from a pointer.
func NewStringFromPtr(s *string) String {
if s == nil {
return NewString("", false)
}
return NewString(*s, true)
}
// MarshalJSON implements the json.Marshaler interface.
func (n String) MarshalJSON() ([]byte, error) {
if !n.Valid {
return []byte("null"), nil
}
return json.Marshal(n.String)
}
// UnmarshalJSON implements the json.Unmarshaler interface.
func (n *String) UnmarshalJSON(b []byte) error {
if bytes.Equal(b, []byte("null")) {
n.String, n.Valid = "", false
return nil
}
if err := json.Unmarshal(b, &n.String); err != nil {
return err
}
n.Valid = true
return nil
}
// MarshalYAML implements the yaml.Marshaler interface.
func (n String) MarshalYAML() (any, error) {
if !n.Valid {
return nil, nil
}
return n.String, nil
}
// UnmarshalYAML implements the yaml.Unmarshaler interface.
func (n *String) UnmarshalYAML(unmarshal func(any) error) error {
if err := unmarshal(&n.String); err != nil {
return err
}
n.Valid = true
return nil
}