java实现多级递归查询

一、java实现多级递归查询

1、实体类对象(省略了一些字段)

@Data
@EqualsAndHashCode(callSuper = true)
@Accessors(chain = true)
@TableName("SYS_USER")
public class User {

    private static final long serialVersionUID = -3089138238598051744L;
    /**
     * 账号
     */
    @TableField("NAME")
    private String name;

    /**
     * 密码
     */
    @TableField("PASSWORD")
    private String password;

    /**
     * 昵称
     */
    @TableField("NICK_NAME")
    private String nickName;

    /**
     * 备注
     */
    @TableField("REMARK")
    private String remark;

    /**
     * 工号
     */
    @TableField("WORK_NUM")
    private Integer workNum;

    

   	@TableField("PARENT_ID")
    private String parentId;


    @TableField(exist = false)
    private List<User> children;
}

2、获取所有用户集合并调用子类查询方法
此处用了封装的HttpResponses返回参数类以及mybatisPlus查询集合

	/**
     * 查询用户等级
     * @return
     */
    @GetMapping("/userTree")
    public HttpResponses<List<User>> findTree(){
        //根节点
        List<User> rootUser = new ArrayList<User>();
        try {//查询所有菜单
            List<User> allUsers = this.userService.query().eq(User::getStatus,StatusEnum.NORMAL)
                                                .eq(User::getDirty,StatusEnum.NORMAL).list();
            for (User nav : allUsers) {
                if(nav.getLevelId().equals("0")){//父节点是0的,为根节点。
                    rootUser.add(nav);
                }
            }
            /* 根据Menu类的order排序 */
            //Collections.sort(rootUser, order());
            //为根菜单设置子菜单,getClild是递归调用的
            for (User nav : rootUser) {
                /* 获取根节点下的所有子节点 使用getChild方法*/
                List<User> childList = getChild(nav.getId(), allUsers);
                nav.setChildren(childList);//给根节点设置子节点
            }

        } catch (Exception e) {

        }
        return success(rootUser);
    }

3、获取子类用户

/**
     * 获取子节点
     * @param id 父节点id
     * @param allUser 所有菜单列表
     * @return 每个根节点下,所有子菜单列表
     */
    public List<User> getChild(String id,List<User> allUser){
        //子菜单
        List<User> childList = new ArrayList<User>();
        for (User nav : allUser) {
            // 遍历所有节点,将所有菜单的父id与传过来的根节点的id比较
            //相等说明:为该根节点的子节点。
            if(nav.getLevelId().equals(id)){
                childList.add(nav);
            }
        }
        //递归
        for (User nav : childList) {
            nav.setChildren(getChild(nav.getId(), allUser));
        }
        //Collections.sort(childList,order());//排序
        //如果节点下没有子节点,返回一个空List(递归退出)
        if(childList.size() == 0){
            return new ArrayList<User>();
        }
        return childList;
    }

4、打完收工,结果自己调用看下就可以了

/** * 根据等级查询类目树 * * @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、付费专栏及课程。

余额充值