Java 树形结构快速组装

如果少量数据组装树结构可以使用递归或循环都是可以的,但是对于几十 W 的数据量可能就会报栈溢出或者超时。这里使用的是Java 8 的 并发流实现,所使用的数据是省市县五级联动查询。返回属性结构。
核心代码

	@Override
	public List<SysAreaTreeDto> getTreeList(Long pid) {
		long start = System.currentTimeMillis();
		//查询所有数据
		QueryWrapper<SysAreaEntity> wrapper = Wrappers.query();
		List<SysAreaEntity> list = baseMapper.selectList(wrapper);

		if(CollectionUtils.isEmpty(list)){
			return new ArrayList<>();
		}
		//将数据转为树形结构的数据

		List<SysAreaTreeDto> dtoList = list.parallelStream().map(item->{
					SysAreaTreeDto dto = new SysAreaTreeDto();
					BeanUtils.copyProperties(item, dto);
					return dto;
		}).collect(Collectors.toList());

		List<SysAreaTreeDto> result = buildTree(dtoList, pid);

		System.out.println(System.currentTimeMillis() - start);
		return result;
	}

	/**
	 * 组装树结构
	 * @param dtoList 所有数据
	 * @param pid 父主键
	 * @return 树形结构
	 */
	private List<SysAreaTreeDto> buildTree(List<SysAreaTreeDto> dtoList, Long pid) {

		//先选出非顶级的节点
		List<SysAreaTreeDto> list = dtoList.parallelStream().filter(item -> item.getPid() != pid).collect(Collectors.toList());

		//将这些非顶级节点的数据按pid进行分组
		Map<Long, List<SysAreaTreeDto>> sub = list.parallelStream().collect(Collectors.groupingBy(SysAreaTreeDto::getPid));

		//循环设置对应的子节点(根据id = pid)
		dtoList.parallelStream().forEach(node -> node.setChildren(sub.get(node.getId())));

		//过滤掉父节点数据
		List<SysAreaTreeDto> collect = dtoList.parallelStream().filter(node -> node.getPid() == 0).collect(Collectors.toList());

		return collect;
	}

省市县五级联动 SQL 下载路径 https://download.csdn.net/download/weixin_45267345/86508057
在这里插入图片描述
在这里插入图片描述

  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值