Repository navigation
Expand file tree
/
Copy pathdata.go
More file actions
96 lines (86 loc) · 1.95 KB
/
Copy pathdata.go
File metadata and controls
96 lines (86 loc) · 1.95 KB
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
package eosc
import (
"encoding/json"
"os"
"reflect"
"strings"
)
type IDataMarshaller interface {
Encode(startIndex int) ([]byte, []*os.File, error)
}
func NewBase[T any](m map[string]interface{}) *Base[T] {
return &Base[T]{m: m}
}
type Base[T any] struct {
Config *T
m map[string]interface{}
Append map[string]interface{}
}
func (b *Base[T]) UnmarshalJSON(bytes []byte) error {
cfg := new(T)
err := json.Unmarshal(bytes, cfg)
if err != nil {
return err
}
appendCfg := make(map[string]interface{})
err = json.Unmarshal(bytes, &appendCfg)
if err != nil {
return err
}
typ := reflect.TypeOf(cfg).Elem()
if typ.Kind() == reflect.Struct {
for i := 0; i < typ.NumField(); i++ {
field := typ.Field(i)
tag := field.Tag.Get("json")
if tag == "-" {
continue // 跳过带有 `json:"-"` 标签的字段
}
if _, ok := b.m[tag]; ok {
delete(b.m, tag)
}
delete(appendCfg, tag)
}
}
for k, v := range b.m {
if _, ok := appendCfg[k]; ok {
// 如果该参数存在，跳过
continue
}
appendCfg[k] = v
}
b.Config = cfg
b.Append = appendCfg
return nil
}
func (b *Base[T]) MarshalJSON() ([]byte, error) {
result := make(map[string]interface{})
val := reflect.Indirect(reflect.ValueOf(b.Config)) // 处理指针
typ := val.Type()
if val.Kind() == reflect.Struct {
for i := 0; i < val.NumField(); i++ {
field := typ.Field(i)
tag := field.Tag.Get("json")
if tag == "-" {
continue // 跳过带有 `json:"-"` 标签的字段
}
if _, ok := b.Append[tag]; ok {
continue // 跳过 b.Append 中的字段
}
if strings.HasSuffix(tag, ",omitempty") {
if val.Field(i).IsZero() {
continue
}
tag = strings.TrimSuffix(tag, ",omitempty")
}
result[tag] = val.Field(i).Interface()
}
}
// 合并 b.Append 的内容
for key, v := range b.Append {
result[key] = v
}
return json.Marshal(result)
}
func (b *Base[T]) SetAppend(key string, value interface{}) {
b.Append[key] = value
}