java代码递归部门结构树

组织全部部门树,以及条件查询部门树:
    /**
     * 组织部门树
     * @return
     */
    @Override
    public List<SxyBranchVO> findAllBranchTree(String branchname) {
        SxyBranchVO sxyBranchVOParam = new SxyBranchVO();
        sxyBranchVOParam.setBranchcode("");
        sxyBranchVOParam.setDeleteflag("1");
        // 查询全部根节点root
        List<SxyBranchVO> sxyBranchVOList = sxyBranchMapper.findParentBranchVO(sxyBranchVOParam);
        if(StringUtils.isNotBlank(branchname)){ // 按部门名称查询,将符合条件的节点组织成结构树
            Set<SxyBranchVO> branchVOSet = new HashSet<>();
            Set<SxyBranchVO> branchVOSetSub = new HashSet<>();
            // 查询全部符合条件的子节点
            List<SxyBranchVO> sxyBranchVOS = sxyBranchMapper.getBranchsByName(branchname);
            for(SxyBranchVO sxyBranchVOSub : sxyBranchVOS){
                sxyBranchVOSub.setDeleteflag("1");
                branchVOSet.add(sxyBranchVOSub);
                // 向上递归,将符合条件的节点以及其全部上级父节点放在set集合中
                branchVOSetSub = buildTreeUp(sxyBranchVOSub,branchVOSetSub);
                for(SxyBranchVO sbv : branchVOSetSub){
                    branchVOSet.add(sbv);
                }
            }
            // 组织结构树,将符合条件的全部节点组织进结构树
            List<SxyBranchVO> sxyBranchVOListNew = new ArrayList<>();
            for(SxyBranchVO sxyBranchVO : sxyBranchVOList){
                // 组织结构树,根节点下的全部部门
                sxyBranchVO.setDeleteflag("1");
                buildTree(sxyBranchVO,branchVOSet);
                if(sxyBranchVO.getChildren() != null && sxyBranchVO.getChildren().size() > 0){
                    sxyBranchVOListNew.add(sxyBranchVO);
                }
            }
            return sxyBranchVOListNew;
        }else{ // 查询全部,将全部节点组织成结构树
            for(SxyBranchVO sxyBranchVO : sxyBranchVOList){
                // 组织结构树,根节点下的全部部门
                sxyBranchVO.setDeleteflag("1");
                buildTreeAll(sxyBranchVO);
            }
            return sxyBranchVOList;
        }
    }

    /**
     * 向上递归查询全部父节点(每一层的父节点只有一个)
     * @param sxyBranchVOSub
     * @param branchVOSetSub
     * @return
     */
    private Set<SxyBranchVO> buildTreeUp(SxyBranchVO sxyBranchVOSub, Set<SxyBranchVO> branchVOSetSub) {
        // 向上递归查询全部父节点
        SxyBranchVO sxyBranchVO = sxyBranchMapper.findParentBranchVOByParentCode(sxyBranchVOSub);
        if(sxyBranchVO != null){ // 若是不是根节点
            branchVOSetSub.add(sxyBranchVO);
            sxyBranchVO.setDeleteflag("1");
            buildTreeUp(sxyBranchVO,branchVOSetSub);
        }
        return branchVOSetSub;
    }

    /**
     * 递归查询子节点(向下递归,组织符合条件的结构树)
     * @param sxyBranchVO
     */
    private void buildTree(SxyBranchVO sxyBranchVO,Set<SxyBranchVO> branchVOSet) {
        List<SxyBranchVO> sxyBranchVOListNew = new ArrayList<>();
        // 查询直属子节点
        List<SxyBranchVO> sxyBranchVOList = sxyBranchMapper.findParentBranchVO(sxyBranchVO);
        for(SxyBranchVO sxyBranch : sxyBranchVOList){
            for(SxyBranchVO sbr : branchVOSet){
                if(sxyBranch.getBranchcode().equals(sbr.getBranchcode())){
                    sxyBranchVOListNew.add(sxyBranch);
                }
            }
        }
        sxyBranchVO.setChildren(sxyBranchVOListNew);
        for(SxyBranchVO sxyBranchItem : sxyBranchVOList){
            sxyBranchItem.setDeleteflag("1");
            buildTree(sxyBranchItem,branchVOSet);
        }
    }

    /**
     * 递归查询子节点(向下递归,组织全部结构树)
     * @param sxyBranchVO
     */
    private void buildTreeAll(SxyBranchVO sxyBranchVO) {
        // 查询直属子节点
        List<SxyBranchVO> sxyBranchVOList = sxyBranchMapper.findParentBranchVO(sxyBranchVO);
        sxyBranchVO.setChildren(sxyBranchVOList);
        for(SxyBranchVO sxyBranchItem : sxyBranchVOList){
            sxyBranchItem.setDeleteflag("1");
            buildTreeAll(sxyBranchItem);
        }
    }
 
 

 注释掉代码,供参考:app

/*private Boolean buildTreeMatch(SxyBranchVO sxyBranchVO,List<SxyBranchVO> sxyBranchVOS) { Boolean flag = false; // 查询直属子节点 List<SxyBranchVO> sxyBranchVOList = sxyBranchMapper.findParentBranchVO(sxyBranchVO); for(SxyBranchVO sb : sxyBranchVOS){ for(SxyBranchVO sbv : sxyBranchVOList){ if(sb.getBranchcode().equals(sbv.getBranchcode())){ return true; } } } if(!flag){ sxyBranchVO.setChildren(sxyBranchVOList); for(SxyBranchVO sxyBranchItem : sxyBranchVOList){ sxyBranchItem.setDeleteflag("1"); buildTreeMatch(sxyBranchItem,sxyBranchVOS); } } return flag; }*/
相关文章
相关标签/搜索