手动实现一个树结构(根据参数ID返回当前节点以及所有子节点,递归查询下一级节点)

package com.study.utils;

import com.study.entity.Per;

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

public class TreeUtils {

    public static Per getCurrentNode(List<Per> list){
        Per per = list.stream().min(Comparator.comparing(Per::getId)).get();

        return setChildrenNode(per, list);
    }
    public static Per setChildrenNode(Per per,List<Per> list){
        List<Per> collect = list.stream().filter(item -> {
            return per.getId().equals(item.getPid());
        }).collect(Collectors.toList());

        per.setPerList(collect);

        if(collect.size()!=0){
            collect.stream().forEach(item -> {setChildrenNode(item,list);});
        }

        return per;
    }
}



package com.study.entity;

import lombok.Data;

import javax.persistence.*;
import java.util.List;

@Data
@Entity
@Table(name = "per")
public class Per {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Integer id;
    private Integer pid;
    @Transient
    private List<Per> perList;
}



package com.study;

import com.alibaba.fastjson.JSON;
import com.study.dao.PerRepository;
import com.study.entity.Per;
import com.study.utils.TreeUtils;
import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;

@SpringBootTest
class JpaApplicationTests {

	@Resource
	private PerRepository perRepository;

	@Test
	void test(){
		List<Per> collect = new ArrayList<>();
		List<Per> list = getList(collect,2);
		Per currentNode = TreeUtils.getCurrentNode(list);
		JSON o = (JSON)JSON.toJSON(currentNode);
		System.out.println(o);

	}

	public List<Per> getList(List<Per> list,Integer id){
		Per per = perRepository.findById(id).get();
		if(per!=null){
			list.add(per);
		}
		List<Per> children = perRepository.findByPid(per.getId());
		children.stream().forEach(item ->{getList(list,item.getId());});
		return list;
	}
}

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
要查询一个节点的所有父级节点返回树形结构数据,可以使用递归的方式实现。具体的实现方式如下: 首先定义一个节点类: ``` public class Node { private String id; private String name; private String parentId; private List<Node> children; // 省略 getter 和 setter 方法 } ``` 然后定义一个方法来查询节点的所有父级节点: ``` public List<Node> queryParentNodes(String nodeId, List<Node> allNodes) { List<Node> result = new ArrayList<>(); Node node = null; // 遍历所有节点,找到指定的节点 for (Node n : allNodes) { if (n.getId().equals(nodeId)) { node = n; break; } } if (node != null) { // 如果找到了指定节点,则递归查询其所有父节点 String parentId = node.getParentId(); if (parentId != null) { List<Node> parentNodes = queryParentNodes(parentId, allNodes); node.setChildren(parentNodes); } result.add(node); } return result; } ``` 这个方法接受两个参数一个是要查询的节点ID,另一个是所有节点的列表。它会遍历所有节点,找到指定的节点,然后递归查询其所有父节点,并将其构造成树形结构返回。如果找不到指定的节点,则返回空列表。 使用示例: ``` List<Node> allNodes = new ArrayList<>(); // 添加所有节点到列表中,这里省略代码 List<Node> parentNodes = queryParentNodes("nodeId", allNodes); // 处理 parentNodes,将其转换成前端需要的数据格式,这里省略代码 ``` 这里的 "nodeId" 是要查询的节点ID,需要替换成实际的值。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值