From 0a88b5ca8d023f065f45af9e7db671035a30e0a4 Mon Sep 17 00:00:00 2001
From: limin <limin@qq.com>
Date: Fri, 3 Jan 2025 09:48:25 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BB=A3=E7=A0=81=E6=A0=A1=E9=AA=8C=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E9=9B=86=E5=90=8D=E7=A7=B0=E6=98=AF=E5=90=A6=E5=AD=98?=
 =?UTF-8?q?=E5=9C=A8=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../yudao/module/llm/enums/ErrorCodeConstants.java    |  1 +
 .../llm/service/dataset/DatasetServiceImpl.java       | 11 +++++++++++
 2 files changed, 12 insertions(+)

diff --git a/yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/ErrorCodeConstants.java b/yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/ErrorCodeConstants.java
index d16389bb6..cfcb7451a 100644
--- a/yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/ErrorCodeConstants.java
+++ b/yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/ErrorCodeConstants.java
@@ -82,5 +82,6 @@ public interface ErrorCodeConstants {
     ErrorCode MODEL_ASSESS_TASK_MANUAL_BACKUP_NOT_EXISTS = new ErrorCode(10033, "人工评估备份不存在");
 
     ErrorCode MODEL_ASSESS_TASK_MANUAL_BACKUP_EXISTS = new ErrorCode(100_1002, "人工评估已备份");
+    ErrorCode DATASET_NAME_EXISTS = new ErrorCode(10034, "数据集名称重复");
 
 }
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java
index 04204abbc..90e43e4da 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java
@@ -35,6 +35,7 @@ import java.util.List;
 import java.util.stream.Collectors;
 
 import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
+import static cn.iocoder.yudao.module.llm.enums.ErrorCodeConstants.DATASET_NAME_EXISTS;
 import static cn.iocoder.yudao.module.llm.enums.ErrorCodeConstants.DATASET_NOT_EXISTS;
 
 /**
@@ -57,6 +58,8 @@ public class DatasetServiceImpl implements DatasetService {
 
     @Override
     public Long createDataset(DatasetSaveReqVO createReqVO) {
+        // 校验
+        validateDatasetNameExists(createReqVO);
         if(createReqVO.getType() == null){
             createReqVO.setType(0);
         }
@@ -95,6 +98,7 @@ public class DatasetServiceImpl implements DatasetService {
     public void updateDataset(DatasetSaveReqVO updateReqVO) {
         // 校验存在
         validateDatasetExists(updateReqVO.getId());
+        validateDatasetNameExists(updateReqVO);
         // 更新
         DatasetDO updateObj = BeanUtils.toBean(updateReqVO, DatasetDO.class);
         List<DatasetFilesSaveReqVO> datasetFiles = updateReqVO.getDatasetFiles();
@@ -133,6 +137,13 @@ public class DatasetServiceImpl implements DatasetService {
             throw exception(DATASET_NOT_EXISTS);
         }
     }
+    private void validateDatasetNameExists(DatasetSaveReqVO dateReqVO){
+        List<DatasetDO> datasetDOS = datasetMapper.selectList(new LambdaQueryWrapper<DatasetDO>()
+                .eq(DatasetDO::getDatasetName, dateReqVO.getDatasetName()));
+        if (CollectionUtils.isNotEmpty(datasetDOS)){
+            throw exception(DATASET_NAME_EXISTS);
+        }
+    }
 
     @Override
     public DatasetRespVO getDataset(Long id) {