递归处理树状数据

废话不错说,上代码

Entity

public class TreeSelect implements Serializable {
    private static final long serialVersionUID = 1L;

    /**
     * 节点ID
     */
    private Long id;

    /**
     * 节点ID值
     */
    private Long value;

    /**
     * 节点名称
     */
    private String label;

    /**
     * 子节点
     */
    @JsonInclude(JsonInclude.Include.NON_EMPTY)
    private List<TreeSelect> children;
    
}

Controller

    /**
     *  用户树
     */
    @Log(title = "用户树" , businessType = BusinessType.UPDATE)
    @GetMapping("/userTree")
    public AjaxResult userTree() {
        return AjaxResult.success(userService.userTree());
    }

ServiceImpl

    @Override
    public List<TreeSelect> userTree() {
        List<SysUser> userList = userMapper.selectUsertLists();
        //一级菜单
        List<TreeSelect> rootTree = new ArrayList<>();
        TreeSelect treeSelect = new TreeSelect();
        treeSelect.setId(1L);
        treeSelect.setValue(1L);
        treeSelect.setLabel("用户树");
        treeSelect.setChildren(new ArrayList<>());
        rootTree.add(treeSelect);
        List<TreeSelect> roots = bulidTree(rootTree, userList);
        return roots;
    }
public List<TreeSelect> bulidTree(List<TreeSelect> rootTree, List<SysUser> userList) {
        for (TreeSelect deptTreeSelect : rootTree) {
            for (SysUser user : userList) {
                if (user.getDeptId().intValue() == deptTreeSelect.getId().intValue()) {
                    TreeSelect treeSelect = new TreeSelect();
                    treeSelect.setId(user.getUserId());
                    treeSelect.setValue(user.getUserId());
                    treeSelect.setLabel(user.getNickName());
                    if (null != deptTreeSelect.getChildren()) {
                        deptTreeSelect.getChildren().add(treeSelect);
                    }
                }
            }
            if (null != deptTreeSelect.getChildren() && deptTreeSelect.getChildren().size() > 0) {
                bulidTree(deptTreeSelect.getChildren(), userList);
            }
        }
        return rootTree;
    }

 

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
/** * 根据等级查询类目树 * * @param level * @return */ @Override public List queryCategoryTree(Integer level) { //查询当前级别下类目 List list = categoryDAO.list(level); //组装好的类目树,返回前端 List categoryTree = new ArrayList(); //所有类目 List allDTOList = new ArrayList(); if (CollectionUtils.isEmpty(list)) { return categoryTree; } for (CategoryDO categoryDO : list) { allDTOList.add(new CategoryTreeDTO().convertDOToDTO(categoryDO)); } //当前等级类目 categoryTree = allDTOList.stream().filter(dto -> level.equals(dto.getLevel())).collect(Collectors.toList()); for (CategoryTreeDTO categoryTreeDTO : categoryTree) { //组装类目为树结构 assembleTree(categoryTreeDTO, allDTOList,Constants.CATEGORY_MAX_LEVEL - level); } return categoryTree; } /** * 组装树 * * @param categoryTreeDTO * @param allList * @param remainRecursionCount 剩余递归次数 * @return */ public CategoryTreeDTO assembleTree(CategoryTreeDTO categoryTreeDTO, List allList, int remainRecursionCount) { remainRecursionCount--; //最大递归次数不超过Constants.CATEGORY_MAX_LEVEL-level次,防止坏数据死循环 if(remainRecursionCount < 0){ return categoryTreeDTO; } String categoryCode = categoryTreeDTO.getCategoryCode(); Integer level = categoryTreeDTO.getLevel(); //到达最后等级树返回 if (Constants.CATEGORY_MAX_LEVEL == level) { return categoryTreeDTO; } //子类目 List child = allList.stream().filter(a -> categoryCode.equals(a.getParentCode())).collect(Collectors.toList()); if (null == child) { return categoryTreeDTO; } categoryTreeDTO.setChildren(child); //组装子类目 for (CategoryTreeDTO dto : child) { assembleTree(dto, allList,remainRecursionCount); } return categoryTreeDTO; }

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值