-
Notifications
You must be signed in to change notification settings - Fork 0
/
to_json.go
124 lines (111 loc) · 2.94 KB
/
to_json.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
117
118
119
120
121
122
123
124
package main
import (
"fmt"
"reflect"
"strings"
"unicode"
"unicode/utf8"
)
type Person struct {
Name string `json:"name"`
Age int `json:"age"`
IsMarraied bool `json:"is_marraied"`
}
func ToJson() {
k := map[int]Person{
1: {Name: "uccs", Age: 18, IsMarraied: false},
2: {Name: "uccs", Age: 18, IsMarraied: true},
3: {Name: "uccs", Age: 18, IsMarraied: true},
}
fmt.Println(JsonMarshal(k))
fmt.Println(JsonMarshal(&[...]interface{}{
1,
&Person{Name: "uccs", Age: 18, IsMarraied: false},
Person{Name: "uccs", Age: 18, IsMarraied: true},
true,
Person{Name: "uccs", Age: 18, IsMarraied: true},
}))
}
func JsonMarshal(v interface{}) (string, error) {
rv := reflect.ValueOf(v)
rt := rv.Type()
switch rt.Kind() {
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
return fmt.Sprintf("%v", rv.Int()), nil
case reflect.Float32, reflect.Float64:
return fmt.Sprintf("%v", rv.Float()), nil
case reflect.String:
return fmt.Sprintf("%q", rv.String()), nil
case reflect.Bool:
return fmt.Sprintf("%v", rv.Bool()), nil
case reflect.Slice:
return marshalSlice(rv)
case reflect.Struct:
return marshalStruct(rv)
case reflect.Array:
return marshalSlice(rv.Slice(0, rv.Len()))
case reflect.Map:
return marshalMap(rv)
case reflect.Pointer:
if rv.Elem().Kind() == reflect.Array {
return marshalSlice(rv.Elem().Slice(0, rv.Len()))
}
if rv.Elem().Kind() == reflect.Struct {
return JsonMarshal(rv.Elem().Interface())
}
return JsonMarshal(rv.Elem().Interface())
default:
return "", fmt.Errorf("unsupported type: %s", rt)
}
}
func marshalMap(rv reflect.Value) (string, error) {
var items []string
for _, mapKey := range rv.MapKeys() {
mapValue := rv.MapIndex(mapKey)
keyJsonString, err := JsonMarshal(mapKey.Interface())
if err != nil {
return "", err
}
valueJsonString, err := JsonMarshal(mapValue.Interface())
if err != nil {
return "", err
}
items = append(items, fmt.Sprintf("%v:%v", keyJsonString, valueJsonString))
}
return "{" + strings.Join(items, ",") + "}", nil
}
func marshalSlice(rv reflect.Value) (string, error) {
var items []string
for i := 0; i < rv.Len(); i++ {
value, err := JsonMarshal(rv.Index(i).Interface())
if err != nil {
return "", err
}
items = append(items, value)
}
return "[" + strings.Join(items, ",") + "]", nil
}
func marshalStruct(rv reflect.Value) (string, error) {
var items []string
for i := 0; i < rv.NumField(); i++ {
fieldValue := rv.Field(i)
jsonTag := rv.Type().Field(i).Tag.Get("json")
key := rv.Type().Field(i).Name
if !isFieldExported(key) {
continue
}
if jsonTag != "" {
key = jsonTag
}
value, err := JsonMarshal(fieldValue.Interface())
if err != nil {
return "", err
}
items = append(items, fmt.Sprintf("%q:%v", key, value))
}
return "{" + strings.Join(items, ",") + "}", nil
}
func isFieldExported(name string) bool {
r, _ := utf8.DecodeRuneInString(name)
return unicode.IsUpper(r)
}