From af1cff7ba0399b04bbbca642e188c666c5728886 Mon Sep 17 00:00:00 2001
From: jiang <jiang@bonus>
Date: Tue, 3 Dec 2024 18:07:15 +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

---
 .../bonus/ai/service/Impl/dataset/DatasetServiceImpl.java  | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/dataset/DatasetServiceImpl.java b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/dataset/DatasetServiceImpl.java
index 5a08652..3ac450d 100644
--- a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/dataset/DatasetServiceImpl.java
+++ b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/dataset/DatasetServiceImpl.java
@@ -18,6 +18,7 @@ import io.minio.errors.MinioException;
 import io.minio.messages.Item;
 import javafx.util.Pair;
 import org.apache.commons.io.IOUtils;
+import org.apache.commons.lang3.ObjectUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -108,8 +109,10 @@ public class DatasetServiceImpl implements DatasetService {
     @Override
     public List<DataSetEntity> getDatasetList(DataSetEntity dataSet) {
         try {
-            String userId = SecurityUtils.getUserId().toString();
-            dataSet.setCreateBy(userId);
+            if (ObjectUtils.isEmpty(dataSet.getIsPublic())){
+                String userId = SecurityUtils.getUserId().toString();
+                dataSet.setCreateBy(userId);
+            }
             return datasetMapper.getDatasetList(dataSet);
         } catch (Exception e) {
             log.error("获取数据集列表失败", e);