DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

代码校验数据集名称是否存在问题

Browse Source
This commit is contained in:
limin 2025-01-03 09:48:25 +08:00
parent 4ddcbb3442
commit 0a88b5ca8d
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/ErrorCodeConstants.java
View File

@ -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, "数据集名称重复");
}

11
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java
View File

@ -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) {