后端返回树结构

出参结构

java 复制代码
@Getter
@Setter
public class TreeResponse implements Serializable {
    // 主键
    private Long id;
    // 父级节点
    private Long parentId;
    // 层级
    private Byte layer;
    // 编码
    private String docCode;
    // 名称
    private String docName;
    // 子节点
    private List<TreeResponse> childNode;
}

方案一:生撸

java 复制代码
public Result<List<TreeResponse>> listTreeData(String code) {
    // 查询全量数据
    List<TreeResponse> tempTreeResponse = new ArrayList<>();
    // 树节点转换
    return Result.success(convertTreeStructure(tempTreeResponse));
}

/**
 * 树形转换
 *
 * @param tempTreeResponse 档案信息
 * @return 档案信息-树
 */
private List<TreeResponse> convertTreeStructure(List<TreeResponse> tempTreeResponse) {
    // 获取第一层级
	List<TreeResponse> treeResponse = tempTreeResponse.stream()
			.filter(value -> Objects.isNull(value.getParentId()))
			.collect(Collectors.toList());
    // 获取非第一层级数据,并以父级ID进行分组
	Map<Long, List<TreeResponse>> notFirstLayerData = tempTreeResponse.stream()
			.filter(value -> Objects.nonNull(value.getParentId()))
			.collect(Collectors.groupingBy(TreeResponse::getParentId));
	treeResponse.forEach(data -> setChildData(data, notFirstLayerData));
	return treeResponse;
}

/**
 * 设置子节点
 *
 * @param parentInfo        父节点信息
 * @param notFirstLayerData 非首节点信息
 */
private void setChildData(TreeResponse parentInfo, Map<Long, List<TreeResponse>> notFirstLayerData) {
	List<TreeResponse> childData = notFirstLayerData.get(parentInfo.getId());
	if (CollectionUtils.isNotEmpty(childData)) {
		parentInfo.setChildNode(childData);
		childData.forEach(data -> setChildData(data, notFirstLayerData));
	}
}

方案二:Hutool

java 复制代码
//配置
TreeNodeConfig treeNodeConfig = new TreeNodeConfig();
// 自定义属性名 都要默认值的
treeNodeConfig.setWeightKey("order");
treeNodeConfig.setIdKey("rid");
// 最大递归深度
treeNodeConfig.setDeep(3);

//转换器 (含义:找出父节点为字符串零的所有子节点, 并递归查找对应的子节点, 深度最多为 3)
List<Tree<String>> treeNodes = TreeUtil.<TreeNode, String>build(nodeList, "0", treeNodeConfig,
		(treeNode, tree) -> {
			tree.setId(treeNode.getId());
			tree.setParentId(treeNode.getParentId());
			tree.setWeight(treeNode.getWeight());
			tree.setName(treeNode.getName());
			// 扩展属性 ...
			tree.putExtra("extraField", 666);
			tree.putExtra("other", new Object());
		});
相关推荐
ChoSeitaku4 小时前
链表交集相关算法题|AB链表公共元素生成链表C|AB链表交集存放于A|连续子序列|相交链表求交点位置(C)
数据结构·考研·链表
香菜大丸4 小时前
链表的归并排序
数据结构·算法·链表
jrrz08284 小时前
LeetCode 热题100(七)【链表】(1)
数据结构·c++·算法·leetcode·链表
南宫生5 小时前
贪心算法习题其四【力扣】【算法学习day.21】
学习·算法·leetcode·链表·贪心算法
萌面小侠Plus9 小时前
Android笔记(三十三):封装设备性能级别判断工具——低端机还是高端机
android·性能优化·kotlin·工具类·低端机
羊小猪~~16 小时前
数据结构C语言描述2(图文结合)--有头单链表,无头单链表(两种方法),链表反转、有序链表构建、排序等操作,考研可看
c语言·数据结构·c++·考研·算法·链表·visual studio
躺不平的理查德1 天前
数据结构-链表【chapter1】【c语言版】
c语言·开发语言·数据结构·链表·visual studio
半夜不咋不困1 天前
单链表OJ题(3):合并两个有序链表、链表分割、链表的回文结构
数据结构·链表
一颗星星辰2 天前
数据结构 | 题目练习第二章 | 合并两个有序链表 | 环形链表 | 环形链表入环第一个节点
网络·数据结构·链表
2401_858286112 天前
L6.【LeetCode笔记】合并两个有序链表
c语言·开发语言·笔记·leetcode·链表