131 lines
2.8 KiB
Go
131 lines
2.8 KiB
Go
// Package json...
|
|
//
|
|
// Description : 动态构建json
|
|
//
|
|
// Author : go_developer@163.com<张德满>
|
|
//
|
|
// Date : 2021-03-10 10:26 下午
|
|
package json
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/go-developer/gopkg/easylock"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
const (
|
|
// PathSplit json 路径的分割符
|
|
PathSplit = "."
|
|
)
|
|
|
|
// JSONode JSOM节点
|
|
//
|
|
// Author : zhangdeman001@ke.com<张德满>
|
|
//
|
|
// Date : 10:33 下午 2021/3/10
|
|
type JSONode struct {
|
|
Key string // json key
|
|
Value interface{} // 对应的值
|
|
Child []*JSONode // 子节点
|
|
IsRoot bool // 是否根节点
|
|
}
|
|
|
|
// NewDynamicJSON 获取JSON实例
|
|
//
|
|
// Author : go_developer@163.com<张德满>
|
|
//
|
|
// Date : 10:36 下午 2021/3/10
|
|
func NewDynamicJSON() *DynamicJSON {
|
|
return &DynamicJSON{
|
|
root: &JSONode{
|
|
Key: "",
|
|
Value: nil,
|
|
Child: nil,
|
|
IsRoot: true,
|
|
},
|
|
nodeCnt: 0,
|
|
lock: easylock.NewLock(),
|
|
}
|
|
}
|
|
|
|
// DynamicJSON 动态json
|
|
//
|
|
// Author : go_developer@163.com<张德满>
|
|
//
|
|
// Date : 11:03 下午 2021/3/10
|
|
type DynamicJSON struct {
|
|
root *JSONode // 节点数
|
|
nodeCnt int // 节点数量
|
|
lock easylock.EasyLock // 锁
|
|
}
|
|
|
|
// SetValue 设置节点值,如果节点不存在,创建;如果已存在,更新, 多级key使用, value 必须是基础数据类型, 如果是结构体, 需要继续添加path,多级path使用.分割
|
|
//
|
|
// Author : go_developer@163.com<张德满>
|
|
//
|
|
// Date : 10:45 下午 2021/3/10
|
|
func (dj *DynamicJSON) SetValue(path string, value interface{}) {
|
|
pathList := strings.Split(path, PathSplit)
|
|
searchRoot := dj.root
|
|
parent := dj.root
|
|
for keyIndex, key := range pathList {
|
|
searchRoot = dj.search(searchRoot, key)
|
|
if nil != searchRoot {
|
|
parent = searchRoot
|
|
} else {
|
|
var val interface{}
|
|
if keyIndex == len(pathList)-1 {
|
|
val = value
|
|
}
|
|
_ = dj.createNode(parent, key, val)
|
|
searchRoot = parent.Child[0]
|
|
parent = parent.Child[0]
|
|
}
|
|
}
|
|
}
|
|
|
|
// Search 搜索一个key TODO : 优化
|
|
//
|
|
// Author : go_developer@163.com<张德满>
|
|
//
|
|
// Date : 11:19 下午 2021/3/10
|
|
func (dj *DynamicJSON) search(root *JSONode, key string) *JSONode {
|
|
if root == nil {
|
|
return nil
|
|
}
|
|
for _, node := range root.Child {
|
|
if node == nil {
|
|
continue
|
|
}
|
|
if node.Key == key {
|
|
return node
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// createNode 创建新的节点
|
|
//
|
|
// Author : go_developer@163.com<张德满>
|
|
//
|
|
// Date : 10:57 下午 2021/3/10
|
|
func (dj *DynamicJSON) createNode(parent *JSONode, key string, value interface{}) error {
|
|
if nil == parent {
|
|
return errors.New("create node error : parent id nil")
|
|
}
|
|
_ = dj.lock.Lock("")
|
|
if parent.Child == nil {
|
|
parent.Child = make([]*JSONode, 0)
|
|
}
|
|
parent.Child = append(parent.Child, &JSONode{
|
|
Key: key,
|
|
Value: value,
|
|
Child: make([]*JSONode, 0),
|
|
IsRoot: false,
|
|
})
|
|
dj.nodeCnt++
|
|
_ = dj.lock.Unlock("")
|
|
return nil
|
|
}
|