返回树结构的通用解决方案2

逃离我推掉我的手 2024-03-22 09:31 12阅读 0赞

## 返回定制的vo ##

dto随便做：给必要参数后端可以校验即可。

vo的话，把原有的低代码生成的vo对象加入节点的参数即可：

@Data
    @ApiModel(value = "质量检查库目录树端展示对象")
    public class InspectCategoryTreeVO implements Serializable {
    
        private static final long serialVersionUID = 1L;
    
        /**
         * 主键
         */
        @TableId(value = "id", type = IdType.ASSIGN_ID)
        @ApiModelProperty(value = "主键")
        private Long id;
    
    
        /**
         * 父节点id
         */
        @ApiModelProperty(value = "父节点id")
        private Long parentId;
    
    
        /**
         * 节点路径
         */
        @ApiModelProperty(value = "节点路径")
        private String nodePath;
    
    
        /**
         * 目录编号
         */
        @ApiModelProperty(value = "目录编号")
        private String categoryNum;
    
    
        /**
         * 名称
         */
        @ApiModelProperty(value = "名称")
        private String name;
    
        /**
         * wbs模板id
         */
        @ApiModelProperty(value = "wbs模板id")
        private Long wbsTempId;
    
        /**
         * 排序
         */
        @ApiModelProperty(value = "排序")
        private Integer sort;
    
    
        /**
         * 子节点列表
         */
        @ApiModelProperty(value = "子节点列表")
        private List<InspectCategoryTreeVO> children;
    
    
        /**
         * 租户
         */
        @ApiModelProperty(value = "租户", hidden = true)
        private Long tenantId;
    
    
    }

然后进行改造，

@Data
    @ApiModel(value = "质量检查负责人管理前端展示对象")
    public class QualityInspectionPrincipalTreeVO  implements Serializable {
    
        private static final long serialVersionUID = 1L;
    
        /**
         * 主键
         */
        @TableId(value = "id", type = IdType.ASSIGN_ID)
        @ApiModelProperty(value = "主键")
        private Long id;
    
    
        /**
         * 父节点id
         */
        @ApiModelProperty(value = "父节点id")
        private Long parentId;
    
    
        /**
         * 节点路径
         */
        @ApiModelProperty(value = "节点路径")
        private String nodePath;
    
    
        /**
         * 目录编号
         */
        @ApiModelProperty(value = "目录编号")
        private String categoryNum;
    
    
        /**
         * 名称
         */
        @ApiModelProperty(value = "名称")
        private String name;
    
        /**
         * wbs模板id
         */
        @ApiModelProperty(value = "wbs模板id")
        private Long wbsTempId;
    
        /**
         * 排序
         */
        @ApiModelProperty(value = "排序")
        private Integer sort;
    
    
        /**
         * 子节点列表
         */
    
    
    
        /**
         * 租户
         */
        @ApiModelProperty(value = "租户", hidden = true)
        private Long tenantId;
    
    
    
    
    
    
        /**
         * 项目编号
         */
        @ApiModelProperty(value="项目编号")
        private String projectNum;
    
    
        @ApiModelProperty(value="项目编号Name")
        private String projectName;
    
        public void setProjectNum(String projectNum){
            this.projectNum = projectNum;
            this.projectName = ProjectCacheComponent.getProjectNameByProjectNum(projectNum);
        }
    
        /**
         * 所属部门编号
         */
        @ApiModelProperty(value="所属部门编号")
        private String deptNum;
    
    
        @ApiModelProperty(value="所属部门编号Name")
        private String deptName;
    
        public void setDeptNum(String deptNum){
            this.deptNum = deptNum;
            this.deptName = SaasDeptComponent.getDeptNameByDeptNum(deptNum);
        }
    
        /**
         * 质量检查库目录id
         */
        @ApiModelProperty(value="质量检查库目录id")
        private Long categoryId;
    
    
    
        /**
         * 责任人
         */
        @ApiModelProperty(value="责任人")
        private String principal;
    
    
    
        /**
         * 审批状态;01-未申请,02-申请中,03-审批成功,04-驳回,05-收回
         */
        @ApiModelProperty(value="审批状态;01-未申请,02-申请中,03-审批成功,04-驳回,05-收回")
        private String applicationStatus;
    
    
    
        /**
         * 是否反审;0-否,1-是
         */
        @ApiModelProperty(value="是否反审;0-否,1-是")
        private Boolean repeated;
    
    
    
        /**
         * 暂存状态;0-暂存,1-已保存
         */
        @ApiModelProperty(value="暂存状态;0-暂存,1-已保存")
        private String holdStatus;
    
    
    
        /**
         * 是否启用;1-启用,0-未启用
         */
        @ApiModelProperty(value="是否启用;1-启用,0-未启用")
        private Boolean enable;
    
    
    
        /**
         * 删除标识;0-未删除,1-已删除
         */
        @ApiModelProperty(value="删除标识;0-未删除,1-已删除")
        private String delFlag;
    
    
    
        /**
         * 版本号
         */
        @ApiModelProperty(value="版本号")
        private Integer version;
    
    
    
        /**
         * 创建人编码
         */
        @ApiModelProperty(value="创建人编码")
        private String createBy;
    
        @ApiModelProperty(value="创建人编码Name")
        private String createByName;
    
        public void setCreateBy(String createBy){
            this.createBy = createBy;
            this.createByName = SaasSysUserComponent.getSysUserNameByUserId(createBy);
        }
    
    
        /**
         * 创建时间
         */
        @JsonDeserialize(using = LocalDateTimeDeserializer.class)
        @JsonSerialize(using = LocalDateTimeSerializer.class)
        @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss", timezone="GMT+8")
        @ApiModelProperty(value="创建时间")
        private LocalDateTime createTime;
    
    
    
        /**
         * 更新人编码
         */
        @ApiModelProperty(value="更新人编码")
        private String updateBy;
    
        @ApiModelProperty(value="更新人编码Name")
        private String updateByName;
    
        public void setUpdateBy(String updateBy){
            this.updateBy = updateBy;
            this.updateByName = SaasSysUserComponent.getSysUserNameByUserId(updateBy);
        }
    
    
        /**
         * 更新时间
         */
        @JsonDeserialize(using = LocalDateTimeDeserializer.class)
        @JsonSerialize(using = LocalDateTimeSerializer.class)
        @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss", timezone="GMT+8")
        @ApiModelProperty(value="更新时间")
        private LocalDateTime updateTime;
    
    
    
    
    
    
    
    
        @ApiModelProperty(value = "子节点列表")
        private List<QualityInspectionPrincipalTreeVO> children;
    
    
    }

然后实现方法：

List getTree(QualityInspectionPrincipalTreeDTO dto);

private List<QualityInspectionPrincipalTreeVO> getChildrenCategory(QualityInspectionPrincipalTreeVO root,List<QualityInspectionPrincipalTreeVO> all){
            List<QualityInspectionPrincipalTreeVO> children = all.stream().filter(categoryEntity -> {
                //这里有bug问题 这里两者返回的是一个Long 是一个包装类(引用类型对于== 比较的是地址，而基本数据类型比较的是值)
                // == 表示的是对 对象的比较（堆内存）
                //看Integer和Long源码  -128~127之间==比较的是值（cache拿出来的），而超过了就是比较地址new 出来的
    //            return categoryEntity.getParentId() == root.getId();
                return  categoryEntity.getParentId().longValue() == root.getId().longValue();
                //方案2  底层equals转换成long 基本数据类型了
    //             categoryEntity.getParentId().equals(root.getId());
    
            }).map(categoryEntity -> {
                //找到了子分类并设置
                categoryEntity.setChildren(getChildrenCategory(categoryEntity,all));
    
                return categoryEntity;
    
            }).sorted((category1,category2) ->{
                return ( category1.getSort() == null ? 0: category1.getSort() ) -
                        (category2.getSort() == null ? 0: category2.getSort() );
            }).collect(Collectors.toList());
    
    
    
            return children;
    
    
    
        }
    
    
        public List getTree(QualityInspectionPrincipalTreeDTO dto) {
            //todo
            // 根据入参获取模版目录全部数据
            List<QualityInspectionPrincipalManage> managesList = this.list();
    //将managesList也转换成树的专用返回列表
            List<QualityInspectionPrincipalTreeVO> managesTreeList = BeanUtil.copyToList(managesList, QualityInspectionPrincipalTreeVO.class);
    //遍历managesTreeList,如果parentId为空的话，将每一个元素的categoryId赋值给parentId
            managesTreeList.stream().filter(category -> category.getParentId() == null).forEach(category -> category.setParentId(category.getCategoryId()));
    
            List<QualityInspectionPrincipalTreeVO> categoryList = new ArrayList<>();
    //根据入参获取模版目录全部数据
            List<InspectCategory> categories = inspectCategoryService.lambdaQuery()
                    .eq(ObjectUtil.isNotEmpty(dto.getTenantId()), InspectCategory::getTenantId, dto.getTenantId())
                    .eq(InspectCategory::getDelFlag, Boolean.FALSE)
                    .orderByAsc(InspectCategory::getSort)
                    .list();
    
    //将数据转换为树的专用返回列表
            if (CollectionUtil.isNotEmpty(categories)) {
                for (InspectCategory category : categories) {
                    categoryList.add(BeanUtil.copyProperties(category, QualityInspectionPrincipalTreeVO.class));
                }
            }
    
    //合并categoryList和managesTreeList
            categoryList.addAll(managesTreeList);
    
    //
    
            List<QualityInspectionPrincipalTreeVO> categoryTree = categoryList.stream().filter(categoryEntity -> {
                //过滤 返回一级分类
                return categoryEntity.getParentId() == 0;
    
            }).map(category -> {
                //进行map映射 ，给每个分类进行子分类（用递归）
                List<QualityInspectionPrincipalTreeVO> childrenCategory = getChildrenCategory(category, categoryList);
                category.setChildren(childrenCategory);
    
    
                return category;
            }).sorted((category1, category2) -> {
                return (category1.getSort() == null ? 0 : category1.getSort()) - (category2.getSort() == null ? 0 : category2.getSort());
            }).collect(Collectors.toList());
    
    
    
    
    
    
    
    
    
    这段好像没用
    
    //将数据根据父节点ID进行分组
            Map<Long, List<QualityInspectionPrincipalTreeVO>> categoryGroupMap = categoryList.stream()
                    .filter(category -> category.getParentId() != null)
                    .collect(Collectors.groupingBy(QualityInspectionPrincipalTreeVO::getParentId));
    
    //遍历categoryList，将子集放入父节点中
            categoryList.forEach(category -> category.setChildren(categoryGroupMap.get(category.getId())));
    
    这段好像没用
    
    
    
    
    
    // 将分组后的根节点返回
    //        return categoryGroupMap.get(TREE_ROOT_ID);
            return categoryTree;
        }