递归处理组织的几种情况

前言

近期在工作中的一点小总结，在此进行记录。

递归封装树形结构组织

/**
  * 树形结构封装
  *
  * @param root
  * @param all
  * @return
  */
 private List<OrgVO> getChildren(OrgVO root, List<OrgVO> all) {
    
  return all.stream().filter(orgVO ->
   root.getId().equals(orgVO.getParentId())).map(orgVO -> {
    
   orgVO.setChildrenEntity(getChildren(orgVO, all));
   return orgVO;
  }).sorted(Comparator.comparingInt(c -> (c.getSort() == null ? 0 : c.getSort()))).collect(Collectors.toList());
 }

递归获取子孙机构

/**
     * 递归获取子孙机构
     * @param orgCode
     * @return
     */
    private List<Org> getAllChildren(String orgCode){
    
        List<Org> allChildren = new ArrayList<>();
        List<Org> orgListByParent = OrgService.getOrgListByParent(orgCode);
        // 这里判断 size 大于 1 的原因是 getOrgListByParent() 方法如果没有获取到子孙机构则返回当前机构信息
        if (CollectionUtils.isNotEmpty(orgListByParent) && orgListByParent.size() > 1){
    
            allChildren.addAll(orgListByParent);
            for (Org org : orgListByParent) {
    
                List<Org> children = getAllChildren(organization.getOrgCode());
                allChildren.addAll(children);
            }
        }
        return allChildren;
    }

递归累加子孙机构人数

 /**
     * 递归统计当前+子孙部门人数，并返回当前部门及子级
     * @param root
     * @param data
     * @return
     */
    private List<UserDTO> getUserNum(UserDTO root, List<UserDTO> data){
    
        Integer userCount = root.getUserCount();
        data.forEach(dto -> {
    
            if (root.getOrgCode().equals(dto.getParentCode())){
    
                root.setUserCount(userCount+dto.getUserCount());
                List<UserDTO> collect = data.stream().filter(dto1 -> dto1.getParentCode().equals(dto.getOrgCode())).collect(Collectors.toList());
                if (CollectionUtils.isNotEmpty(collect)){
    
                    getUserNum(dto,collect);
                }

            }
        });
        List<UserDTO> collect = data.stream().filter(userDTO -> userDTO.getParentCode().equals(root.getOrgCode())).collect(Collectors.toList());
        root.userCount(collect.stream().map(UserDTO::getUserCount).mapToInt(Integer::intValue).sum()+userCount);
        collect.add(0,root);
      return collect;

    }

总结

递归统计人数该方法可能会出现异常，使用中需注意。

- END -