【List<Map<String, Object>>】记录一次树形遍历方法

在我实习过程中,有一次碰到了一个遍历需求,大概是这样的:
在这里插入图片描述
我现在拿到了广州这个节点的id,然后我要把整棵树查出来,并且返回方式要和前端匹配,以父子嵌套方式返回,就是这样

在这里插入图片描述
然后这个问题折磨了我很久
在同事的帮助下才解决


首先,原对象字段是没有child字段的,是只有parentId进行关联,所以我要先创建一个DTO对象,假如有图示的两个字段要返回(因为真正的数据库表对象中是真的没有child字段的),那么我再加一个child字段

这样:
在这里插入图片描述

注意,要自己生成get和set方法
在这里插入图片描述
然后开始实现


要明白,根据传进来的id查询全部节点很简单,sql:


SELECT
	id
	name 
	parentId
FROM
	[原表] 
WHERE
	tree_name = ( SELECT tree_name FROM [原表] WHERE id ='xxx' )
	
	order by tree_level asc;
	

大概就是这么个意思
每个人需求不一样。
查出来很简单,但是进行构造有点复杂。
就是要构造一个数据结构。

 public List<CustShowDto> showWithAll(String id) {
		//list是查出来的所有节点,没有父子节点
        List<Map<String, Object>>  list = fmCgCustomInfoMapper.selectWithAll(id);
		//l是节点的属性集合,就是类似于广州
        List<CustShowDto> l = new ArrayList<>();
		//
        Map<String, List<CustShowDto>> parentMap = new HashMap<>();
		//遍历集合,把所有的节点的父节点的parentid给赋值
        for (int i = 0; i < list.size(); i++) {
            CustShowDto custShowDto = JSONObject.parseObject(JSONObject.toJSONString(list.get(i)), CustShowDto.class);
            l.add(custShowDto);
            if (StringUtils.isNotEmpty(custShowDto.getCustTreeParentId())) {
                if (parentMap.get(custShowDto.getCustTreeParentId()) == null) {
                    parentMap.put(custShowDto.getCustTreeParentId(), new ArrayList<>());
                }
                parentMap.get(custShowDto.getCustTreeParentId()).add(custShowDto);
            }
        }
        List<CustShowDto> resList = new ArrayList<>();
		//把赋值的节点添加到返回集合中
        for (int i = 0; i < l.size(); i++) {
            if (StringUtils.isEmpty(l.get(i).getCustTreeParentId())) {
                resList.add(l.get(i));
            }
            if (parentMap.get(l.get(i).getCustId()) != null) {
                l.get(i).setChildren(parentMap.get(l.get(i).getCustId()));
            }
        }
        return resList;

    }

主要是构造一个parentMap,用parentId作为key,然后根据parentId
把子节点查出来,这个子节点是parentId下面挂靠的节点,就像是:
在这里插入图片描述
然后把查出来的所有的子节点放入list中,这样就形成了

Map<String, List<CustShowDto>> parentMap = new HashMap<>();

这样大概的数据结构就出来了
在这里插入图片描述
然后构建一个

List<CustShowDto> resList = new ArrayList<>();

这个是返回的集合。
集合里面放的啥?Map呀
就是list [map1,map2,map3,...]
意思就是这么个意思,随缘看懂。
记录一下,免得下次又要做这种自己又不会!
全部代码:

 public List<CustShowDto> showWithAll(String id) {
        List<Map<String, Object>>  list = fmCgCustomInfoMapper.selectWithAll(id);
        List<CustShowDto> l = new ArrayList<>();
        Map<String, List<CustShowDto>> parentMap = new HashMap<>();
        for (int i = 0; i < list.size(); i++) {
            CustShowDto custShowDto = JSONObject.parseObject(JSONObject.toJSONString(list.get(i)), CustShowDto.class);
            l.add(custShowDto);
            if (StringUtils.isNotEmpty(custShowDto.getCustTreeParentId())) {
                if (parentMap.get(custShowDto.getCustTreeParentId()) == null) {
                    parentMap.put(custShowDto.getCustTreeParentId(), new ArrayList<>());
                }
                parentMap.get(custShowDto.getCustTreeParentId()).add(custShowDto);
            }
        }
        List<CustShowDto> resList = new ArrayList<>();
        for (int i = 0; i < l.size(); i++) {
            if (StringUtils.isEmpty(l.get(i).getCustTreeParentId())) {
                resList.add(l.get(i));
            }
            if (parentMap.get(l.get(i).getCustId()) != null) {
                l.get(i).setChildren(parentMap.get(l.get(i).getCustId()));
            }
        }
        return resList;

    }

文章仅供参考!

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值