Repository navigation
Expand file tree
/
Copy pathdata.go
More file actions
78 lines (68 loc) · 1.55 KB
/
Copy pathdata.go
File metadata and controls
78 lines (68 loc) · 1.55 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
package eosc
import (
"encoding/json"
"os"
"reflect"
)
type IDataMarshaller interface {
Encode(startIndex int) ([]byte, []*os.File, error)
}
func NewBase[T any]() *Base[T] {
return &Base[T]{}
}
type Base[T any] struct {
Config *T
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:"-"` 标签的字段
}
delete(appendCfg, tag)
}
}
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 中的字段
}
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
}