树结构数据补全路径信息

最新推荐文章于 2024-10-10 09:56:47 发布

luxc666

最新推荐文章于 2024-10-10 09:56:47 发布

阅读量271

点赞数

文章标签： java

本文链接：https://blog.csdn.net/main_luxc/article/details/113932226

版权

思路:

1.先把要补全路劲的全部数据找出

2.递归列表,需要一个额外字段fakeId标识,每次找到列表中最末级的数据,把路径字段往前补全id,同时把fakeId字段赋值为上一级的id

3.每递归一次,把最末级的去掉,根据fakeId再补全路径

    @Override
    public void getOrgPath() {
        //java获取org全路径
        long startTime = System.currentTimeMillis();
        List<OrgDTO> all = orgMapper.getAll();
        long midTime = System.currentTimeMillis();
        log.info("查询耗时：" + (midTime - startTime) + "ms");

        List<OrgDTO> resultList = orgMapper.getAll();
        List<OrgDTO> dealedList = new ArrayList<>();

        List<OrgDTO> endList = fillPath(all, resultList, dealedList);
    }


    List<OrgDTO> fillPath(List<OrgDTO> all, List<OrgDTO> resultList, List<OrgDTO> dealedList) {
        if (resultList == null || resultList.isEmpty()) {
            return dealedList;
        }
        Map<String, List<OrgDTO>> collect = resultList.stream().collect(Collectors.groupingBy(OrgDTO::getZsporgid));
        //最末级数据
        List<OrgDTO> lastItemList = resultList.stream().filter(item -> !collect.containsKey(item.getPkId())).collect(Collectors.toList());
        Map<String, String> lastMap = lastItemList.stream().collect(Collectors.toMap(OrgEntity::getPkId, OrgEntity::getZsporgid));
        if (dealedList != null && !dealedList.isEmpty()) {
            for (OrgDTO orgEntity : dealedList) {
                if (lastMap.containsKey(orgEntity.getFakeId())) {
                    lastItemList.add(orgEntity);
                }
            }
        }
        for (OrgDTO orgDTO : lastItemList) {
            if (StringUtils.isEmpty(orgDTO.getFakeId())) {
                orgDTO.setOrgPath(orgDTO.getPkId());
                orgDTO.setFakeId(orgDTO.getZsporgid());
            } else {
                orgDTO.setOrgPath(orgDTO.getFakeId() + "," + orgDTO.getOrgPath());
                //如果是旧的deallist,需要赋值新的fakeId
                orgDTO.setFakeId(lastMap.get(orgDTO.getFakeId()));
            }
        }
        dealedList.addAll(lastItemList);
        //去重
        Map<String, OrgDTO> map = dealedList.stream().collect(Collectors.toMap(OrgDTO::getPkId, item -> item, (k1, k2) -> k1));
        dealedList = map.entrySet().stream().map(e -> e.getValue()).collect(Collectors.toList());
        resultList = all.stream().filter(item -> !map.containsKey(item.getPkId())).collect(Collectors.toList());
        if (resultList == null || resultList.isEmpty()) {
            return dealedList;
        } else {
            return fillPath(all, resultList, dealedList);
        }
    }