From bb5d1d3335fdb158e6e59ae073a7bceae00d538d Mon Sep 17 00:00:00 2001
From: hzj <826100833@qq.com>
Date: Wed, 18 Sep 2024 18:43:13 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B5=84=E6=96=99=E5=BA=93=E4=BB=A3=E7=A0=81?=
 =?UTF-8?q?=E6=8F=90=E4=BA=A4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../com/njcn/supervision/pojo/po/databank/LibAlgorithm.java    | 3 +++
 .../service/databank/impl/LibAlgorithmServiceImpl.java         | 2 ++
 2 files changed, 5 insertions(+)

diff --git a/pqs-supervision/supervision-api/src/main/java/com/njcn/supervision/pojo/po/databank/LibAlgorithm.java b/pqs-supervision/supervision-api/src/main/java/com/njcn/supervision/pojo/po/databank/LibAlgorithm.java
index 66566b5c0..37ab9856c 100644
--- a/pqs-supervision/supervision-api/src/main/java/com/njcn/supervision/pojo/po/databank/LibAlgorithm.java
+++ b/pqs-supervision/supervision-api/src/main/java/com/njcn/supervision/pojo/po/databank/LibAlgorithm.java
@@ -83,5 +83,8 @@ public class LibAlgorithm extends BaseEntity {
     @TableField(exist = false)
     private List<LibAlgorithm> children;
 
+    @TableField(exist = false)
+    private Integer level;
+
 
 }
\ No newline at end of file
diff --git a/pqs-supervision/supervision-boot/src/main/java/com/njcn/supervision/service/databank/impl/LibAlgorithmServiceImpl.java b/pqs-supervision/supervision-boot/src/main/java/com/njcn/supervision/service/databank/impl/LibAlgorithmServiceImpl.java
index 8eb117f9c..05573c232 100644
--- a/pqs-supervision/supervision-boot/src/main/java/com/njcn/supervision/service/databank/impl/LibAlgorithmServiceImpl.java
+++ b/pqs-supervision/supervision-boot/src/main/java/com/njcn/supervision/service/databank/impl/LibAlgorithmServiceImpl.java
@@ -93,12 +93,14 @@ public class LibAlgorithmServiceImpl extends ServiceImpl<LibAlgorithmMapper, Lib
 
         List<LibAlgorithm> libAlgorithmList = this.lambdaQuery().eq(LibAlgorithm::getStatus,1).list();
         List<LibAlgorithm> tree = libAlgorithmList.stream().filter(item -> item.getPid().equals("0")).peek(item -> {
+            item.setLevel(1);
             item.setChildren(getChildren(item, libAlgorithmList));
         }).collect(Collectors.toList());
         return tree;
     }
     private List<LibAlgorithm> getChildren(LibAlgorithm libAlgorithm, List<LibAlgorithm> all) {
         return all.stream().filter(item -> item.getPid().equals(libAlgorithm.getId())).peek(item -> {
+            item.setLevel(libAlgorithm.getLevel()+1);
             item.setChildren(getChildren(item, all));
         }).collect(Collectors.toList());
     }