-
Notifications
You must be signed in to change notification settings - Fork 36
/
marshal.go
89 lines (79 loc) · 1.48 KB
/
marshal.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
package types
import (
"database/sql/driver"
"encoding"
"fmt"
)
type Text[T interface {
encoding.TextMarshaler
}, Tp interface {
*T
encoding.TextUnmarshaler
}] struct {
Val T
}
func (t Text[T, Tp]) Value() (driver.Value, error) {
return t.Val.MarshalText()
}
func (t *Text[T, Tp]) Scan(value any) error {
switch x := value.(type) {
case string:
v := Tp(&t.Val)
return v.UnmarshalText([]byte(x))
case []byte:
v := Tp(&t.Val)
return v.UnmarshalText(x)
case nil:
return nil
default:
return fmt.Errorf("cannot scan type %T: %v", value, value)
}
}
type Binary[T interface {
encoding.BinaryMarshaler
}, Tp interface {
*T
encoding.BinaryUnmarshaler
}] struct {
Val T
}
func (b Binary[T, Tp]) Value() (driver.Value, error) {
return b.Val.MarshalBinary()
}
func (b *Binary[T, Tp]) Scan(value any) error {
switch x := value.(type) {
case string:
v := Tp(&b.Val)
return v.UnmarshalBinary([]byte(x))
case []byte:
v := Tp(&b.Val)
return v.UnmarshalBinary(x)
case nil:
return nil
default:
return fmt.Errorf("cannot scan type %T: %v", value, value)
}
}
type Stringer[T interface {
~[]byte | ~string
fmt.Stringer
}] struct {
Val T
}
func (s Stringer[T]) Value() (driver.Value, error) {
return s.Val.String(), nil
}
func (s *Stringer[T]) Scan(value any) error {
switch x := value.(type) {
case string:
s.Val = T(x)
return nil
case []byte:
s.Val = T(x)
return nil
case nil:
return nil
default:
return fmt.Errorf("cannot scan type %T: %v", value, value)
}
}