使用Java8将列表型数据转为树形数据

原创已于 2022-10-26 16:12:46 修改 · 653 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java

于 2022-10-26 16:11:29 首次发布

Java基础专栏收录该内容

8 篇文章

订阅专栏

博客展示了Java相关代码，但未给出具体代码内容。

代码如下：

import cn.hutool.core.collection.CollectionUtil;
import com.alibaba.fastjson.JSONObject;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import lombok.Data;
import lombok.experimental.Accessors;
import org.junit.Test;

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

public class Demo {

	@Test
	public void test01() {
	    List<Node> nodes = Lists.newArrayList();
	    Node node = new Node();
	    node.setId(1).setName("root").setPid(0);
	    nodes.add(node);
	    node = new Node();
	    node.setId(2).setName("level1-1").setPid(1);
	    nodes.add(node);
	    node = new Node();
	    node.setId(3).setName("level1-2").setPid(1);
	    nodes.add(node);
	
	    node = new Node();
	    node.setId(4).setName("level2-1").setPid(2);
	    nodes.add(node);
	
	    node = new Node();
	    node.setId(5).setName("level2-1").setPid(2);
	    nodes.add(node);
	
	    Map<Integer, List<Node>> nodeMap = nodes.stream().collect(Collectors.groupingBy(Node::getPid));
	
	    Map<Integer, Node> nodeMap2 = nodes.stream().collect(Collectors.toMap(Node::getId, x -> x));
	    for (Integer pid : nodeMap.keySet()) {
	        List<Node> nodeList = nodeMap.get(pid);
	        Node n = nodeMap2.get(pid);
	        if (n != null && CollectionUtil.isNotEmpty(nodeList)) {
	            n.setChildren(nodeList);
	        }
	    }
	
	    List<Node> result =  nodes.stream().filter(x -> x.getPid() == 0).collect(Collectors.toList());
	    System.out.println(JSONObject.toJSONString(result));
	
	}
	
	@Data
	@Accessors(chain = true)
	public static class Node {
	    private int id;
	    private String name;
	    private int pid;
	    private List<Node> children;
	}
}