获取树形结构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;
    }