/
struct.go
92 lines (66 loc) · 1.5 KB
/
struct.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
package goist
import (
"encoding/json"
"fmt"
"reflect"
"strings"
)
func Struct2Map(in interface{}) (out map[string]interface{}, err error) {
out = map[string]interface{}{}
v := reflect.ValueOf(in)
if v.Kind() == reflect.Ptr {
v = v.Elem()
}
if v.Kind() != reflect.Struct {
err = fmt.Errorf("only accepts struct, got %T", v)
return
}
tp := v.Type()
for i := 0; i < v.NumField(); i++ {
fi := tp.Field(i)
if tv := fi.Tag.Get("json"); tv != "" {
tv, _ := strings.CutSuffix(tv, ",omitempty")
out[tv] = v.Field(i).Interface()
}
}
return
}
func Struct2MapMarshal(st interface{}) (m map[string]interface{}, err error) {
bt, err := json.Marshal(st)
if err != nil {
return
}
err = json.Unmarshal(bt, &m)
return
}
func Struct2Str(st interface{}, tag, sep string) (s string, err error) {
return Struct2StrAndWrap(st, tag, sep, "")
}
func Struct2StrAndWrap(st interface{}, tag, sep, wrap string) (s string, err error) {
v := reflect.ValueOf(st)
if v.Kind() == reflect.Ptr {
v = v.Elem()
}
if v.Kind() != reflect.Struct {
err = fmt.Errorf("only accepts struct, got %T", v)
return
}
if tag == "" {
tag = "json"
}
if sep == "" {
sep = ","
}
tp := v.Type()
for i := 0; i < v.NumField(); i++ {
fi := tp.Field(i)
if tv := fi.Tag.Get(tag); tv != "" {
tv, _ = strings.CutSuffix(tv, ",omitempty")
s += fmt.Sprintf("%s%s%s%s", wrap, tv, wrap, sep)
} else {
s += fmt.Sprintf("%s%s%s%s", wrap, fi.Name, wrap, sep)
}
}
s = strings.TrimRight(s, sep)
return
}