-
Notifications
You must be signed in to change notification settings - Fork 0
/
gjson.go
472 lines (440 loc) · 12.8 KB
/
gjson.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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
// 版权所有 GoFrame 作者(https://goframe.org)。保留所有权利。
//
// 本源代码形式遵循 MIT 许可协议条款。如果随此文件未分发 MIT 许可副本,
// 您可以在 https://github.com/gogf/gf 获取一份。
// Package gjson 提供了方便的 API 用于处理 JSON/XML/INI/YAML/TOML 数据。
package json类
import (
"reflect"
"strconv"
"strings"
"github.com/888go/goframe/errors/gcode"
"github.com/888go/goframe/errors/gerror"
"github.com/888go/goframe/internal/reflection"
"github.com/888go/goframe/internal/rwmutex"
"github.com/888go/goframe/internal/utils"
"github.com/888go/goframe/text/gstr"
"github.com/888go/goframe/util/gconv"
)
type ContentType string
const (
ContentTypeJson ContentType = `json`
ContentTypeJs ContentType = `js`
ContentTypeXml ContentType = `xml`
ContentTypeIni ContentType = `ini`
ContentTypeYaml ContentType = `yaml`
ContentTypeYml ContentType = `yml`
ContentTypeToml ContentType = `toml`
ContentTypeProperties ContentType = `properties`
)
const (
defaultSplitChar = '.' // 分隔符字符,用于分层数据访问。
)
// Json 是自定义的 JSON 结构体。
type Json struct {
mu rwmutex.RWMutex
p *interface{} // Pointer for hierarchical data access, it's the root of data in default.
c byte // 字符分隔符(默认为'.')
vc bool // 暴力检查(默认为false),用于在分层数据键包含分隔符字符时访问数据。
}
// Json对象创建/加载的选项。
type Options struct {
Safe bool // 标记此对象适用于并发安全的使用场景。这尤其针对Json对象的创建。
Tags string // 自定义解析优先级标签,例如:"json,yaml,MyTag"。这主要用于将结构体解析为Json对象时。
Type ContentType // Type 指定数据内容类型,例如:json、xml、yaml、toml、ini。
StrNumber bool // StrNumber 使 Decoder 在将数字反序列化到 interface{} 时,将其作为字符串处理而非 float64。
}
// iInterfaces 用于对 Interfaces() 方法进行类型断言。
type iInterfaces interface {
X取any数组() []interface{}
}
// iMapStrAny 是支持将结构体参数转换为映射的接口。
type iMapStrAny interface {
X取MapStrAny() map[string]interface{}
}
// iVal 是用于获取底层 interface{} 的接口。
type iVal interface {
X取值() interface{}
}
// setValue 通过 `pattern` 将 `value` 设置为 `j`。
// 注意:
// 1. 如果 value 为 nil 且 removed 为 true,表示删除这个值;
// 2. 在层次数据搜索、节点创建和数据赋值过程中较为复杂。
func (j *Json) setValue(pattern string, value interface{}, removed bool) error {
var (
err error
array = strings.Split(pattern, string(j.c))
length = len(array)
)
if value, err = j.convertValue(value); err != nil {
return err
}
// 初始化检查。
if *j.p == nil {
if 文本类.X是否为数字(array[0]) {
*j.p = make([]interface{}, 0)
} else {
*j.p = make(map[string]interface{})
}
}
var (
pparent *interface{} = nil // Parent pointer.
pointer *interface{} = j.p // Current pointer.
)
j.mu.Lock()
defer j.mu.Unlock()
for i := 0; i < length; i++ {
switch (*pointer).(type) {
case map[string]interface{}:
if i == length-1 {
if removed && value == nil {
// 从map中删除项目。
delete((*pointer).(map[string]interface{}), array[i])
} else {
if (*pointer).(map[string]interface{}) == nil {
*pointer = map[string]interface{}{}
}
(*pointer).(map[string]interface{})[array[i]] = value
}
} else {
// 如果键在映射中不存在。
if v, ok := (*pointer).(map[string]interface{})[array[i]]; !ok {
if removed && value == nil {
goto done
}
// 创建新的节点。
if 文本类.X是否为数字(array[i+1]) {
// 创建数组节点
n, _ := strconv.Atoi(array[i+1])
var v interface{} = make([]interface{}, n+1)
pparent = j.setPointerWithValue(pointer, array[i], v)
pointer = &v
} else {
// 创建映射节点。
var v interface{} = make(map[string]interface{})
pparent = j.setPointerWithValue(pointer, array[i], v)
pointer = &v
}
} else {
pparent = pointer
pointer = &v
}
}
case []interface{}:
// A string key.
if !文本类.X是否为数字(array[i]) {
if i == length-1 {
*pointer = map[string]interface{}{array[i]: value}
} else {
var v interface{} = make(map[string]interface{})
*pointer = v
pparent = pointer
pointer = &v
}
continue
}
// Numeric index.
valueNum, err := strconv.Atoi(array[i])
if err != nil {
err = 错误类.X多层错误码并格式化(错误码类.CodeInvalidParameter, err, `strconv.Atoi failed for string "%s"`, array[i])
return err
}
if i == length-1 {
// Leaf node.
if len((*pointer).([]interface{})) > valueNum {
if removed && value == nil {
// Deleting element.
if pparent == nil {
*pointer = append((*pointer).([]interface{})[:valueNum], (*pointer).([]interface{})[valueNum+1:]...)
} else {
j.setPointerWithValue(pparent, array[i-1], append((*pointer).([]interface{})[:valueNum], (*pointer).([]interface{})[valueNum+1:]...))
}
} else {
(*pointer).([]interface{})[valueNum] = value
}
} else {
if removed && value == nil {
goto done
}
if pparent == nil {
// 这是根节点。
j.setPointerWithValue(pointer, array[i], value)
} else {
// 它不是根节点。
s := make([]interface{}, valueNum+1)
copy(s, (*pointer).([]interface{}))
s[valueNum] = value
j.setPointerWithValue(pparent, array[i-1], s)
}
}
} else {
// Branch node.
if 文本类.X是否为数字(array[i+1]) {
n, _ := strconv.Atoi(array[i+1])
pSlice := (*pointer).([]interface{})
if len(pSlice) > valueNum {
item := pSlice[valueNum]
if s, ok := item.([]interface{}); ok {
for i := 0; i < n-len(s); i++ {
s = append(s, nil)
}
pparent = pointer
pointer = &pSlice[valueNum]
} else {
if removed && value == nil {
goto done
}
var v interface{} = make([]interface{}, n+1)
pparent = j.setPointerWithValue(pointer, array[i], v)
pointer = &v
}
} else {
if removed && value == nil {
goto done
}
var v interface{} = make([]interface{}, n+1)
pparent = j.setPointerWithValue(pointer, array[i], v)
pointer = &v
}
} else {
pSlice := (*pointer).([]interface{})
if len(pSlice) > valueNum {
pparent = pointer
pointer = &(*pointer).([]interface{})[valueNum]
} else {
s := make([]interface{}, valueNum+1)
copy(s, pSlice)
s[valueNum] = make(map[string]interface{})
if pparent != nil {
// i > 0
j.setPointerWithValue(pparent, array[i-1], s)
pparent = pointer
pointer = &s[valueNum]
} else {
// i = 0
var v interface{} = s
*pointer = v
pparent = pointer
pointer = &s[valueNum]
}
}
}
}
// 如果通过`pointer`指向的变量不是引用类型,
// 那么它将通过其父级(即:pparent)来修改该变量。
// 在Go语言中,这段注释描述了如果给定的指针`pointer`不指向一个引用类型,那么对变量的修改会通过其上级父级指针`pparent`间接完成。
default:
if removed && value == nil {
goto done
}
if 文本类.X是否为数字(array[i]) {
n, _ := strconv.Atoi(array[i])
s := make([]interface{}, n+1)
if i == length-1 {
s[n] = value
}
if pparent != nil {
pparent = j.setPointerWithValue(pparent, array[i-1], s)
} else {
*pointer = s
pparent = pointer
}
} else {
var v1, v2 interface{}
if i == length-1 {
v1 = map[string]interface{}{
array[i]: value,
}
} else {
v1 = map[string]interface{}{
array[i]: nil,
}
}
if pparent != nil {
pparent = j.setPointerWithValue(pparent, array[i-1], v1)
} else {
*pointer = v1
pparent = pointer
}
v2 = v1.(map[string]interface{})[array[i]]
pointer = &v2
}
}
}
done:
return nil
}
// convertValue 将 `value` 转换为 map[string]interface{} 或 []interface{},
// 这样就可以支持对层级数据的访问。
func (j *Json) convertValue(value interface{}) (convertedValue interface{}, err error) {
if value == nil {
return
}
switch value.(type) {
case map[string]interface{}:
return value, nil
case []interface{}:
return value, nil
default:
var (
reflectInfo = reflection.OriginValueAndKind(value)
)
switch reflectInfo.OriginKind {
case reflect.Array:
return 转换类.X取any数组(value), nil
case reflect.Slice:
return 转换类.X取any数组(value), nil
case reflect.Map:
return 转换类.X取Map(value), nil
case reflect.Struct:
if v, ok := value.(iMapStrAny); ok {
convertedValue = v.X取MapStrAny()
}
if utils.X是否为Nil(convertedValue) {
if v, ok := value.(iInterfaces); ok {
convertedValue = v.X取any数组()
}
}
if utils.X是否为Nil(convertedValue) {
convertedValue = 转换类.X取Map(value)
}
if utils.X是否为Nil(convertedValue) {
err = 错误类.X创建错误码并格式化(错误码类.CodeInvalidParameter, `unsupported value type "%s"`, reflect.TypeOf(value))
}
return
default:
return value, nil
}
}
}
// setPointerWithValue 将 `key`:`value` 设置到 `pointer` 中,其中 `key` 可能是 map 的键或 slice 的索引。
// 它返回指向新设置值的指针。
func (j *Json) setPointerWithValue(pointer *interface{}, key string, value interface{}) *interface{} {
switch (*pointer).(type) {
case map[string]interface{}:
(*pointer).(map[string]interface{})[key] = value
return &value
case []interface{}:
n, _ := strconv.Atoi(key)
if len((*pointer).([]interface{})) > n {
(*pointer).([]interface{})[n] = value
return &(*pointer).([]interface{})[n]
} else {
s := make([]interface{}, n+1)
copy(s, (*pointer).([]interface{}))
s[n] = value
*pointer = s
return &s[n]
}
default:
*pointer = value
}
return pointer
}
// getPointerByPattern 根据指定的`pattern`返回一个指向该值的指针。
func (j *Json) getPointerByPattern(pattern string) *interface{} {
if j.p == nil {
return nil
}
if j.vc {
return j.getPointerByPatternWithViolenceCheck(pattern)
} else {
return j.getPointerByPatternWithoutViolenceCheck(pattern)
}
}
// getPointerByPatternWithViolenceCheck 函数通过暴力检查的方式,返回指定 `pattern` 的值的指针。
func (j *Json) getPointerByPatternWithViolenceCheck(pattern string) *interface{} {
if !j.vc {
return j.getPointerByPatternWithoutViolenceCheck(pattern)
}
// 如果模式为空,则返回nil。
if pattern == "" {
return nil
}
// 如果模式为".",则返回所有内容。
if pattern == "." {
return j.p
}
var (
index = len(pattern)
start = 0
length = 0
pointer = j.p
)
if index == 0 {
return pointer
}
for {
if r := j.checkPatternByPointer(pattern[start:index], pointer); r != nil {
if length += index - start; start > 0 {
length += 1
}
start = index + 1
index = len(pattern)
if length == len(pattern) {
return r
} else {
pointer = r
}
} else {
// 获取下一个分隔符字符的位置。
index = strings.LastIndexByte(pattern[start:index], j.c)
if index != -1 && length > 0 {
index += length + 1
}
}
if start >= index {
break
}
}
return nil
}
// getPointerByPatternWithoutViolenceCheck 根据指定的`pattern`返回一个指向其值的指针,且不进行暴力检查。
func (j *Json) getPointerByPatternWithoutViolenceCheck(pattern string) *interface{} {
if j.vc {
return j.getPointerByPatternWithViolenceCheck(pattern)
}
// 如果模式为空,则返回nil。
if pattern == "" {
return nil
}
// 如果模式为".",则返回所有内容。
if pattern == "." {
return j.p
}
pointer := j.p
if len(pattern) == 0 {
return pointer
}
array := strings.Split(pattern, string(j.c))
for k, v := range array {
if r := j.checkPatternByPointer(v, pointer); r != nil {
if k == len(array)-1 {
return r
} else {
pointer = r
}
} else {
break
}
}
return nil
}
// checkPatternByPointer 检查在指定的 `pointer` 中是否存在通过 `key` 访问的值。
// 它会返回该值的指针。
func (j *Json) checkPatternByPointer(key string, pointer *interface{}) *interface{} {
switch (*pointer).(type) {
case map[string]interface{}:
if v, ok := (*pointer).(map[string]interface{})[key]; ok {
return &v
}
case []interface{}:
if 文本类.X是否为数字(key) {
n, err := strconv.Atoi(key)
if err == nil && len((*pointer).([]interface{})) > n {
return &(*pointer).([]interface{})[n]
}
}
}
return nil
}