Selaa lähdekoodia

【优化】字段重命名 job id 改为 task id

cherishsince 10 kuukautta sitten
vanhempi
commit
c0c2305c23

+ 3 - 3
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/image/AiImageDO.java

@@ -111,11 +111,11 @@ public class AiImageDO extends BaseDO {
     @TableField(typeHandler = JacksonTypeHandler.class)
     private List<MidjourneyNotifyReqVO.Button> buttons;
 
-    // TODO @芋艿:再瞅瞅
     /**
-     * midjourney proxy 关联的 job id
+     * midjourney proxy 关联的 task id
      */
-    private String jobId;
+    private String taskId;
+
     /**
      * 绘画错误信息
      */

+ 1 - 1
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/image/AiImageMapper.java

@@ -34,7 +34,7 @@ public interface AiImageMapper extends BaseMapperX<AiImageDO> {
      * @return
      */
     default AiImageDO selectByJobId(String id) {
-        return this.selectOne(new LambdaQueryWrapperX<AiImageDO>().eq(AiImageDO::getJobId, id));
+        return this.selectOne(new LambdaQueryWrapperX<AiImageDO>().eq(AiImageDO::getTaskId, id));
     }
 
     default PageResult<AiImageDO> selectPage(Long userId, PageParam pageReqVO) {

+ 2 - 2
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/image/AiImageServiceImpl.java

@@ -158,7 +158,7 @@ public class AiImageServiceImpl implements AiImageService {
         }
         imageMapper.updateById(new AiImageDO()
                 .setId(aiImageDO.getId())
-                .setJobId(submitRespVO.getResult())
+                .setTaskId(submitRespVO.getResult())
         );
         return aiImageDO.getId();
     }
@@ -228,7 +228,7 @@ public class AiImageServiceImpl implements AiImageService {
         midjourneyProxyClient.action(
                 new MidjourneyActionReqVO()
                         .setCustomId(customId)
-                        .setTaskId(aiImageDO.getJobId())
+                        .setTaskId(aiImageDO.getTaskId())
                         .setNotifyHook(midjourneyNotifyUrl)
         );
         return Boolean.TRUE;