完成一个简版JSON树构造

json/build.go

@@ -0,0 +1,130 @@
+// 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
+}

json/json_test.go

@@ -0,0 +1,20 @@
+// Package json...
+//
+// Description : json...
+//
+// Author : go_developer@163.com<张德满>
+//
+// Date : 2021-03-10 11:44 下午
+package json
+
+import (
+	"fmt"
+	"testing"
+)
+
+func TestJSON(t *testing.T) {
+	tree := NewDynamicJSON()
+	tree.SetValue("extra.height.value", 180)
+	tree.SetValue("extra.height.unit", "cm")
+	fmt.Println(tree.root)
+}