提交代码
This commit is contained in:
jiang
parent
b79163cf0c
commit
cedbff4a07
|
|
@ -51,4 +51,12 @@ public interface ModelEvaluateMapper {
|
|||
* @return 结果
|
||||
*/
|
||||
public int deleteAiModelEvaluateByIds(Long[] ids);
|
||||
|
||||
/**
|
||||
* 唯一性
|
||||
*
|
||||
* @param aiModelEvaluate 模型评价
|
||||
* @return 条数
|
||||
*/
|
||||
public Integer checkUniqueEvaluateNameAndVersion(ModelEvaluate aiModelEvaluate);
|
||||
}
|
||||
|
|
|
|||
|
|
@ -265,6 +265,7 @@ public class DataSetBasicFileServiceImpl implements DataSetBasicFileService {
|
|||
entity.setIsDirectory("0");
|
||||
entity.setParentId(Long.valueOf(parentId));
|
||||
dataSetBasicFileMapper.insertDataSetBasicFile(entity);
|
||||
return AjaxResult.success(entity);
|
||||
}
|
||||
return AjaxResult.success();
|
||||
} catch (Exception e) {
|
||||
|
|
|
|||
|
|
@ -66,6 +66,10 @@ public class ModelEvaluateServiceImpl implements ModelEvaluateService {
|
|||
aiModelEvaluate.setCreateTime(DateUtils.getNowDate());
|
||||
aiModelEvaluate.setCreateBy(SecurityUtils.getUserId().toString());
|
||||
try {
|
||||
int i = aiModelEvaluateMapper.checkUniqueEvaluateNameAndVersion(aiModelEvaluate);
|
||||
if (i > 0) {
|
||||
return AjaxResult.error("评价已存在");
|
||||
}
|
||||
int rows = aiModelEvaluateMapper.insertAiModelEvaluate(aiModelEvaluate);
|
||||
return rows > 0 ? AjaxResult.success() : AjaxResult.error();
|
||||
} catch (Exception e) {
|
||||
|
|
@ -83,6 +87,10 @@ public class ModelEvaluateServiceImpl implements ModelEvaluateService {
|
|||
public AjaxResult updateAiModelEvaluate(ModelEvaluate aiModelEvaluate) {
|
||||
aiModelEvaluate.setUpdateTime(DateUtils.getNowDate());
|
||||
try {
|
||||
int i = aiModelEvaluateMapper.checkUniqueEvaluateNameAndVersion(aiModelEvaluate);
|
||||
if (i > 0) {
|
||||
return AjaxResult.error("评价已存在");
|
||||
}
|
||||
int rows = aiModelEvaluateMapper.updateAiModelEvaluate(aiModelEvaluate);
|
||||
return rows > 0 ? AjaxResult.success() : AjaxResult.error();
|
||||
} catch (Exception e) {
|
||||
|
|
|
|||
|
|
@ -151,7 +151,9 @@ public class DatasetServiceImpl implements DatasetService {
|
|||
}
|
||||
|
||||
|
||||
/**验证数据集名称的唯一性*/
|
||||
/**
|
||||
* 验证数据集名称的唯一性
|
||||
*/
|
||||
@Override
|
||||
public boolean checkDatasetNameUnique(DataSetEntity dataSet) {
|
||||
Long dataSetId = StringUtils.isNull(dataSet.getDatasetId()) ? -1L : dataSet.getDatasetId();
|
||||
|
|
@ -308,12 +310,9 @@ public class DatasetServiceImpl implements DatasetService {
|
|||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
/**
|
||||
* 插入数据集文件映射关系
|
||||
*
|
||||
* @param dataSet 数据集实体，包含数据集ID、输入ID、数据类型等信息
|
||||
*/
|
||||
private void insertDatasetFileMap(DataSetEntity dataSet) {
|
||||
|
|
@ -322,19 +321,25 @@ public class DatasetServiceImpl implements DatasetService {
|
|||
|
||||
// 根据数据类型获取支持的文件后缀列表，例如图片、视频或音频
|
||||
List<String> supportedFormats = getSupportedFormats(dataSet.getDataType());
|
||||
|
||||
if (ObjectUtils.isNotEmpty(dataSet.getFileIds())) {
|
||||
List<Long> fileIdList = Arrays.asList(dataSet.getFileIds());
|
||||
// 批量插入文件映射关系
|
||||
insertDatasetFileMap(dataSet.getDatasetId(), fileIdList, dataSet.getIsAnnotated());
|
||||
} else {
|
||||
// 筛选符合条件的文件ID列表
|
||||
List<Long> fileIdList = fileList.stream()
|
||||
.filter(file -> isValidFile(file, supportedFormats)) // 过滤出符合条件的文件
|
||||
.map(DataSetBasicFileEntity::getFileId) // 提取文件ID
|
||||
.collect(Collectors.toList());
|
||||
|
||||
// 批量插入文件映射关系
|
||||
insertDatasetFileMap(dataSet.getDatasetId(), fileIdList, dataSet.getIsAnnotated());
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据数据类型获取支持的文件后缀列表
|
||||
*
|
||||
* @param dataType 数据类型：0 - 图片，1 - 视频，2 - 音频
|
||||
* @return 支持的文件后缀列表，例如图片格式返回 [".jpg", ".png", ".jpeg", ".bmp"]
|
||||
*/
|
||||
|
|
@ -378,6 +383,7 @@ public class DatasetServiceImpl implements DatasetService {
|
|||
|
||||
/**
|
||||
* 批量插入文件映射关系
|
||||
*
|
||||
* @param datasetId 数据集ID
|
||||
* @param fileIdList 文件ID列表
|
||||
* @param isAnnotated 是否标注标志
|
||||
|
|
|
|||
|
|
@ -71,6 +71,18 @@
|
|||
<include refid="selectAiModelEvaluateVo"/>
|
||||
where ame.id = #{id}
|
||||
</select>
|
||||
<select id="checkUniqueEvaluateNameAndVersion" resultType="java.lang.Integer">
|
||||
SELECT COUNT(*)
|
||||
FROM ai_model_evaluate
|
||||
WHERE evaluate_name = #{evaluateName}
|
||||
AND enviro_conditions = #{enviroConditions}
|
||||
AND evaluate_version = #{evaluateVersion}
|
||||
AND process_speed = #{processSpeed}
|
||||
AND memory_usage = #{memoryUsage}
|
||||
<if test="id != null">
|
||||
AND id != #{id}
|
||||
</if>
|
||||
</select>
|
||||
|
||||
<insert id="insertAiModelEvaluate" parameterType="com.bonus.ai.domain.ModelEvaluate">
|
||||
insert into ai_model_evaluate
|
||||
|
|
@ -115,7 +127,6 @@
|
|||
<update id="updateAiModelEvaluate" parameterType="com.bonus.ai.domain.ModelEvaluate">
|
||||
update ai_model_evaluate
|
||||
<trim prefix="SET" suffixOverrides=",">
|
||||
<if test="createBy != null and createBy != ''">create_by = #{createBy},</if>
|
||||
<if test="evaluateName != null">evaluate_name = #{evaluateName},</if>
|
||||
<if test="evaluateVersion != null">evaluate_version = #{evaluateVersion},</if>
|
||||
<if test="sampleAll != null">sample_all = #{sampleAll},</if>
|
||||
|
|
|
|||
Loading…
Reference in New Issue