java循环遍历树状结构对象并转换成另外树状结构对象

项目工程需要调用外部服务接口返回树状结构对象,不能直接使用,需要再次转换处理,满足实际的项目需求,涉及到两点处理:
1、遍历树状对象

//外部服务对象
public class DepartmentVO {

    private Integer id;
    private Integer orgId;
    private String code;
    private String name;
    private String nameZh;
    private Integer parentId;
    private String parentIds;
    private DepartmentStatus status;
    private DepartmentType type;
    private Integer ownerId;
    private Integer rank;
//    private Instant createdAt;
//    private Instant updatedAt;
    private Integer createdBy;
    private Integer updatedBy;
    private String userCode;
    private String username;
    private List<DepartmentVO> subDepartments;

}


//需要使用的树状结构对象
public class McmpDepartmentDto implements Serializable {

  @ApiModelProperty(value = "部门ID", example = "0")
  private Integer id;
  @ApiModelProperty(value = "租户ID", example = "0")
  private Integer orgId;
  @ApiModelProperty(value = "部门编码")
  private String code;
  @ApiModelProperty(value = "部门英文名称")
  private String name;
  @ApiModelProperty(value = "部门中文名称")
  private String nameZh;
  @ApiModelProperty(value = "父级ID")
  private Integer parentId;
  @ApiModelProperty(value = "父级(路径)")
  private String parentIds;
  @ApiModelProperty(value = "级别(1,2,3,4....)")
  private String level;
  private List<McmpDepartmentDto> subDepartments = new ArrayList<>();
  public void addChild(McmpDepartmentDto child) {
    subDepartments.add(child);
  }

  /**
   * 查找指定ID的节点,并返回该节点及其所有子节点的ID集合
   * @param root 树的根节点
   * @param targetId 目标节点的ID
   * @return 包含目标节点及其所有子节点ID的集合
   */
  public static Set<Integer> findNodeAndDescendants(DepartmentVO root, Integer targetId) {
    Set<Integer> ids = new HashSet<>();
    searchNodeAndCollectIds(root, targetId, ids);
    return ids;
  }

  private static void searchNodeAndCollectIds(DepartmentVO node, Integer targetId, Set<Integer> ids) {
    if (node.getId().equals(targetId)) {
      ids.add(node.getId());
      collectChildIds(node, ids);
    } else {
      for (DepartmentVO child : node.getSubDepartments()) {
        searchNodeAndCollectIds(child, targetId, ids);
      }
    }
  }

  private static void collectChildIds(DepartmentVO node, Set<Integer> ids) {
    ids.add(node.getId());
    for (DepartmentVO child : node.getSubDepartments()) {
      collectChildIds(child, ids);
    }
  }

//调用地方
public List<McmpDepartmentDto> searchDepartmentTree(Long tenantId){
        List<McmpDepartmentDto> mcmpDepartmentDtos = new ArrayList<>();
        DepartmentSearchForm departmentSearchForm = new DepartmentSearchForm();
        departmentSearchForm.setOrgId(tenantId.intValue());
        //外部服务返回的集合对象,里面对象就是树状结构
        List<DepartmentVO> result =  iamDepartmentClient.searchTree(departmentSearchForm).getData();
        if(!CollectionUtils.isEmpty(result)){
            for(DepartmentVO departmentVO :result){
                mcmpDepartmentDtos.add(traverseTreeRecursive(departmentVO));
            }
        }
        return mcmpDepartmentDtos;
    }


    public McmpDepartmentDto traverseTreeRecursive(DepartmentVO departmentVO) {

        if (departmentVO == null) {
            return null;
        }
        McmpDepartmentDto mcmpDepartmentDto = new McmpDepartmentDto();
        BeanUtils.copyProperties(departmentVO,mcmpDepartmentDto);
        mcmpDepartmentDto.setLevel(getLevel(mcmpDepartmentDto.getParentIds()));
        // 遍历子节点
        for (DepartmentVO child : departmentVO.getSubDepartments()) {
            mcmpDepartmentDto.addChild(traverseTreeRecursive(child));
        }
        return mcmpDepartmentDto;
    }

    /**
     * 通过当前所属父级的路径信息,返回当前节点所属于的Level层级
     * @param parentIds
     * 举例:最顶级阶段 parentIds=null 表示一级; parentIds=/34959/ 表示二级 ; parentIds=/34959/35079/ 表示三级,
     * @return
     */
    public String getLevel(String parentIds){
        if(null==parentIds || parentIds.equals("")){
            return "1";
        }
        return String.valueOf(countCharUsingRegex(parentIds,"/"));
    }

    /**
     * 字符串中出现某个特殊符号的次数
     * @param str
     * @param regex
     * @return
     */
    public static int countCharUsingRegex(String str, String regex) {
        Pattern pattern = Pattern.compile(regex);
        Matcher matcher = pattern.matcher(str);
        int count = 0;
        while (matcher.find()) {
            count++;
        }
        return count;
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值