提交代码

bonus-modules/bonus-ai/src/main/java/com/bonus/ai/mapper/ModelEvaluateMapper.java

@@ -51,4 +51,12 @@ public interface ModelEvaluateMapper {
      * @return 结果
      */
     public int deleteAiModelEvaluateByIds(Long[] ids);
+
+    /**
+     * 唯一性
+     *
+     * @param aiModelEvaluate 模型评价
+     * @return 条数
+     */
+    public Integer checkUniqueEvaluateNameAndVersion(ModelEvaluate aiModelEvaluate);
 }

bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/DataSetBasicFileServiceImpl.java

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

bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/ModelEvaluateServiceImpl.java

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

bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/dataset/DatasetServiceImpl.java

@@ -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 是否标注标志

bonus-modules/bonus-ai/src/main/resources/mapper/ModelEvaluateMapper.xml

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