From 84a2063a585c6ba46c4eb5a25355fbadbf34afcc Mon Sep 17 00:00:00 2001
From: jiang <jiang@bonus>
Date: Mon, 2 Dec 2024 13:48:56 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/com/bonus/ai/controller/DataSetLabelsController.java  | 2 +-
 .../com/bonus/ai/service/Impl/DataSetLabelsServiceImpl.java    | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/controller/DataSetLabelsController.java b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/controller/DataSetLabelsController.java
index ee3dad2..fd23bdf 100644
--- a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/controller/DataSetLabelsController.java
+++ b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/controller/DataSetLabelsController.java
@@ -102,6 +102,6 @@ public class DataSetLabelsController extends BaseController
 
     @PostMapping("/getLabelsTree/{labelId}")
     public AjaxResult getLabelsTree(@PathVariable Long labelId){
-        return dataSetLabelsService.getLabelsTree(labelId);
+          return dataSetLabelsService.getLabelsTree(labelId);
     }
 }
diff --git a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/DataSetLabelsServiceImpl.java b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/DataSetLabelsServiceImpl.java
index 9bacb9e..75ff205 100644
--- a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/DataSetLabelsServiceImpl.java
+++ b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/DataSetLabelsServiceImpl.java
@@ -130,6 +130,9 @@ public class DataSetLabelsServiceImpl implements DataSetLabelsService
     public AjaxResult getLabelsTree(Long labelId) {
         DataSetLabels entity = new DataSetLabels();
         List<DataSetLabels> dataSetLabels = dataSetLabelsMapper.selectDataSetLabelsList(entity);
+        if (labelId==0){
+            return AjaxResult.success(dataSetLabels);
+        }
         Set<Long> labelIds = getFileWithChildren(labelId);
         for (Long id : labelIds){
             dataSetLabels.removeIf(item -> item.getLabelId().equals(id));