/
struct_type.go
69 lines (60 loc) · 1.43 KB
/
struct_type.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
package sdblueprint
import (
"reflect"
)
type structType struct {
reflect.Type
}
type markedField struct {
name string
mark reflect.Type
tag structTag
}
func structTypeOf(v any) (structType, bool) {
switch t := v.(type) {
case nil:
return structType{}, false
case reflect.Type:
if t.Kind() == reflect.Struct {
return structType{t}, true
} else if t.Kind() == reflect.Pointer {
return structTypeOf(t.Elem())
} else {
return structType{t}, false
}
case reflect.Value:
return structTypeOf(t.Type())
default:
return structTypeOf(reflect.TypeOf(t))
}
}
func (st structType) isZero() bool {
return st.Type == nil
}
func (st structType) forEachField(action func(reflect.StructField)) {
n := st.NumField()
for i := 0; i < n; i++ {
sf := st.Field(i)
action(sf)
}
}
func (st structType) findStructMarkIn(markSet markSet) []markedField {
var r []markedField
st.forEachField(func(sf reflect.StructField) {
mark, ok := getFieldMark(sf.Type, markSet)
if ok {
r = append(r, markedField{name: sf.Name, mark: mark, tag: structTag(sf.Tag)})
}
})
return r
}
func newFieldMark(sf reflect.StructField) markedField {
return markedField{name: sf.Name, mark: sf.Type, tag: structTag(sf.Tag)}
}
func (mark markedField) getId(st *structType) string {
if st != nil {
return selectNotEmpty[string](mark.tag.Get("id"), mark.name, st.Name())
} else {
return selectNotEmpty[string](mark.tag.Get("id"), mark.name, "")
}
}