组织架构树的查询

该博客介绍了如何在Java项目中处理组织架构的多层父子关系查询。通过创建实体类ExampleVO,定义id和parent_id字段,然后编写SQL查询语句获取数据。在Service层,利用stream进行数据处理,构建父子级联的树形结构。此方法适用于需要展示层级关系的业务场景。
摘要由CSDN通过智能技术生成

在项目的实际开发中,肯定会遇到需要对组织架构树进行查询的业务场景,也就是多层的父子级关系查询
首先创建实体类

@Data
public class ExampleVO implements Serializable {

    private static final long serialVersionUID = 1L;

    /**
     * id
     */
    @ApiModelProperty(value = "id")
    private Long id;

    /**
     * 父级id
     */
    @ApiModelProperty(value = "父级id")
    private Long parentId;


    private List<ExampleVO> sons = new ArrayList<>();

}

然后写查询语句

select id,parent_id from example

最后写service层逻辑

	@Override
    public List<ExampleVO> onExamples() {
        List<ExampleVO> exampleVOs = exampleMapper.onExamples();
        ExampleVO exampleVO = new ExampleVO();
        exampleVO.setId((long)0);
        exampleVOs.add(exampleVO);
       
        Map<Long, ExampleVO> map = exampleVOs.stream().collect(Collectors.toMap(ExampleVO::getId, s -> s));
        for (ExampleVO exampleVO : exampleVOs) {
            Long parentId = exampleVO.getParentId();
            ExampleVO vo = null;
            if (null != parentId && null != (vo = map.get(parentId))) {
                List<ExampleVO> children = vo.getSons();
                if (null == children) {
                    children = new ArrayList<>();
                    vo.setSons(children);
                }
                children.add(exampleVO);
            }
        }
        
        List<ExampleVO> list = new ArrayList<>();
        list.add(map.get((long)0));

        return list;
	}

可以根据需要,添加需要的字段,目前为了举例,只用了id和parent_id两个必须的字段

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

余额充值