当前位置：首页 > news >正文

数形结构转换工具类

news 2026/5/31 18:10:40

项目中使用了很多树状结构，为了方便使用开发一个通用的工具类。
使用工具类的时候写一个类基础BaseNode，如果有个性化字段添加到类里面，然后就可以套用工具类。
工具类会将id和pid做关联返回一个树状结构的集合。使用了hutool的工具包判空。
依赖
<dependency> <groupId>cn.hutool</groupId> <artifactId>hutool-all</artifactId> <version>5.8.25</version> </dependency>
工具类TreeUtil.java
`import cn.hutool.core.util.ObjUtil;
import com.iccb.BaseNode;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class TreeUtil {

public static <T extends BaseNode<T>> List<T> toTree(List<T> treeNodeList) {//数据按照pid分组Map<Long, List<T>> map = treeNodeList.stream().collect(Collectors.groupingBy(T :: getPid,LinkedHashMap::new,Collectors.toList()));//找出所有的根节点，pid=-1的为根节点List<T> areaList = map.get(-1L);if(ObjUtil.isEmpty(areaList)){return new ArrayList<>();}for (T areaVO : areaList) {forEach(map, areaVO);}return areaList;
}
private static  <T extends BaseNode<T>> void forEach(Map<Long, List<T>> collect, T areaVO) {List<T> nodeList = collect.get(areaVO.getId());if (collect.get(areaVO.getId()) == null) {return;}areaVO.setChildren(nodeList);for (T node : nodeList) {forEach(collect, node);}
}

BaseNode.java
`import com.fasterxml.jackson.annotation.JsonFormat;
import lombok.Data;

import java.io.Serializable;
import java.util.List;

/**

@author admin
*/
@Data
public class BaseNode implements Serializable {

private static final long serialVersionUID = 1L;

@JsonFormat(shape = JsonFormat.Shape.STRING)
private Long id;

@JsonFormat(shape = JsonFormat.Shape.STRING)
private Long pid;

private String value;

private List children;
}`

使用范例
返回树状结构的地区数据

1）首先构建数据VO
`@EqualsAndHashCode(callSuper = true)
@Data
public class AreaVO extends BaseNode {

private static final long serialVersionUID = 1L;private String areaCode;private String label;private String level;private String lnglat;

2）查询地区数据
List<Area> list = areaService.list();

构建数据VO 或者直接使用mapper查询用VO封装
List<AreaVO> collect = list.stream().map(i -> { AreaVO areaVO = BeanUtil.copyProperties(i, AreaVO.class); areaVO.setLabel(i.getAreaName()); areaVO.setValue(String.valueOf(i.getId())); areaVO.setLevel(i.getAreaLevel()); areaVO.setPid(i.getParentId()); return areaVO; }).collect(Collectors.toList());
使用工具类构造数据结构
List<AreaVO> areaTree = TreeUtil.toTree(collect);

数据返回示例
{ "code": 200, "msg": null, "data": [ { "id": "4058069977868140544", "value": "4058069977868140544", "pid": "4058069977868148888", "children": [ { "id": "4058069978245627904", "value": "4058069978245627904", "pid": "4058069977868140544", "children": [ { "id": "4058069978530840576", "value": "4058069978530840576", "pid": "4058069978245627904", "children": null, "areaCode": "110101000000", "label": "东城区", "level": "3", "lnglat": null }, { "id": "4058069978732167168", "value": "4058069978732167168", "pid": "4058069978245627904", "children": null, "areaCode": "110102000000", "label": "西城区", "level": "3", "lnglat": null }, { "id": "4058069978874773504", "value": "4058069978874773504", "pid": "4058069978245627904", "children": null, "areaCode": "110105000000", "label": "朝阳区", "level": "3", "lnglat": null }, { "id": "4058069978979631104", "value": "4058069978979631104", "pid": "4058069978245627904", "children": null, "areaCode": "110106000000", "label": "丰台区", "level": "3", "lnglat": null }, { "id": "4058069979046739968", "value": "4058069979046739968", "pid": "4058069978245627904", "children": null, "areaCode": "110107000000", "label": "石景山区", "level": "3", "lnglat": null }, { "id": "4058069979118043136", "value": "4058069979118043136", "pid": "4058069978245627904", "children": null, "areaCode": "110108000000", "label": "海淀区", "level": "3", "lnglat": null }, { "id": "4058069979231289344", "value": "4058069979231289344", "pid": "4058069978245627904", "children": null, "areaCode": "110109000000", "label": "门头沟区", "level": "3", "lnglat": null }, { "id": "4058069979336146944", "value": "4058069979336146944", "pid": "4058069978245627904", "children": null, "areaCode": "110111000000", "label": "房山区", "level": "3", "lnglat": null }, { "id": "4058069979441004544", "value": "4058069979441004544", "pid": "4058069978245627904", "children": null, "areaCode": "110112000000", "label": "通州区", "level": "3", "lnglat": null }, { "id": "4058069979550056448", "value": "4058069979550056448", "pid": "4058069978245627904", "children": null, "areaCode": "110113000000", "label": "顺义区", "level": "3", "lnglat": null }, { "id": "4058069979650719744", "value": "4058069979650719744", "pid": "4058069978245627904", "children": null, "areaCode": "110114000000", "label": "昌平区", "level": "3", "lnglat": null }, { "id": "4058069979747188736", "value": "4058069979747188736", "pid": "4058069978245627904", "children": null, "areaCode": "110115000000", "label": "大兴区", "level": "3", "lnglat": null }, { "id": "4058069979852046336", "value": "4058069979852046336", "pid": "4058069978245627904", "children": null, "areaCode": "110116000000", "label": "怀柔区", "level": "3", "lnglat": null }, { "id": "4058069979956903936", "value": "4058069979956903936", "pid": "4058069978245627904", "children": null, "areaCode": "110117000000", "label": "平谷区", "level": "3", "lnglat": null }, { "id": "4058069980166619136", "value": "4058069980166619136", "pid": "4058069978245627904", "children": null, "areaCode": "110228000000", "label": "密云区", "level": "3", "lnglat": null }, { "id": "4058069980284059648", "value": "4058069980284059648", "pid": "4058069978245627904", "children": null, "areaCode": "110229000000", "label": "延庆区", "level": "3", "lnglat": null } ], "areaCode": "110100000000", "label": "北京市", "level": "2", "lnglat": null } ], "areaCode": "110000000000", "label": "北京市", "level": "1", "lnglat": null }, { "id": "4058069980422471680", "value": "4058069980422471680", "pid": "4058069977868148888", "children": [ { "id": "4058069980527329280", "value": "4058069980527329280", "pid": "4058069980422471680", "children": [ { "id": "4058069980627992576", "value": "4058069980627992576", "pid": "4058069980527329280", "children": null, "areaCode": "120101000000", "label": "和平区", "level": "3", "lnglat": null }, { "id": "4058069980732850176", "value": "4058069980732850176", "pid": "4058069980527329280", "children": null, "areaCode": "120102000000", "label": "河东区", "level": "3", "lnglat": null }, { "id": "4058069980841902080", "value": "4058069980841902080", "pid": "4058069980527329280", "children": null, "areaCode": "120103000000", "label": "河西区", "level": "3", "lnglat": null }, { "id": "4058069980938371072", "value": "4058069980938371072", "pid": "4058069980527329280", "children": null, "areaCode": "120104000000", "label": "南开区", "level": "3", "lnglat": null }, { "id": "4058069981043228672", "value": "4058069981043228672", "pid": "4058069980527329280", "children": null, "areaCode": "120105000000", "label": "河北区", "level": "3", "lnglat": null }, { "id": "4058069981148086272", "value": "4058069981148086272", "pid": "4058069980527329280", "children": null, "areaCode": "120106000000", "label": "红桥区", "level": "3", "lnglat": null }, { "id": "4058069981252943872", "value": "4058069981252943872", "pid": "4058069980527329280", "children": null, "areaCode": "120110000000", "label": "东丽区", "level": "3", "lnglat": null }, { "id": "4058069981357801472", "value": "4058069981357801472", "pid": "4058069980527329280", "children": null, "areaCode": "120111000000", "label": "西青区", "level": "3", "lnglat": null }, { "id": "4058069981475241984", "value": "4058069981475241984", "pid": "4058069980527329280", "children": null, "areaCode": "120112000000", "label": "津南区", "level": "3", "lnglat": null }, { "id": "4058069981580099584", "value": "4058069981580099584", "pid": "4058069980527329280", "children": null, "areaCode": "120113000000", "label": "北辰区", "level": "3", "lnglat": null }, { "id": "4058069981684957184", "value": "4058069981684957184", "pid": "4058069980527329280", "children": null, "areaCode": "120114000000", "label": "武清区", "level": "3", "lnglat": null }, { "id": "4058069981789814784", "value": "4058069981789814784", "pid": "4058069980527329280", "children": null, "areaCode": "120115000000", "label": "宝坻区", "level": "3", "lnglat": null }, { "id": "4058069981890478080", "value": "4058069981890478080", "pid": "4058069980527329280", "children": null, "areaCode": "120116000000", "label": "滨海新区", "level": "3", "lnglat": null }, { "id": "4058069981986947072", "value": "4058069981986947072", "pid": "4058069980527329280", "children": null, "areaCode": "120117000000", "label": "宁河区", "level": "3", "lnglat": null }, { "id": "4058069982087610368", "value": "4058069982087610368", "pid": "4058069980527329280", "children": null, "areaCode": "120118000000", "label": "静海区", "level": "3", "lnglat": null }, { "id": "4058069982293131264", "value": "4058069982293131264", "pid": "4058069980527329280", "children": null, "areaCode": "120225000000", "label": "蓟州区", "level": "3", "lnglat": null } ], "areaCode": "120100000000", "label": "天津市", "level": "2", "lnglat": null } ], "areaCode": "120000000000", "label": "天津市", "level": "1", "lnglat": null }, { "id": "4461100398823866368", "value": "4461100398823866368", "pid": "4058069977868148888", "children": [ { "id": "4461100398974861312", "value": "4461100398974861312", "pid": "4461100398823866368", "children": null, "areaCode": "820100000000", "label": "澳门半岛", "level": "2", "lnglat": null }, { "id": "4461100399113273344", "value": "4461100399113273344", "pid": "4461100398823866368", "children": null, "areaCode": "820200000000", "label": "离岛", "level": "2", "lnglat": null } ], "areaCode": "820000000000", "label": "澳门特别行政区", "level": "1", "lnglat": null } ] }

查看全文

http://www.rkmt.cn/news/54116.html