将List对象列表转为树形结构

1、SocialTag.java

1. package com.nijia.vip.social.entity;
3. import java.util.ArrayList;
4. import java.util.List;
6. import com.baomidou.mybatisplus.annotation.IdType;
7. import com.baomidou.mybatisplus.annotation.TableField;
8. import com.baomidou.mybatisplus.annotation.TableId;
9. import com.nijia.base.entity.BaseEntity;
11. import io.swagger.annotations.ApiModelProperty;
12. import lombok.Data;
14. /**
15. * 社交--个人标签(SocialTag)表实体类
16. *
17. * @author whp
18. * @since 2022-06-23 14:15:53
19. */
20. @SuppressWarnings("serial")
21. @Data
22. public class SocialTag extends BaseEntity {
23.     private static final long serialVersionUID = 1L;
25.     @TableId(value = "id", type = IdType.ASSIGN_ID)
26.     /**
27.      * 主键
28.      */
29.     @ApiModelProperty("主键")
30.     private Long id;
32.     /**
33.      * 标签名称
34.      */
35.     @ApiModelProperty("标签名称")
36.     private String tagName;
38.     /**
39.      * 父级id
40.      */
41.     @ApiModelProperty("父级id")
42.     private Long parentId;
44.     /**
45.      * 标签等级【1-1级，2-2级】
46.      */
47.     @ApiModelProperty("标签等级【1-1级，2-2级】")
48.     private Integer levelNo;
50.     /**
51.      * 排序号
52.      */
53.     @ApiModelProperty("排序号")
54.     private Integer sortNo;
56.     /**
57.      * 可用状态【1-可用，0-不可用】
58.      */
59.     @ApiModelProperty("可用状态【1-可用，0-不可用】")
60.     private Integer enableFlag;
62.     /**
63.      * 删除标识【1-删除，0-正常】
64.      */
65.     @ApiModelProperty("删除标识【1-删除，0-正常】")
66.     private Integer deleteFlag;
68.     @TableField(exist=false)
69.     private List<SocialTag> childSocialTagList = new ArrayList<SocialTag>();
72. }

复制代码

 
2、SocialTagListToTreeTest.java

1. package com.nijia.vip.service;
3. import java.util.ArrayList;
4. import java.util.Date;
5. import java.util.List;
6. import java.util.Map;
7. import java.util.stream.Collectors;
9. import org.apache.commons.collections4.CollectionUtils;
11. import com.nijia.test.BaseTest;
12. import com.nijia.vip.social.entity.SocialTag;
14. import io.geekidea.springbootplus.framework.util.JavaJsonConvert;
17. public class SocialTagListToTreeTest extends BaseTest {
18.    
19.      public static void main(String[] args) {
20.          List<SocialTag> resultList = convertListToTree(initData());
21.          System.out.println(JavaJsonConvert.javaToJson(resultList));
22.     }
24.    
25.      /**
26.       * 将List对象列表转为树形结构
27.       *
28.       * @param socialTagList
29.       * @return
30.       */
31.     private static List<SocialTag> convertListToTree(List<SocialTag> socialTagList) {
32.         List<SocialTag> resultList = new ArrayList<SocialTag>();
33.         if(CollectionUtils.isEmpty(socialTagList)) {
34.             return resultList;
35.         }
36.         
37.         //List 转为 Map
38.          Map<Long, SocialTag> socialTagMap = socialTagList.stream().collect(Collectors.toMap(SocialTag::getId, item -> item));
39.          for(SocialTag socialTag : socialTagList) {
40.              SocialTag parentSocialTag = socialTagMap.get(socialTag.getParentId());
41.              //该节点为空，说明是顶级节点
42.              if(parentSocialTag == null) {
43.                  // 将顶级节点加入结果集中
44.                  resultList.add(socialTag);
45.                  continue;
46.              }
47.             
48.              // 把自己加到父节点对象里面去
49.              parentSocialTag.getChildSocialTagList().add(socialTag);
50.          }
51.          
52.         return resultList;
53.     }
55.     private static List<SocialTag> initData() {
56.         List<SocialTag> socialTagList = new ArrayList<SocialTag>();
57.         SocialTag socialTag = new SocialTag();
58.         socialTag.setId(1L);
59.         socialTag.setTagName("Java");
60.         socialTag.setParentId(0L);
61.         //标签等级【1-1级，2-2级】
62.         socialTag.setLevelNo(1);
63.         socialTag.setCreateTime(new Date());
64.         socialTagList.add(socialTag);
65.         
66.         socialTag = new SocialTag();
67.         socialTag.setId(10L);
68.         socialTag.setTagName("Springboot");
69.         socialTag.setParentId(1L);
70.         //标签等级【1-1级，2-2级】
71.         socialTag.setLevelNo(2);
72.         socialTag.setCreateTime(new Date());
73.         socialTagList.add(socialTag);
74.         
75.         socialTag = new SocialTag();
76.         socialTag.setId(11L);
77.         socialTag.setTagName("SpringCloud");
78.         socialTag.setParentId(1L);
79.         //标签等级【1-1级，2-2级】
80.         socialTag.setLevelNo(2);
81.         socialTag.setCreateTime(new Date());
82.         socialTagList.add(socialTag);
83.         
84.         socialTag = new SocialTag();
85.         socialTag.setId(2L);
86.         socialTag.setTagName("DB");
87.         socialTag.setParentId(0L);
88.         //标签等级【1-1级，2-2级】
89.         socialTag.setLevelNo(1);
90.         socialTag.setCreateTime(new Date());
91.         socialTagList.add(socialTag);
92.         
93.         socialTag = new SocialTag();
94.         socialTag.setId(20L);
95.         socialTag.setTagName("Mysql");
96.         socialTag.setParentId(2L);
97.         //标签等级【1-1级，2-2级】
98.         socialTag.setLevelNo(2);
99.         socialTag.setCreateTime(new Date());
100.         socialTagList.add(socialTag);
101.         
102.         socialTag = new SocialTag();
103.         socialTag.setId(21L);
104.         socialTag.setTagName("Oracle");
105.         socialTag.setParentId(2L);
106.         //标签等级【1-1级，2-2级】
107.         socialTag.setLevelNo(2);
108.         socialTag.setCreateTime(new Date());
109.         socialTagList.add(socialTag);
110.         
111.         socialTag = new SocialTag();
112.         socialTag.setId(3L);
113.         socialTag.setTagName("JS");
114.         socialTag.setParentId(0L);
115.         //标签等级【1-1级，2-2级】
116.         socialTag.setLevelNo(1);
117.         socialTag.setCreateTime(new Date());
118.         socialTagList.add(socialTag);
119.         
120.         socialTag = new SocialTag();
121.         socialTag.setId(30L);
122.         socialTag.setTagName("VUE");
123.         socialTag.setParentId(3L);
124.         //标签等级【1-1级，2-2级】
125.         socialTag.setLevelNo(2);
126.         socialTag.setCreateTime(new Date());
127.         socialTagList.add(socialTag);
128.         
129.         return socialTagList;
130.     }
132.    
133. }

复制代码

 

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！