java id pid,JAVA8 两行代码实现ID-PID转树状结构
核心逻辑
通过stream对集合进行分组
准备知识：java8的收集器groupingBy 可以根据项目的一个属性的值对流中的项目分组，并将属性值作为结果Map的键，利用这个特性，可以按列表对象中的Pid属性进行分组。
Stream收集器groupingBy示意图：
StreamGroupingBy示意图
第二步，遍历原集合，从Map中取出对应的List存进去。
测试用Entity
@Data
public class TestEntity {
/** 必备的结构属性 **/
//id
private int id;
//父id
private int pid;
//子集合
private List children;
/** 私有属性，与业务相关的属性 **/
//名称
private String name;
}
代码
单一顶层节点的实现方式
/**
* 将id-pid转为tree结构，单一顶层节点，顶层节点的id和pid不能相同
* @param pidList id-pid对象的列表
* @return tree结构对象
*/
public TestEntity buildTree(List pidList){
//以pid为Key进行分组存入Map
Map> pidListMap =
pidList.stream().collect(Collectors.groupingBy(TestEntity::getPid));
pidList.stream().forEach(item->item.setChildren(pidListMap.get(item.getId())));
//取出顶层节点的对象，本例顶层节点的"PID"为0,注意是PID
return pidListMap.get(0).get(0);
}
多个顶层节点的实现方式：原来的实现稍加修改，只在需要修改的地方加了注释，便于查看区别
//返回值改为List
public List buildTree(List pidList){
Map> pidListMap =
pidList.stream().collect(Collectors.groupingBy(TestEntity::getPid));
pidList.stream().forEach(item->item.setChildren(pidListMap.get(item.getId())));
//返回结果也改为返回顶层节点的list
return pidListMap.get(0);
}
小结
Id-pid转树是一个很常见的需求，以前的实现方式都是基于递归，总是要写老长，还得单独提供一个递归方法，转一个数据结构，要写那么长，总感觉“长不配位”。现在java8逐渐在项目中普及起来了，使用stream可以很方便很清晰的来实现Id-pid转树的需求。
附录：测试代码
public class TestMain {
public static void main(String[] args) {
/** 构造数据 **/
final List testEntities = new ArrayList();
TestEntity topEntity = new TestEntity();
topEntity.setPid(0);
topEntity.setId(1);
topEntity.setName("我是顶层");
testEntities.add(topEntity);
IntStream.range(2,10).forEach(i->{
TestEntity testEntity = new TestEntity();
testEntity.setId(i);
testEntity.setPid(1);
testEntity.setName("name_" + i);
testEntities.add(testEntity);
IntStream.range(i * 10,(i * 10 + 5)).forEach(subI->{
TestEntity subEntity = new TestEntity();
subEntity.setId(subI);
subEntity.setPid(i);
subEntity.setName("name_" + subI);
testEntities.add(subEntity);
IntStream.range(subI * 10,(subI * 10 + 3)).forEach(subII->{
TestEntity sub2Entity = new TestEntity();
sub2Entity.setId(subII);
sub2Entity.setPid(subI);
sub2Entity.setName("name_" + subII);
testEntities.add(sub2Entity);
});
});
});
testEntities.stream().forEach(item->{
System.out.println(JSON.toJSONString(item));
});
System.out.println("========================================");
/** 测试 **/
Pid2TreeTest pid2TreeTest = new Pid2TreeTest();
TestEntity testEntitiy = pid2TreeTest.buildTree(testEntities);
System.out.println(JSON.toJSONString(testEntitiy));
}
}