init...

2022-03-01 13:50:13 +08:00
parent 8eab9943e9
commit d2cabd7d58
113 changed files with 7925 additions and 0 deletions
@@ -0,0 +1,219 @@
+package tree_dept
+
+import (
+	"cutego/core/entity"
+	"sort"
+)
+
+// Tree 统一定义菜单树的数据结构, 也可以自定义添加其他字段
+type Tree struct {
+	Id              int         `json:"id"`
+	Data            interface{} `json:"-"` // 自定义对象
+	Label           string      `json:"label"`
+	Leaf            bool        `json:"-"`        // 叶子节点
+	Selected        bool        `json:"-"`        // 选中
+	PartialSelected bool        `json:"-"`        // 部分选中
+	Children        []Tree      `json:"children"` // 子节点
+}
+
+// INode 其他的结构体想要生成菜单树, 直接实现这个接口
+type INode interface {
+	GetId() int
+	GetLabel() string
+	GetParentId() int
+	GetData() interface{}
+	// IsRoot 判断当前节点是否是顶层根节点
+	IsRoot() bool
+}
+type INodes []INode
+
+func (nodes INodes) Len() int {
+	return len(nodes)
+}
+func (nodes INodes) Swap(i, j int) {
+	nodes[i], nodes[j] = nodes[j], nodes[i]
+}
+func (nodes INodes) Less(i, j int) bool {
+	return nodes[i].GetId() < nodes[j].GetId()
+}
+
+// GenerateTree 自定义的结构体实现 INode 接口后调用此方法生成树结构
+// nodes 需要生成树的节点
+// selectedNode 生成树后选中的节点
+// menuTrees 生成成功后的树结构对象
+func GenerateTree(nodes, selectedNodes []INode) (trees []Tree) {
+	trees = []Tree{}
+	// 定义顶层根和子节点
+	var roots, childList []INode
+	for _, v := range nodes {
+		if v.IsRoot() {
+			// 判断顶层根节点
+			roots = append(roots, v)
+		}
+		childList = append(childList, v)
+	}
+
+	for _, v := range roots {
+		childTree := &Tree{
+			Id:    v.GetId(),
+			Label: v.GetLabel(),
+			Data:  v.GetData(),
+		}
+		// 递归之前, 根据父节点确认 childTree 的选中状态
+		childTree.Selected = nodeSelected(v, selectedNodes, childTree.Children)
+		// 递归
+		recursiveTree(childTree, childList, selectedNodes)
+		// 递归之后, 根据子节点确认 childTree 的选中状态
+		if !childTree.Selected {
+			childTree.Selected = nodeSelected(v, selectedNodes, childTree.Children)
+		}
+		// 递归之后, 根据子节点确认 childTree 的半选中状态
+		childTree.PartialSelected = nodePartialSelected(childTree.Children)
+		// 递归之后, 根据子节确认是否是叶子节点
+		childTree.Leaf = len(childTree.Children) == 0
+		trees = append(trees, *childTree)
+	}
+	return
+}
+
+// recursiveTree 递归生成树结构
+// tree 递归的树对象
+// nodes 递归的节点
+// selectedNodes 选中的节点
+func recursiveTree(tree *Tree, nodes, selectedNodes []INode) {
+	data := tree.Data.(INode)
+
+	for _, v := range nodes {
+		if v.IsRoot() {
+			// 如果当前节点是顶层根节点就跳过
+			continue
+		}
+		if data.GetId() == v.GetParentId() {
+			childTree := &Tree{
+				Id:    v.GetId(),
+				Label: v.GetLabel(),
+				Data:  v.GetData(),
+			}
+			// 递归之前, 根据子节点和父节点确认 childTree 的选中状态
+			childTree.Selected = nodeSelected(v, selectedNodes, childTree.Children) || tree.Selected
+			recursiveTree(childTree, nodes, selectedNodes)
+
+			if !childTree.Selected {
+				// 递归之后, 根据子节点确认 childTree 的选中状态
+				childTree.Selected = nodeSelected(v, selectedNodes, childTree.Children)
+			}
+			// 递归之后, 根据子节点确认 childTree 的半选中状态
+			childTree.PartialSelected = nodePartialSelected(childTree.Children)
+			// 递归之后, 根据子节确认是否是叶子节点
+			childTree.Leaf = len(childTree.Children) == 0
+			tree.Children = append(tree.Children, *childTree)
+		}
+	}
+}
+
+// FindRelationNode 在 allTree 中查询 nodes 中节点的所有父节点
+// nodes 要查询父节点的子节点数组
+// allTree 所有节点数组
+func FindRelationNode(nodes, allNodes []INode) (respNodes []INode) {
+	nodeMap := make(map[int]INode)
+	for _, v := range nodes {
+		recursiveFindRelationNode(nodeMap, allNodes, v, 0)
+	}
+
+	for _, v := range nodeMap {
+		respNodes = append(respNodes, v)
+	}
+	sort.Sort(INodes(respNodes))
+	return
+}
+
+// recursiveFindRelationNode 递归查询关联父子节点
+// nodeMap 查询结果搜集到map中
+// allNodes 所有节点
+// node 递归节点
+// t 递归查找类型：0 查找父、子节点；1 只查找父节点；2 只查找子节点
+func recursiveFindRelationNode(nodeMap map[int]INode, allNodes []INode, node INode, t int) {
+	nodeMap[node.GetId()] = node
+	for _, v := range allNodes {
+		if _, ok := nodeMap[v.GetId()]; ok {
+			continue
+		}
+		// 查找父节点
+		if t == 0 || t == 1 {
+			if node.GetParentId() == v.GetId() {
+				nodeMap[v.GetId()] = v
+				if v.IsRoot() {
+					// 是顶层根节点时, 不再进行递归
+					continue
+				}
+				recursiveFindRelationNode(nodeMap, allNodes, v, 1)
+			}
+		}
+		// 查找子节点
+		if t == 0 || t == 2 {
+			if node.GetId() == v.GetParentId() {
+				nodeMap[v.GetId()] = v
+				recursiveFindRelationNode(nodeMap, allNodes, v, 2)
+			}
+		}
+	}
+}
+
+// nodeSelected 判断节点的选中状态
+// node 进行判断节点
+func nodeSelected(node INode, selectedNodes []INode, children []Tree) bool {
+	for _, v := range selectedNodes {
+		if node.GetId() == v.GetId() {
+			// 1. 如果选择节点数组中存在当前节点
+			return true
+		}
+	}
+
+	if len(children) == 0 {
+		// 2. 不满足前置条件1, 且没有子节点
+		return false
+	}
+	selectedNum := 0
+	for _, v := range children {
+		if v.Selected {
+			selectedNum++
+		}
+	}
+	if selectedNum == len(children) {
+		// 不满足前置条件1, 2 , 且子节点全部是选中状态
+		return true
+	}
+	return false
+}
+
+// nodePartialSelected 判断节点的半选中状态
+func nodePartialSelected(trees []Tree) bool {
+	selectedNum := 0
+	for _, v := range trees {
+		if v.Selected {
+			selectedNum++
+		}
+	}
+	if selectedNum == len(trees) || selectedNum == 0 {
+		// 子节点全选中, 或一个也没有选中
+		return false
+	}
+	return true
+}
+
+type DeptList []entity.SysDept
+
+// ConvertToINodeArray 将当前数组转换成父类 INode 接口 数组
+func (s DeptList) ConvertToINodeArray(*[]entity.SysDept) (nodes []INode) {
+	for _, v := range s {
+		nodes = append(nodes, v)
+	}
+	return
+}
+
+// GetTree 获取树结构
+func (s DeptList) GetTree(treeSelect *[]entity.SysDept) []Tree {
+	s = *treeSelect
+	array := s.ConvertToINodeArray(treeSelect)
+	return GenerateTree(array, nil)
+}