在java项目中,经常会遇到将从数据库中查询的list结构数据封装成Tree结构的数据。比如行政区划、部门这类有上下级关系的数据经常需要封装成Tree。下面就是在实际项目中遇到的需要将从数据库中获取的行政区划数据封装成tree结构的示例。
// 将list封装成tree
public static List<XzqhNodeDto> treeBuild(List<XzqhNodeDto> xzqhNodeDtos){
List<XzqhNodeDto> tree = new ArrayList<XzqhNodeDto>();
for(XzqhNodeDto xzqhNode : xzqhNodeDtos) {
// 判断是否为根节点
if("1".equals(xzqhNode.getJb())) {
tree.add(xzqhNode);
}
for(XzqhNodeDto item : xzqhNodeDtos) {
if(StringUtils.equals(item.getParentCode(), xzqhNode.getValue())) {
if(xzqhNode.getChildren() == null) {
// 当前节点的子节点为null,则新增一个用于存放子节点的list
xzqhNode.setChildren(new ArrayList<XzqhNodeDto>());
}
xzqhNode.getChildren().add(item);
}
}
}
return tree;
}
// 行政区划数据结构
public class XzqhNodeDto {
// 行政区划编码
private String value;
// 行政区划名称
private String label;
// 所在层级
private String jb;
// 上级编码
private String parentCode;
}