获取树形结构java

 public static List<ProductBriefCodePO> getTreeList(List<ProductBriefCodePO> list,String parentCode){

            //获取所有头节点
            List<ProductBriefCodePO> rootNode = new ArrayList<>();
            for (ProductBriefCodePO treeData: list ){
                if (treeData.getCode().equals(parentCode)){
                    rootNode.add(treeData);
                }
            }
            //头节点不存在的情况
            if (rootNode.size() == 0){
                return getChild(parentCode,list);
            }
            //头节点存在的情况
            for (ProductBriefCodePO treeData : rootNode){
                String code = treeData.getCode();
                treeData.setChildrenList(getChild(code,list));
            }
            return rootNode;
        }

    private static List<ProductBriefCodePO> getChild(String code, List<ProductBriefCodePO> list){
        //找到id节点子节点
        List<ProductBriefCodePO> childList = new ArrayList<>();
        for (ProductBriefCodePO treeData: list){
            if (treeData.getParentCode().equals(code)){
                childList.add(treeData);
            }
        }
        //给子节点设置子节点
        for (ProductBriefCodePO treeData : childList){
            code = treeData.getCode();
            //递归
            treeData.setChildrenList(getChild(code,list));
        }
        if (childList.size() == 0){
            return null;
        }
        return childList;
    }
public class ProductBriefCodePO implements Serializable {

    /**
     * id
     */
    private Long id;

    /**
     * 名称
     */
    private String name;

    /**
     * 编码
     */
    private String code;

    /**
     * 父编码
     */
    private String parentCode;

    /**
     * 级别
     */
    private Integer level;

    /**
     * 备注
     */
    private String remark;

    /**
     * 版本号:默认0, 每次更新1
     */
    private Integer versionNo;

    /**
     * 是否可用:默认1 0否;1是
     */
    private int isAvailable;

    /**
     * 是否逻辑删除:默认0未删除;1已删除
     */
    private int isDeleted;

    /**
     * 创建人id
     */
    private Long createUserid;

    /**
     * 创建人姓名
     */
    private String createUsername;

    /**
     * 创建时间-应用操作时间
     */
    private Date createTime;

    /**
     * 创建时间-数据库操作时间
     */
    private Date createTimeDb;

    /**
     * 服务器ip
     */
    private String serverIp;

    /**
     * 最后修改人id
     */
    private Long updateUserid;

    /**
     * 最后修改人姓名
     */
    private String updateUsername;

    /**
     * 最后修改时间
     */
    private Date updateTime;

    /**
     * 最后修改时间-数据库默认写入时间
     */
    private Date updateTimeDb;

    /**
     * 公司id
     */
    private Long companyId;
    private List<ProductBriefCodePO> childrenList;

    public List<ProductBriefCodePO> getChildrenList() {
        return childrenList;
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
/** * 根据等级查询类目树 * * @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、付费专栏及课程。

余额充值