feat: 体系树下自带我的收藏

2023-09-12 17:44:45 +08:00
parent 33fb6a9d28
commit e78dc0585b
1 changed files with 8 additions and 2 deletions
@@ -63,9 +63,15 @@ public class LawsTreeNodeServiceImpl extends ServiceImpl<LawsTreeNodeMapper, Law
        } else {
            lawsTreeNodes = new ArrayList<>(nodesSet);
        }
-
        // 调用wrapTreeDataToTreeList方法生成树状数据
-        return FindsTreeNodeChildrenUtil.wrapTreeDataToTreeList(lawsTreeNodes);
+        List<LawsTreeNodeModel> lawsTreeNodeModels = FindsTreeNodeChildrenUtil.wrapTreeDataToTreeList(lawsTreeNodes);
+        // 添加我的收藏节点
+        LawsTreeNodeModel myCollect = new LawsTreeNodeModel();
+        myCollect.setKey("myCollect");
+        myCollect.setValue("myCollect");
+        myCollect.setTitle("我的收藏");
+        lawsTreeNodeModels.add(myCollect);
+        return lawsTreeNodeModels;
    }

    private Set<LawsTreeNode> findPathToRoot(List<LawsTreeNode> allNodes, LawsTreeNode node) {