package org.sxkj.system.util;
import org.sxkj.system.entity.DictBiz;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class TreeBuilder {

	/**
	 * 构建树形结构
	 * @param list 原始数据列表
	 * @return 树形结构列表
	 */
	public static List<DictBiz> buildTree(List<DictBiz> list) {
		// 用于存储id和对应的节点
		Map<Long, DictBiz> nodeMap = new HashMap<>();
		// 最终返回的树形结构
		List<DictBiz> tree = new ArrayList<>();

		// 第一遍遍历：将所有节点存入map，并初始化children列表
		for (DictBiz node : list) {
			nodeMap.put(node.getId(), node);
			if (node.getChildrenList() == null) {
				node.setChildrenList(new ArrayList<>());
			}
		}

		// 第二遍遍历：建立父子关系
		for (DictBiz node : list) {
			Long parentId = node.getParentId();
			if (parentId != null && parentId != 0) {  // 假设0表示顶级节点
				DictBiz parent = nodeMap.get(parentId);
				if (parent != null) {
					parent.getChildrenList().add(node);
				}
			} else {
				// 没有父节点或父节点为0，认为是顶级节点
				tree.add(node);
			}
		}

		return tree;
	}

	/**
	 * 使用Stream API实现的树形结构构建
	 * @param list 原始数据列表
	 * @return 树形结构列表
	 */
	public static List<DictBiz> buildTreeWithStream(List<DictBiz> list) {
		Map<Long, DictBiz> nodeMap = list.stream()
			.collect(Collectors.toMap(DictBiz::getId, node -> node));

		List<DictBiz> tree = new ArrayList<>();

		list.forEach(node -> {
			Long parentId = node.getParentId();
			if (parentId != null && parentId != 0) {
				DictBiz parent = nodeMap.get(parentId);
				if (parent != null) {
					if (parent.getChildrenList() == null) {
						parent.setChildrenList(new ArrayList<>());
					}
					parent.getChildrenList().add(node);
				}
			} else {
				tree.add(node);
			}
		});

		return tree;
	}



}