gitee.com/zhongguo168a/gocodes@v0.0.0-20230609140523-e1828349603f/datax/schemax/configs/ConfigTree.go

package configs

import (
	"gitee.com/zhongguo168a/gocodes/datax"
	"gitee.com/zhongguo168a/gocodes/datax/listx"
	"gitee.com/zhongguo168a/gocodes/datax/mapx"
	"gitee.com/zhongguo168a/gocodes/datax/schemax"
	"gitee.com/zhongguo168a/gocodes/datax/treex"
	"strings"
)

func NewConfigTree() (obj *ConfigTree) {
	obj = &ConfigTree{}
	obj.datas = listx.NewList()
	return
}

type ConfigTree struct {
	datas *listx.List
}

func (set *ConfigTree) L列表() *listx.List {
	return set.datas
}

func (set *ConfigTree) ToConfigMap(existed *ConfigMap) *ConfigMap {
	if existed == nil {
		existed = NewConfigMap()
	}

	list := set.L列表()
	for _, val := range list.Slice() {
		c := val.(*ConfigTreeItem)

		var 数据 map[string]interface{}
		i数据 := c.S数据()
		if i数据 == nil {
			数据 = map[string]interface{}{}
		} else {
			数据 = mapx.CloneMap(i数据)
		}
		数据["_type"] = c.L类型
		数据["_catalog"] = c.F分类
		existed.AddConfig(c.B编号, 数据)
	}
	return existed
}

func (set *ConfigTree) Load(files map[string]datax.M) {
	for _, fileBody := range files {
		for key, iconf := range fileBody {
			conf := iconf.(datax.M)
			obj := NewTreeItemByConf(key, conf)
			delete(conf, "_type")
			delete(conf, "_name")
			delete(conf, "_catalog")
			delete(conf, "_parent")
			delete(conf, "_prefix")
			delete(conf, "_suffix")
			delete(conf, "_export")
			delete(conf, "_default")
			delete(conf, "_file")
			obj.原始数据 = conf
			set.datas.Push(obj)
		}
	}

	修正集 := set.C查找_通过分类("修正")
	for _, ival := range 修正集.datas.Slice() {
		修正配置 := ival.(*ConfigTreeItem)
		修正配置编号 := mapx.String(修正配置.原始数据, "配置编号")
		if 修正配置编号 != "" {
			配置 := set.C查找_通过编号(修正配置编号)
			配置.T添加修正(修正配置.B编号)
		}
	}

	root := set.S生成类树()
	root.Walk(func(node *treex.TreeNode) bool {
		if node.Data == nil {
			return true
		}
		node配置 := node.Data.(*ConfigTreeItem)

		if node配置 == nil {
			return true
		}
		data := datax.M{}
		node配置.数据 = data

		decl := node配置.GetSchema()
		if decl != nil {
			_ = schemax.InitObj(data, decl)
		}

		if node.Parent != nil {
			parent := node.Parent.(*treex.TreeNode)
			if parent.Data != nil {
				parent配置 := parent.Data.(*ConfigTreeItem)
				mapx.Copy(parent配置.数据, data)
				node配置.T添加修正集(parent配置.X修正)
			}

		}
		if node配置.L类型 == "" {
			if node.Parent != "" {
				parent := node.Parent.(*treex.TreeNode)
				if parent.Data != nil {
					parent配置 := parent.Data.(*ConfigTreeItem)
					node配置.L类型 = parent配置.L类型
				}
			}
		}
		mapx.Copy(node配置.原始数据, data)
		return true
	})
}
func (set *ConfigTree) C查找_通过文件(文件 string) (result *ConfigTree) {
	result = NewConfigTree()
	for _, ival := range set.datas.Slice() {
		val := ival.(*ConfigTreeItem)
		if val.W文件 == 文件 {
			result.datas.Push(val)
		}
	}
	return
}

func (set *ConfigTree) C查找_通过目录(目录 string) (result *ConfigTree) {
	result = NewConfigTree()
	for _, ival := range set.datas.Slice() {
		val := ival.(*ConfigTreeItem)
		if strings.Contains(val.W文件, 目录) {
			result.datas.Push(val)
		}
	}
	return
}

func (set *ConfigTree) C查找_通过分类(分类 string) (result *ConfigTree) {
	result = NewConfigTree()
	for _, ival := range set.datas.Slice() {
		val := ival.(*ConfigTreeItem)
		if val.F分类 == 分类 {
			result.datas.Push(val)
		}
	}
	return
}
func (set *ConfigTree) C查找_通过类型(类型 string) (result *ConfigTree) {
	result = NewConfigTree()
	for _, ival := range set.datas.Slice() {
		val := ival.(*ConfigTreeItem)
		if val.L类型 == 类型 {
			result.datas.Push(val)
		}
	}
	return
}

func (set *ConfigTree) C查找_通过编号(编号 string) *ConfigTreeItem {
	for i := 0; i < set.datas.Length(); i++ {
		iitem := set.datas.At(i)
		item := iitem.(*ConfigTreeItem)
		if item.B编号 == 编号 {
			return item
		}
	}

	return nil
}

func (set *ConfigTree) C查找_通过修正(编号 string) (result *ConfigTree) {
	result = NewConfigTree()
	for _, ival := range set.datas.Slice() {
		val := ival.(*ConfigTreeItem)
		if val.B包含修正(编号) {
			result.datas.Push(val)
		}
	}

	return
}

func (set *ConfigTree) S生成类树() *treex.TreeRoot {

	var list []*treex.TreeListItem
	list = append(list, &treex.TreeListItem{
		Id: "/",
	})

	for i := 0; i < set.datas.Length(); i++ {
		iitem := set.datas.At(i)
		item := iitem.(*ConfigTreeItem)
		if item.F父编号 != "" {
			list = append(list, &treex.TreeListItem{
				Id: item.B编号, Parent: item.F父编号, Data: item,
			})
		} else {
			list = append(list, &treex.TreeListItem{
				Id: item.B编号, Parent: "/", Data: item,
			})
		}
	}
	return treex.NewByList(list)
}

func (set *ConfigTree) ToSaveFileMap() datax.M {
	result := datax.M{}
	for _, val := range set.L列表().Slice() {
		item := val.(*ConfigTreeItem)
		result[item.B编号] = item.ToMap()
	}
	return result
}