-
Notifications
You must be signed in to change notification settings - Fork 2
/
tag.go
116 lines (92 loc) · 1.98 KB
/
tag.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
package mapper
import (
"strings"
)
/***************************
@author: tiansheng.ren
@date: 2022/11/1
@desc:
***************************/
type tagHandleS struct {
tagName string
newFn NewTagHandleFn
}
var tagHandleFnList = []tagHandleS{
{"copy", newCopyTag},
{"json", newJSONTag},
{"gorm", newGormTag},
}
type NewTagHandleFn func(string) TagHande
type TagHande interface {
Name() string
OmitEmpty() bool
}
type structTagInline struct {
name string
omitEmpty bool
}
func newJSONTag(tag string) TagHande {
sti := structTagInline{}
tags := strings.Split(tag, ",")
for idx := range tags {
tags[idx] = strings.TrimSpace(tags[idx])
}
if tags[0] != "" && tags[0] != excludeTagValue {
sti.name = strings.TrimSpace(tags[0])
}
// json tag copy value
for _, item := range tags[0:] {
if itemVals := strings.Split(item, "="); strings.TrimSpace(itemVals[0]) == copyValueTagPrefix {
if len(itemVals) > 1 && itemVals[1] != "" {
sti.name = itemVals[1]
}
}
if item == "omitempty" {
sti.omitEmpty = true
}
}
return jsonTag{structTagInline: sti}
}
type jsonTag struct {
structTagInline
}
var _ TagHande = (*jsonTag)(nil)
func newCopyTag(tag string) TagHande {
ct := copyTag{}
tags := strings.Split(tag, ",")
ct.name = tags[0]
return ct
}
type copyTag struct {
structTagInline
to string
}
func (j copyTag) To() string {
return j.to
}
var _ TagHande = (*copyTag)(nil)
func newGormTag(tag string) TagHande {
// primary_key;column:id;type:int(10);not null
sti := structTagInline{}
tags := strings.Split(tag, ";")
for _, tag := range tags {
if strings.HasPrefix(strings.ToLower(tag), "column:") {
tagItems := strings.Split(tag, ":")
if len(tagItems) > 1 {
sti.name = tagItems[1]
break
}
}
}
return jsonTag{structTagInline: sti}
}
type gormTag struct {
structTagInline
}
var _ TagHande = (*gormTag)(nil)
func (j structTagInline) Name() string {
return j.name
}
func (j structTagInline) OmitEmpty() bool {
return j.omitEmpty
}