-
Notifications
You must be signed in to change notification settings - Fork 7
/
array.go
114 lines (98 loc) · 1.96 KB
/
array.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
package types
import (
"database/sql/driver"
"encoding/json"
"fmt"
"gorm.io/gorm"
"gorm.io/gorm/schema"
)
type StringArray []string
func (o *StringArray) Scan(src any) error {
bytes, ok := src.([]byte)
if !ok {
return fmt.Errorf("src value cannot cast to []byte")
}
json.Unmarshal(bytes, o)
return nil
}
func (o StringArray) Value() (driver.Value, error) {
if len(o) == 0 {
return nil, nil
}
b, err := json.Marshal(o)
return string(b), err
}
func (StringArray) GormDataType() string {
return "text"
}
func (StringArray) GormDBDataType(db *gorm.DB, field *schema.Field) string {
switch db.Dialector.Name() {
case "sqlite":
return "JSON"
case "mysql":
return "JSON"
case "postgres":
return "JSONB"
}
return "text"
}
type IntArray []string
func (o *IntArray) Scan(src any) error {
bytes, ok := src.([]byte)
if !ok {
return fmt.Errorf("src value cannot cast to []byte")
}
json.Unmarshal(bytes, o)
return nil
}
func (o IntArray) Value() (driver.Value, error) {
if len(o) == 0 {
return nil, nil
}
b, err := json.Marshal(o)
return string(b), err
}
func (IntArray) GormDataType() string {
return "text"
}
func (IntArray) GormDBDataType(db *gorm.DB, field *schema.Field) string {
switch db.Dialector.Name() {
case "sqlite":
return "JSON"
case "mysql":
return "JSON"
case "postgres":
return "JSONB"
}
return "text"
}
type Int64Array []string
func (o *Int64Array) Scan(src any) error {
bytes, ok := src.([]byte)
if !ok {
return fmt.Errorf("src value cannot cast to []byte")
}
json.Unmarshal(bytes, o)
return nil
}
func (o Int64Array) Value() (driver.Value, error) {
if len(o) == 0 {
return nil, nil
}
b, err := json.Marshal(o)
return string(b), err
}
func (Int64Array) GormDataType() string {
return "text"
}
func (Int64Array) GormDBDataType(db *gorm.DB, field *schema.Field) string {
switch db.Dialector.Name() {
case "sqlite":
return "JSON"
case "mysql":
return "JSON"
case "postgres":
return "JSONB"
}
return "text"
}