Java后台转换成树结构

素颜马尾好姑娘i 2022-11-15 11:46 160阅读 0赞

package cn.ccccltd.ccccoi.mediumterminvestment.domain;
    
    import java.io.Serializable;
    import java.util.List;
    
    /**
    * @Description:返回前端，业务模块
    * @author yujiaqi
    * @date: 2019年9月9日下午2:07:19
    */
    public class ProjectBasicInfoDTO implements Serializable {
    
    private static final long serialVersionUID = 1L;
    
    private String id;
    private String pId;
    private String key;
    private String title;
    private String value;
    private List<ProjectBasicInfoDTO> children;
    private Boolean selectable;
    private Boolean disableCheckbox;
    private Boolean disabled;
    public String getId() {
    return id;
    }
    public String getValue() {
    return value;
    }
    public void setValue(String value) {
    this.value = value;
    }
    public void setId(String id) {
    this.id = id;
    }
    public String getpId() {
    return pId;
    }
    public void setpId(String pId) {
    this.pId = pId;
    }
    public String getKey() {
    return key;
    }
    public void setKey(String key) {
    this.key = key;
    }
    public String getTitle() {
    return title;
    }
    public void setTitle(String title) {
    this.title = title;
    }
    public List<ProjectBasicInfoDTO> getChildren() {
    return children;
    }
    public void setChildren(List<ProjectBasicInfoDTO> children) {
    this.children = children;
    }
    public Boolean getSelectable() {
    return selectable;
    }
    public void setSelectable(Boolean selectable) {
    this.selectable = selectable;
    }
    public Boolean getDisableCheckbox() {
    return disableCheckbox;
    }
    public void setDisableCheckbox(Boolean disableCheckbox) {
    this.disableCheckbox = disableCheckbox;
    }
    public Boolean getDisabled() {
    return disabled;
    }
    public void setDisabled(Boolean disabled) {
    this.disabled = disabled;
    }
    public static long getSerialversionuid() {
    return serialVersionUID;
    }
    }

service实现类：

/**
    * 查询投资项目资金来源
    */
    @Override
    public List<ProjectBasicInfoDTO> queryFundingSource() {
    //查询投资项目资金来源
    List<DictionaryBean> list = projectBasicInfoMapper.queryDictionaryBean(Constant.FUNDING_SOURCE,Constant.DELFLAG);
    List<ProjectBasicInfoDTO> dataProcessing = ProjectBasicInfoServiceImpl.dataProcessing(list);
    List<ProjectBasicInfoDTO> buildByRecursive = TreeUtils.buildByRecursive(dataProcessing);
    return buildByRecursive;
    }
    /**
    * 将字典数据遍历放入ProjectBasicInfoDTO实体中
    * @param list
    * @return
    */
    public static List<ProjectBasicInfoDTO> dataProcessing(List<DictionaryBean> list){
    List<ProjectBasicInfoDTO> list1 = Lists.newArrayList();
    for (DictionaryBean dictionaryBean : list) {
    ProjectBasicInfoDTO projectBasicInfoDTO = new ProjectBasicInfoDTO();
    projectBasicInfoDTO.setId(dictionaryBean.getItemId());
    projectBasicInfoDTO.setpId(dictionaryBean.getItemParentId());
    projectBasicInfoDTO.setKey(dictionaryBean.getItemId());
    projectBasicInfoDTO.setTitle(dictionaryBean.getItemFullName());
    projectBasicInfoDTO.setValue(dictionaryBean.getItemFullName());
    projectBasicInfoDTO.setSelectable(false);
    projectBasicInfoDTO.setDisableCheckbox(true);
    projectBasicInfoDTO.setDisabled(true);
    list1.add(projectBasicInfoDTO);
    }
    return list1;
    }

递归方法工具类

package cn.ccccltd.ccccoi.mediumterminvestment.util;
    
    import java.util.ArrayList;
    import java.util.List;
    
    import cn.ccccltd.ccccoi.mediumterminvestment.domain.ProjectBasicInfoDTO;
    
    public class TreeUtils {
    /**
    * 使用递归方法建树
    * @param treeNodes
    * @return
    */
    public static List<ProjectBasicInfoDTO> buildByRecursive(List<ProjectBasicInfoDTO> projectBasicInfoDTOs) {
    List<ProjectBasicInfoDTO> trees = new ArrayList<ProjectBasicInfoDTO>();
    for (ProjectBasicInfoDTO projectBasicInfoDTO : projectBasicInfoDTOs) {
    if ("".equals(projectBasicInfoDTO.getpId()) || projectBasicInfoDTO.getpId() == null ) {
    trees.add(findChildren(projectBasicInfoDTO,projectBasicInfoDTOs));
    }
    }
    return trees;
    }
    /**
    * 递归查找子节点
    * @param treeNodes
    * @return
    */
    public static ProjectBasicInfoDTO findChildren(ProjectBasicInfoDTO projectBasicInfoDTO,List<ProjectBasicInfoDTO> projectBasicInfoDTOs) {
    for (ProjectBasicInfoDTO projectBasicInfoDTO2 : projectBasicInfoDTOs) {
    if(projectBasicInfoDTO.getId().equals(projectBasicInfoDTO2.getpId())) {
    if(projectBasicInfoDTO.getChildren() == null) {
    projectBasicInfoDTO.setChildren(new ArrayList<ProjectBasicInfoDTO>());
    }
    //是否还有子节点，如果有的话继续往下遍历，如果没有则直接返回
    projectBasicInfoDTO.getChildren().add(findChildren(projectBasicInfoDTO2,projectBasicInfoDTOs));
    }
    }
    return projectBasicInfoDTO;
    }
    }

树形结构效果图：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzMwNjMxMDYz_size_16_color_FFFFFF_t_70][]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzMwNjMxMDYz_size_16_color_FFFFFF_t_70]: /images/20221022/a17a319767c24937a99cfb38dd210468.png