-
Notifications
You must be signed in to change notification settings - Fork 182
/
setval.go
224 lines (186 loc) · 4.51 KB
/
setval.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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
package structs
import (
"errors"
"fmt"
"reflect"
"github.com/gookit/goutil/internal/comfunc"
"github.com/gookit/goutil/reflects"
)
const defaultInitTag = "default"
// InitOptFunc define
type InitOptFunc func(opt *InitOptions)
// InitOptions struct
type InitOptions struct {
// TagName default value tag name. tag: default
TagName string
// ParseEnv var name on default value. eg: `default:"${APP_ENV}"`
//
// default: false
ParseEnv bool
// ValueHook before set value hook TODO
ValueHook func(val string) any
}
// InitDefaults init struct default value by field "default" tag.
//
// TIPS:
//
// Only support init set: string, bool, intX, uintX, floatX
//
// Example:
//
// type User1 struct {
// Name string `default:"inhere"`
// Age int32 `default:"30"`
// }
//
// u1 := &User1{}
// err = structs.InitDefaults(u1)
// fmt.Printf("%+v\n", u1)
// // Output: {Name:inhere Age:30}
func InitDefaults(ptr any, optFns ...InitOptFunc) error {
rv := reflect.ValueOf(ptr)
if rv.Kind() != reflect.Ptr {
return errors.New("must be provider an pointer value")
}
rv = rv.Elem()
if rv.Kind() != reflect.Struct {
return errors.New("must be provider an struct value")
}
opt := &InitOptions{TagName: defaultInitTag}
for _, fn := range optFns {
fn(opt)
}
return initDefaults(rv, opt)
}
func initDefaults(rv reflect.Value, opt *InitOptions) error {
rt := rv.Type()
for i := 0; i < rt.NumField(); i++ {
ft := rt.Field(i)
// skip don't exported field
if ft.Name[0] >= 'a' && ft.Name[0] <= 'z' {
continue
}
fv := rv.Field(i)
if fv.Kind() == reflect.Struct {
if err := initDefaults(fv, opt); err != nil {
return err
}
continue
}
// skip on field has value
if !fv.IsZero() {
continue
}
tagVal := ft.Tag.Get(opt.TagName)
if err := initDefaultValue(fv, tagVal, opt.ParseEnv); err != nil {
return err
}
}
return nil
}
func initDefaultValue(fv reflect.Value, val string, parseEnv bool) error {
if val == "" || !fv.CanSet() {
return nil
}
// parse env var
if parseEnv {
val = comfunc.ParseEnvVar(val, nil)
}
// set value
return reflects.SetValue(fv, val)
}
/*************************************************************
* load values to a struct
*************************************************************/
// SetOptFunc define
type SetOptFunc func(opt *SetOptions)
// SetOptions for set values to struct
type SetOptions struct {
// FieldTagName get field name for read value. default tag: json
FieldTagName string
// ValueHook before set value hook TODO
ValueHook func(val any) any
// ParseDefault init default value by DefaultValTag tag value.
// default: false
//
// see InitDefaults()
ParseDefault bool
// DefaultValTag name. tag: default
DefaultValTag string
// ParseDefaultEnv parse env var on default tag. eg: `default:"${APP_ENV}"`
//
// default: false
ParseDefaultEnv bool
}
// SetValues set data values to struct ptr
//
// TIPS:
//
// Only support set: string, bool, intX, uintX, floatX
func SetValues(ptr any, data map[string]any, optFns ...SetOptFunc) error {
rv := reflect.ValueOf(ptr)
if rv.Kind() != reflect.Ptr {
return errors.New("must be provider an pointer value")
}
rv = rv.Elem()
if rv.Kind() != reflect.Struct {
return errors.New("must be provider an struct value")
}
opt := &SetOptions{
FieldTagName: "json",
DefaultValTag: defaultInitTag,
}
for _, fn := range optFns {
fn(opt)
}
return setValues(rv, data, opt)
}
func setValues(rv reflect.Value, data map[string]any, opt *SetOptions) error {
if len(data) == 0 {
return nil
}
rt := rv.Type()
for i := 0; i < rt.NumField(); i++ {
ft := rt.Field(i)
name := ft.Name
// skip don't exported field
if name[0] >= 'a' && name[0] <= 'z' {
continue
}
// get field name
tagVal, ok := ft.Tag.Lookup(opt.FieldTagName)
if ok {
info, err := ParseTagValueDefault(name, tagVal)
if err != nil {
return err
}
name = info.Get("name")
}
fv := rv.Field(i)
val, ok := data[name]
// set field value by default tag.
if !ok && fv.IsZero() {
defVal := ft.Tag.Get(opt.DefaultValTag)
if err := initDefaultValue(fv, defVal, opt.ParseDefaultEnv); err != nil {
return err
}
continue
}
// field is struct
if fv.Kind() == reflect.Struct {
asMp, ok := val.(map[string]any)
if !ok {
return fmt.Errorf("field is struct, must provide map data value")
}
if err := setValues(fv, asMp, opt); err != nil {
return err
}
continue
}
// set field value
if err := reflects.SetValue(fv, val); err != nil {
return err
}
}
return nil
}