From 2041809ea6bf84e517fe4f987efb6c56a7422512 Mon Sep 17 00:00:00 2001
From: xupei <xupei3360@163.com>
Date: Sun, 20 Jul 2025 19:19:07 +0800
Subject: [PATCH] =?UTF-8?q?TASK:pose=20transfer=E6=B7=BB=E5=8A=A0=E8=BF=94?=
 =?UTF-8?q?=E5=9B=9E=E5=8F=82=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/com/ai/da/service/impl/GenerateServiceImpl.java     | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
index bd307380..113288ff 100644
--- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
@@ -16,6 +16,7 @@ import com.ai.da.mapper.primary.*;
 import com.ai.da.mapper.primary.entity.*;
 import com.ai.da.model.dto.*;
 import com.ai.da.model.enums.CollectionType;
+import com.ai.da.model.enums.Module;
 import com.ai.da.model.enums.SketchStyle;
 import com.ai.da.model.vo.*;
 import com.ai.da.python.PythonService;
@@ -1243,6 +1244,8 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
         poseTransformationMapper.insert(poseTransformation);
         // 当需要默认like
         ToProductImageResultVO toProductImageResultVO = new ToProductImageResultVO();
+        toProductImageResultVO.setParentId(poseTransformDTO.getParentId());
+        toProductImageResultVO.setResultType(Module.poseTransfer.getValue());
         if (Objects.nonNull(poseTransformDTO.getIsDefaultLike()) && poseTransformDTO.getIsDefaultLike()) {
             // 满足条件下添加到like
             poseTransformation.setIsLiked((byte) 1);
@@ -1255,6 +1258,7 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
         } else if (Objects.nonNull(poseTransformDTO.getIsDefaultLike()) && Objects.nonNull(poseTransformDTO.getParentId())) {
             toProductImageResultVO.setParentId(poseTransformDTO.getParentId());
         }
+        toProductImageResultVO.setTaskId(taskId);
 
 
         if (flag) {
@@ -1262,7 +1266,6 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
             creditsService.addRecordToCreditsDeduction(accountId, taskId, creditsEventsEnum);
             // 6.1 添加积分扣除记录到db
             creditsService.preInsert(accountId, creditsEventsEnum.getName(), taskId, Boolean.TRUE, null);
-            toProductImageResultVO.setTaskId(taskId);
             // 更新项目更新时间
             projectService.modifyProjectUpdateTime(projectId);
             return toProductImageResultVO;