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 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);