java树结构数据_java 数据拼装成树结构

项目数据需要在后台拼装成树结构，所以写了两种方案来实现：

方案一：数据类型List

1>、首先增加树结构数据基础父类BaseTreeVO，然后需要拼装树结构的VO类需要继承来基础父类。

2>、创建并调用数据拼装数工具类

方案二：数据类型List>

调用工具类，传入标记父Id的字段

基础父类实现如下：

package com.xxxx.icop.framework.skeleton.template;

import com.xxxx.icop.framework.skeleton.template.BaseVO;

import java.util.List;

public class BaseTreeVO {

private static final long serialVersionUID = 6753923840108631912L;

private List children;

private Long parentId;

private Long id;

public BaseTreeVO() {

}

public Long getParentId() {

return this.parentId;

}

public void setParentId(Long parentId) {

this.parentId = parentId;

}

public List getChildren() {

return this.children;

}

public void setChildren(List children) {

this.children = children;

}

public String toString() {

return super.toString();

}

public Long getId() {

return this.id;

}

public void setId(Long id) {

this.id = id;

}

}

数据拼装树结构工具类如下：

package com.xxxx.icop.framework.skeleton.template;

import com.xxxx.icop.framework.skeleton.template.BaseTreeVO;

import java.util.ArrayList;

import java.util.Collection;

import java.util.List;

import java.util.Map;

import java.util.function.Consumer;

import java.util.function.Function;

import java.util.stream.Collectors;

import org.apache.commons.collections.CollectionUtils;

public class BaseTreeHelper {

private static final String KEY_CHILDREN = "children";

public BaseTreeHelper() {

}

public static List> createTree(List> data, String key) {

if(CollectionUtils.isEmpty(data)) {

return new ArrayList();

} else {

Map groupByParentIdMap = (Map)data.stream().collect(Collectors.groupingBy((item) -> {

return item.get(key) != null?item.get(key).toString():"nonParent";

}));

Map dataMap = (Map)data.stream().collect(Collectors.toMap((item) -> {

return item.get("id").toString();

}, (t2) -> {

return t2;

}));

ArrayList resp = new ArrayList();

groupByParentIdMap.keySet().forEach((parentId) -> {

if(dataMap.containsKey(parentId)) {

Object child = (List)((Map)dataMap.get(parentId)).get("children");

if(CollectionUtils.isEmpty((Collection)child)) {

child = new ArrayList();

((Map)dataMap.get(parentId)).put("children", child);

}

((List)child).addAll((Collection)groupByParentIdMap.get(parentId));

} else {

resp.addAll((Collection)groupByParentIdMap.get(parentId));

}

});

return resp;

}

}

public static List createTree(List data) {

if(CollectionUtils.isEmpty(data)) {

return new ArrayList();

} else {

Map groupByParentIdMap = (Map)data.stream().collect(Collectors.groupingBy((item) -> {

return item.getParentId() != null?item.getParentId().toString():"nonParent";

}));

Map dataMap = (Map)data.stream().collect(Collectors.toMap((item) -> {

return item.getId().toString();

}, (t2) -> {

return t2;

}));

ArrayList resp = new ArrayList();

groupByParentIdMap.keySet().forEach((parentId) -> {

if(dataMap.containsKey(parentId)) {

Object child = ((BaseTreeVO)dataMap.get(parentId)).getChildren();

if(CollectionUtils.isEmpty((Collection)child)) {

child = new ArrayList();

}

((List)child).addAll((Collection)groupByParentIdMap.get(parentId));

((BaseTreeVO)dataMap.get(parentId)).setChildren((List)child);

} else {

resp.addAll((Collection)groupByParentIdMap.get(parentId));

}

});

return resp;

}

}

}