树形结构的一种便捷实现方案

背景

在开发过程中经常需要把平铺的数据结构转为树形的数据结构,例如多级菜单、组织机构等。

实现方案有很多种。

1、可以使用递归查询,但是这样数据一多会导致频繁的多次查询数据库,产生很多额外的IO开销,总体的响应时间会比较慢,一般不会这样做。

2、可以事先查询出来所有的数据,再进行递归的子节点查找,这是一个可行的方案,只需要查询一次数据库,之后的操作利用算法在内存操作,这样响应时间会有一个很大的提升。

3、这里要说的一种方案和第二种类似,只不过采用了google的guava包下的Multimap这种数据结构,利用它可以一个key对应多个值的特性。

方案实现

引入guava包

java 复制代码
<dependency>
	<groupId>com.google.guava</groupId>
	<artifactId>guava</artifactId>
    <version>33.2.0-jre</version>
</dependency>

<!-- 这个包可以不要,这里我用来转json字符串打印出来有用到 -->
<dependency>
   <groupId>com.alibaba</groupId>
   <artifactId>fastjson</artifactId>
   <version>1.2.83</version>
</dependency>

树形VO

java 复制代码
@Data
public class TreeVO {

    private List<TreeVO> children;
    private int id;
    private boolean leaf;
    private String menuName;
    private int parentId;
}

转树示例代码

java 复制代码
public static void main(String[] args) {
		TreeVO v1 = new TreeVO();
		v1.setId(10L);
		v1.setParentId(0L);
		v1.setMenuName("第一级菜单");

		TreeVO  v2 = new TreeVO();
		v2.setId(11L);
		v2.setParentId(10L);
		v2.setMenuName("第二级菜单1");

		TreeVO v21 = new TreeVO();
		v21.setId(12L);
		v21.setParentId(10L);
		v21.setMenuName("第二级菜单2");

		TreeVO v3 = new TreeVO();
		v3.setId(21L);
		v3.setParentId(11L);
		v3.setMenuName("第三级菜单");

		Multimap<Long,TreeVO> multimap = ArrayListMultimap.create();
		multimap.put(v1.getParentId(),v1);
		multimap.put(v2.getParentId(),v2);
		multimap.put(v21.getParentId(),v21);
		multimap.put(v3.getParentId(),v3);

		Iterator<TreeVO> iterator = multimap.values().iterator();
		while (iterator.hasNext()) {
			TreeVOmenuNode = iterator.next();
			// 找直接后代 children
			Collection<TreeVO> children = multimap.get(menuNode.getId());
			if (children.isEmpty()) {
				menuNode.setLeaf(true);
				menuNode.setChildren(null);
			} else {
				menuNode.setChildren(children);
			}
		}
		System.out.println(JSON.toJSONString(multimap.get(0L),SerializerFeature.PrettyFormat));
	}

这里打印出来的结果是

{ "children":\[ { "children":\[ { "id":21, "leaf":true, "menuName":"第三级菜单", "parentId":11 } \], "id":11, "leaf":false, "menuName":"第二级菜单1", "parentId":10 }, { "id":12, "leaf":true, "menuName":"第二级菜单2", "parentId":10 } \], "id":10, "leaf":false, "menuName":"第一级菜单", "parentId":0 }

相关推荐
软件开发-NETKF88884 分钟前
JSP到Tomcat特详细教程
java·开发语言·tomcat·jsp·项目运行
渣哥9 分钟前
为什么 JDK 1.8 要给 HashMap 加红黑树?
java
我登哥MVP10 分钟前
Java 网络编程学习笔记
java·网络·学习
大厂码农老A16 分钟前
面试官:“聊聊你最复杂的项目?” 为什么90%的候选人第一句就栽了?
java·面试
爱读源码的大都督22 分钟前
Java已死?别慌,看我如何用Java手写一个Qwen Code Agent,拯救Java
java·人工智能·后端
lssjzmn22 分钟前
性能飙升!Spring异步流式响应终极指南:ResponseBodyEmitter实战与架构思考
java·前端·架构
LiuYaoheng39 分钟前
【Android】View 的基础知识
android·java·笔记·学习
勇往直前plus1 小时前
Sentinel微服务保护
java·spring boot·微服务·sentinel
星辰大海的精灵1 小时前
SpringBoot与Quartz整合,实现订单自动取消功能
java·后端·算法
小鸡脚来咯1 小时前
一个Java的main方法在JVM中的执行流程
java·开发语言·jvm