package cn.gistack.common.utils;

import cn.gistack.common.node.TreeNode;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * node tree 工具类
 * @author zhongrj
 * @since 2021-11-23
 */
public class NodeTreeUtil {

	/**
	 * map 转 tree 组织机构(父子id不会重复的情况，数据查询来自同一个表)
	 * @param treeMap
	 * @return
	 */
	public  static List<TreeNode> getNodeTree(Map<String, TreeNode> treeMap){
		List<TreeNode> tree = new ArrayList<>();
		if (treeMap.size() > 1) {
			treeMap.forEach((id, treeNode) -> {
				if (treeMap.containsKey(treeNode.getParentId())) {
					treeMap.get(treeNode.getParentId()).getChildren().add(treeNode);
				} else {
					tree.add(treeNode);
				}
			});
		}
		return tree;
	}

	/**
	 * list 转 tree 组织机构
	 * @param parentList
	 * @param childrenList
	 * @return
	 */
	public  static List<TreeNode> getNodeTree(List<TreeNode> parentList,List<TreeNode> childrenList){
		// 遍历
		for (TreeNode parentNode : parentList) {
			// 判断是否有孩子
			if (parentNode.getHasChildren()) {
				List<TreeNode> tree = new ArrayList<>();
				// 遍历子集合
				for (TreeNode childrenNode : childrenList) {
					if (parentNode.getId().equals(childrenNode.getParentId())) {
						tree.add(childrenNode);
					}
				}
				// 设置孩子节点
				parentNode.setChildren(tree);
			}
		}
		return parentList;
	}
}