feat: 涉及系统部件懒加载搜索功能

2024-08-10 17:34:39 +08:00
parent 080f8c076c
commit 800e9ae500
1 changed files with 95 additions and 42 deletions
@@ -184,61 +184,114 @@ public class PartNameServiceImpl extends ServiceImpl<PartNameMapper, PartName>

    @Override
    public List<PartName> queryList(String nodeName) {
+        // 查询所有节点
        LambdaQueryWrapper<PartName> queryWrapper = new LambdaQueryWrapper<>();
-        queryWrapper.eq(PartName::getParentCode, "Part");
-        if (nodeName != null && !nodeName.isEmpty()) {
-            queryWrapper.like(PartName::getName, nodeName);
-        }
-        List<PartName> list = list(queryWrapper);
+        List<PartName> allNodes = list(queryWrapper);

-        List<String> codes = list.stream().map(PartName::getCode).collect(Collectors.toList());
-        // 判断有没有子级
-        queryWrapper.clear();
-        queryWrapper.in(PartName::getParentCode, codes);
-        if (nodeName != null && !nodeName.isEmpty()) {
-            queryWrapper.like(PartName::getName, nodeName);
-        }
-        List<PartName> childList = list(queryWrapper);
+        // 通过 nodeName 过滤符合条件的节点
+        List<PartName> matchedNodes = allNodes.stream()
+                .filter(node -> node.getName() != null && node.getName().contains(nodeName))
+                .collect(Collectors.toList());

-        if (childList.isEmpty()) {
-            list.forEach(e -> e.setLeafFlag(true));
-            return list;
+        if (matchedNodes.isEmpty()) {
+            return Collections.emptyList();
        }
-        Map<String, List<PartName>> childMap = childList.stream().collect(Collectors.groupingBy(PartName::getParentCode));
-        list.forEach(item -> {
-            List<PartName> partNames = childMap.get(item.getCode());
-            item.setLeafFlag(partNames == null || partNames.isEmpty());
-        });
-        return list;
+
+        // 用于存储一级节点
+        Set<PartName> rootNodes = new HashSet<>();
+
+        // 查找每个匹配节点的一级父节点
+        for (PartName matchedNode : matchedNodes) {
+            PartName currentNode = matchedNode;
+            while (currentNode != null && !currentNode.getParentCode().equals("Part")) {
+                // 查找当前节点的父节点
+                String parentCode = currentNode.getParentCode();
+                currentNode = allNodes.stream()
+                        .filter(node -> node.getCode().equals(parentCode))
+                        .findFirst()
+                        .orElse(null);
+            }
+            // 添加一级节点到结果集
+            if (currentNode != null) {
+                rootNodes.add(currentNode);
+            }
+        }
+
+        // 设置叶子标志的逻辑调整
+        for (PartName node : rootNodes) {
+            Set<String> descendantCodes = getAllDescendantCodes(node.getCode(), allNodes);  // 获取所有子孙节点的ID
+            boolean isLeaf = true;  // 默认设置为叶子节点
+
+            for (PartName matchedNode : matchedNodes) {
+                if (descendantCodes.contains(matchedNode.getCode())) {
+                    isLeaf = false;
+                    break;  // 如果在子孙节点中找到匹配节点，当前根节点不是叶子节点
+                }
+            }
+            node.setLeafFlag(isLeaf);  // 设置叶子标志
+        }
+
+        return new ArrayList<>(rootNodes);
+    }
+
+    // 方法：获取给定节点ID的所有子孙节点ID
+    private Set<String> getAllDescendantCodes(String parentCode, List<PartName> allNodes) {
+        Set<String> descendantCodes = new HashSet<>();
+        List<PartName> directChildren = allNodes.stream()
+                .filter(node -> node.getParentCode().equals(parentCode))
+                .collect(Collectors.toList());
+
+        for (PartName child : directChildren) {
+            descendantCodes.add(child.getCode());
+            descendantCodes.addAll(getAllDescendantCodes(child.getCode(), allNodes)); // 递归获取子孙节点
+        }
+
+        return descendantCodes;
    }

    @Override
    public List<PartName> getChild(String parentCode, String nodeName) {
+        // 查询所有节点
        LambdaQueryWrapper<PartName> queryWrapper = new LambdaQueryWrapper<>();
-        queryWrapper.eq(PartName::getParentCode, parentCode);
-        if (nodeName != null && !nodeName.isEmpty()) {
-            queryWrapper.like(PartName::getName, nodeName);
-        }
-        List<PartName> list = list(queryWrapper);
+        List<PartName> allNodes = list(queryWrapper);

-        List<String> codes = list.stream().map(PartName::getCode).collect(Collectors.toList());
-        // 判断有没有子级
-        queryWrapper.clear();
-        queryWrapper.in(PartName::getParentCode, codes);
-        if (nodeName != null && !nodeName.isEmpty()) {
-            queryWrapper.like(PartName::getName, nodeName);
+        // 通过 nodeName 过滤符合条件的节点
+        List<PartName> matchedNodes = allNodes.stream()
+                .filter(node -> node.getName() != null && node.getName().contains(nodeName))
+                .collect(Collectors.toList());
+
+        if (matchedNodes.isEmpty()) {
+            return Collections.emptyList();
        }
-        List<PartName> childList = list(queryWrapper);
-        if (childList.isEmpty()) {
-            list.forEach(e -> e.setLeafFlag(true));
-            return list;
+
+        // 用于存储一级节点
+        Set<PartName> rootNodes = new HashSet<>();
+
+        // 查找每个匹配节点的一级父节点
+        for (PartName matchedNode : matchedNodes) {
+            PartName currentNode = matchedNode;
+            while (currentNode != null && !currentNode.getParentCode().equals(parentCode)) {
+                // 查找当前节点的父节点
+                String pCode = currentNode.getParentCode();
+                currentNode = allNodes.stream()
+                        .filter(node -> node.getCode().equals(pCode))
+                        .findFirst()
+                        .orElse(null);
+            }
+            // 添加一级节点到结果集
+            if (currentNode != null) {
+                rootNodes.add(currentNode);
+            }
        }
-        Map<String, List<PartName>> childMap = childList.stream().collect(Collectors.groupingBy(PartName::getParentCode));
-        list.forEach(item -> {
-            List<PartName> partNames = childMap.get(item.getCode());
-            item.setLeafFlag(partNames == null || partNames.isEmpty());
+
+        // 设置叶子标志
+        Set<String> rootNodeCodes = rootNodes.stream().map(PartName::getCode).collect(Collectors.toSet());
+        rootNodes.forEach(node -> {
+            node.setLeafFlag(matchedNodes.stream()
+                    .noneMatch(n -> rootNodeCodes.contains(n.getParentCode())));
        });
-        return list;
+
+        return new ArrayList<>(rootNodes);
    }

    private PartName getFirstLevelNode(PartName node, Map<String, PartName> nodeMap) {