feat(标准结构树): 拆分特殊过滤

2024-08-01 15:07:28 +08:00
parent 79818d99d3
commit eeb857e9b6
1 changed files with 8 additions and 0 deletions
@@ -87,6 +87,14 @@ public class LawsTreeNodeServiceImpl extends ServiceImpl<LawsTreeNodeMapper, Law
            List<LawsTreeNodeModel> children = lawsTreeNodeModels.get(0).getChildren();
            List<LawsTreeNodeModel> collect = children.stream()
                    .filter(node -> "A02A01".equals(node.getNodeCode()) || "A02A03".equals(node.getNodeCode()))
+                    .peek(node -> {
+                        if ("A02A01".equals(node.getNodeCode())){
+                            List<LawsTreeNodeModel> children1 = node.getChildren();
+                            children1.stream()
+                                    .peek(v -> v.setChildren(null))
+                                    .collect(Collectors.toList());
+                        }
+                    })
                    .collect(Collectors.toList());
            lawsTreeNodeModels.get(0).setChildren(collect);
        }