Java: 返回树形结构数据

本文介绍了在Java中如何构建并返回树形结构数据,包括TreeNodeDto节点类和TreeUtil工具类的详细使用。

Java - 返回树形结构数据


1.TreeNodeDto

@Data
public class TreeNodeDto {

    private Long id;

    private Long organizationId;

    private Long parentOrgId;

    private Integer level;

    private Integer sort;

    private String path;

    private String code;

    private String name;

    private String note;

    private List children = new ArrayList();
}

2.TreeUtil

/**
 * 返回树形结构数据
 *
 * @author dijk
 */
public class TreeUtil {

    public static List<TreeNodeDto> formatter(List<TreeNodeDto> list) {
        //定义map集合封装参数
        Map<Long, TreeNodeDto> map = new HashMap<>(50);
        for (TreeNodeDto treeNode : list) {
            //遍历list将结点id封装进map中
            map.put(treeNode.getOrganizationId(), treeNode);
        }

        //设置父子级关系
        return addChildren(list, map);
    }

    private static List<TreeNodeDto> addChildren(List<TreeNodeDto> list, Map<Long, TreeNodeDto> map) {
        List<TreeNodeDto> rootNodes = new ArrayList<>();
        for (TreeNodeDto treeNode : list) {
            TreeNodeDto parentNode = map.get(treeNode.getParentOrgId());
            if (parentNode != null) {
                //存在父级结点
                parentNode.getChildren().add(treeNode);
            } else {
                //没有父级节点,即本身为根结点
                rootNodes.add(treeNode);
            }
        }
        return rootNodes;
    }
}
/** * 根据等级查询类目树 * * @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; }
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值