-
Notifications
You must be signed in to change notification settings - Fork 22
/
text.go
126 lines (111 loc) · 2.3 KB
/
text.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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
//nolint:revive
package schema
import (
"database/sql/driver"
"fmt"
)
type TextTransformer interface {
TransformText(*Text) any
}
type Text struct {
Str string
Status Status
}
func (dst *Text) GetStatus() Status {
return dst.Status
}
func (*Text) Type() ValueType {
return TypeString
}
func (dst *Text) Size() int {
return len(dst.Str)
}
func (dst *Text) Equal(src CQType) bool {
if src == nil {
return false
}
s, ok := src.(*Text)
if !ok {
return false
}
return dst.Status == s.Status && dst.Str == s.Str
}
func (dst *Text) String() string {
if dst.Status == Present {
return dst.Str
} else {
return ""
}
}
func (dst *Text) Set(src any) error {
if src == nil {
*dst = Text{Status: Null}
return nil
}
if value, ok := src.(CQType); ok {
value2 := value.Get()
if value2 != value {
return dst.Set(value2)
}
}
switch value := src.(type) {
case string:
*dst = Text{Str: value, Status: Present}
case *string:
if value == nil {
*dst = Text{Status: Null}
} else {
*dst = Text{Str: *value, Status: Present}
}
case []byte:
if value == nil {
*dst = Text{Status: Null}
} else {
*dst = Text{Str: string(value), Status: Present}
}
case fmt.Stringer:
if value == fmt.Stringer(nil) {
*dst = Text{Status: Null}
} else {
*dst = Text{Str: value.String(), Status: Present}
}
default:
// Cannot be part of the switch: If Value() returns nil on
// non-string, we should still try to checks the underlying type
// using reflection.
//
// For example the struct might implement driver.Valuer with
// pointer receiver and fmt.Stringer with value receiver.
if value, ok := src.(driver.Valuer); ok {
if value == driver.Valuer(nil) {
*dst = Text{Status: Null}
return nil
} else {
v, err := value.Value()
if err != nil {
return fmt.Errorf("driver.Valuer Value() method failed: %w", err)
}
// Handles also v == nil case.
if s, ok := v.(string); ok {
*dst = Text{Str: s, Status: Present}
return nil
}
}
}
if originalSrc, ok := underlyingStringType(src); ok {
return dst.Set(originalSrc)
}
return fmt.Errorf("cannot convert %v to Text", value)
}
return nil
}
func (dst Text) Get() any {
switch dst.Status {
case Present:
return dst.Str
case Null:
return nil
default:
return dst.Status
}
}