springboot树形布局 支持含糊查询，返回匹配节点和父节点，其他节点不返回 ...

1. package com.me.meterdemo.ds;
2. import java.util.ArrayList;
3. import java.util.List;
5. public class TreeNode {
6.     private Long id;
7.     private String name;
8.     private Long parentId;
9.     private List<TreeNode> children = new ArrayList<>();
11.     // 构造方法
12.     public TreeNode(Long id, String name, Long parentId) {
13.         this.id = id;
14.         this.name = name;
15.         this.parentId = parentId;
16.     }
18.     // Getters and Setters
19.     public Long getId() { return id; }
20.     public String getName() { return name; }
21.     public Long getParentId() { return parentId; }
22.     public List<TreeNode> getChildren() { return children; }
24.     public void addChild(TreeNode child) {
25.         // 防止重复添加
26.         if (children.stream().noneMatch(c -> c.id.equals(child.id))) {
27.             children.add(child);
28.         }
29.     }
30. }

复制代码

1. package com.me.meterdemo.ds;
3. import org.springframework.stereotype.Service;
5. import javax.annotation.PostConstruct;
6. import java.util.*;
7. import java.util.stream.Collectors;
11. @Service
12. public class TreeMemoryService {
13.     // 内存数据存储
14.     private final Map<Long, TreeNode> nodeMap = new HashMap<>();
16.     // 初始化测试数据
17.     @PostConstruct
18.     private void initData() {
19.         clearAndInitData();
20.     }
22.     private void clearAndInitData() {
23.         nodeMap.clear();
24.         addNode(1L, "集团总部", null);
25.         addNode(2L, "华东公司", 1L);
26.         addNode(3L, "技术部", 2L);
27.         addNode(4L, "开发组", 3L);
28.         addNode(5L, "测试组", 3L);
29.         addNode(6L, "华南公司", 1L);
30.     }
32.     private void addNode(Long id, String name, Long parentId) {
33.         TreeNode node = new TreeNode(id, name, parentId);
34.         nodeMap.put(id, node);
35.     }
37.     /**
38.      * 执行树结构搜索
39.      * @param keyword 查询关键词（可为空）
40.      * @return 符合条件节点及其父节点组成的树结构
41.      */
42.     public List<TreeNode> search(String keyword) {
43.         // 空关键词返回空结果
44.         if (keyword != null && keyword.trim().isEmpty()) {
45.             return Collections.emptyList();
46.         }
48.         // 获取所有匹配节点
49.         List<TreeNode> matches = nodeMap.values().stream()
50.                 .filter(node -> matchesKeyword(node, keyword))
51.                 .collect(Collectors.toList());
53.         // 收集需要展示的节点ID（匹配节点+父链）
54.         Set<Long> visibleIds = new LinkedHashSet<>();
55.         matches.forEach(node -> collectParentChain(node, visibleIds));
57.         // 构建精简树结构
58.         return buildPrunedTree(visibleIds);
59.     }
61.     // 判断节点是否匹配关键词
62.     private boolean matchesKeyword(TreeNode node, String keyword) {
63.         return keyword == null ||
64.                 node.getName().contains(keyword);
65.     }
67.     // 收集节点及其所有父节点
68.     private void collectParentChain(TreeNode node, Set<Long> ids) {
69.         Long currentId = node.getId();
70.         while (currentId != null) {
71.             if (!ids.add(currentId)) break; // 防止循环
72.             TreeNode current = nodeMap.get(currentId);
73.             currentId = current.getParentId();
74.         }
75.     }
77.     // 构建修剪后的树结构
78.     private List<TreeNode> buildPrunedTree(Set<Long> visibleIds) {
79.         // 创建节点副本防止污染原始数据
80.         Map<Long, TreeNode> visibleNodes = createNodeCopies(visibleIds);
82.         // 构建层级关系
83.         List<TreeNode> roots = new ArrayList<>();
84.         visibleNodes.values().forEach(node -> {
85.             Long parentId = node.getParentId();
86.             if (parentId == null) {
87.                 roots.add(node);
88.             } else {
89.                 TreeNode parent = visibleNodes.get(parentId);
90.                 if (parent != null) {
91.                     parent.addChild(node);
92.                 }
93.             }
94.         });
96.         return roots;
97.     }
99.     // 创建节点副本（清空原有子节点）
100.     private Map<Long, TreeNode> createNodeCopies(Set<Long> visibleIds) {
101.         return visibleIds.stream()
102.                 .map(id -> nodeMap.get(id))
103.                 .filter(Objects::nonNull)
104.                 .collect(Collectors.toMap(
105.                         TreeNode::getId,
106.                         orig -> new TreeNode(orig.getId(),
107.                                 orig.getName(),
108.                                 orig.getParentId())
109.                 ));
110.     }
112.     // 重置测试数据（可选）
113.     public void resetData() {
114.         clearAndInitData();
115.     }
116. }

复制代码

1. package com.me.meterdemo.ds;
3. import org.springframework.web.bind.annotation.GetMapping;
4. import org.springframework.web.bind.annotation.RequestParam;
5. import org.springframework.web.bind.annotation.RestController;
7. import java.util.List;
9. @RestController
10. public class TreeDSController {
11.     private final TreeMemoryService treeMemoryService;
13.     public TreeDSController(TreeMemoryService treeService) {
14.         this.treeMemoryService = treeService;
15.     }
17.     @GetMapping("/search")
18.     public List<TreeNode> searchTree(@RequestParam(required = false) String keyword) {
19.         return treeMemoryService.search(keyword);
20.     }
21. }

复制代码

springboot树形布局 支持含糊查询，返回匹配节点和父节点，其他节点不返回

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。