From 1dc6c6f8c41716f333d191c1893f394d18a39420 Mon Sep 17 00:00:00 2001
From: xupei <xupei3360@163.com>
Date: Tue, 8 Jul 2025 18:53:21 +0800
Subject: [PATCH] =?UTF-8?q?TASK:=E7=94=9F=E6=88=90=E8=8E=B7=E5=8F=96?=
 =?UTF-8?q?=E7=BB=93=E6=9E=9C=E6=8E=A5=E5=8F=A3=E8=BF=94=E5=9B=9EparentId?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main/java/com/ai/da/service/UserLikeGroupService.java   | 2 ++
 .../java/com/ai/da/service/impl/GenerateServiceImpl.java    | 4 ++++
 .../com/ai/da/service/impl/UserLikeGroupServiceImpl.java    | 6 +++++-
 3 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/ai/da/service/UserLikeGroupService.java b/src/main/java/com/ai/da/service/UserLikeGroupService.java
index 3fa00e04..b054d1cd 100644
--- a/src/main/java/com/ai/da/service/UserLikeGroupService.java
+++ b/src/main/java/com/ai/da/service/UserLikeGroupService.java
@@ -58,6 +58,8 @@ public interface UserLikeGroupService extends IService<UserLikeGroup> {
 
     List<MagicToolResultVO> getToProductImageResultList(List<String> taskIdList);
 
+    Long getParentIdByElementIdAndElementType(Long elementId, String elementType);
+
     JSONObject exportSearch(ExportSearchDTO exportSearchDTO);
 
     CanvasElementUpload canvasElementUpload(MultipartFile file);
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 118ec156..fc1b148a 100644
--- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
@@ -1259,6 +1259,8 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
                             poseTransformationVO.setFirstFrameUrl(minioUtil.getPreSignedUrl(url, CommonConstant.MINIO_IMAGE_EXPIRE_TIME)));
                 }
                 poseTransformationVO.setResultType(CollectionType.POSE_TRANSFORM.getValue());
+                Long parentId = userLikeGroupService.getParentIdByElementIdAndElementType(poseTransformationVO.getId(), CollectionType.POSE_TRANSFORM.getValue());
+                poseTransformationVO.setParentId(parentId);
                 poseTransformationVOS.add(poseTransformationVO);
             } else if(type.equals("wx")){
                 poseTransformationVOS.add(getAnimateResult(taskId));
@@ -1924,6 +1926,8 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
                     if (!StringUtil.isNullOrEmpty(poseTransformation.getFirstFrameUrl())){
                         poseTransformationVO.setFirstFrameUrl(minioUtil.getPreSignedUrl(poseTransformation.getFirstFrameUrl(), CommonConstant.MINIO_IMAGE_EXPIRE_TIME));
                     }
+                    Long parentId = userLikeGroupService.getParentIdByElementIdAndElementType(poseTransformation.getId(), CollectionType.POSE_TRANSFORM.getValue());
+                    poseTransformationVO.setParentId(parentId);
                     // 执行积分扣除
                     Long accountId = poseTransformation.getAccountId();
                     Boolean flag = creditsService.taskCreditsDeduction(accountId, taskId);
diff --git a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
index 7fbe1ada..9d15568a 100644
--- a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
@@ -797,7 +797,7 @@ public class UserLikeGroupServiceImpl extends ServiceImpl<UserLikeGroupMapper, U
         return results;
     }
 
-    private Long getParentIdByElementIdAndElementType(Long elementId, String elementType) {
+    public Long getParentIdByElementIdAndElementType(Long elementId, String elementType) {
         QueryWrapper<CollectionSort> qw = new QueryWrapper<>();
         qw.lambda().eq(CollectionSort::getRelationId, elementId);
 
@@ -820,6 +820,8 @@ public class UserLikeGroupServiceImpl extends ServiceImpl<UserLikeGroupMapper, U
             return CollectionType.DESIGN.getValue();
         }else if (elementType.equals("ToProductImage")) {
             return CollectionType.TO_PRODUCT_IMAGE.getValue();
+        }else if (elementType.equals("PoseTransfer")) {
+            return CollectionType.POSE_TRANSFORM.getValue();
         }else {
             return "";
         }
@@ -847,6 +849,8 @@ public class UserLikeGroupServiceImpl extends ServiceImpl<UserLikeGroupMapper, U
             ToProductImageResult toProductImage = toProductImageResultMapper.selectById(toProductImageResult.getElementId());
             magicToolResultVO.setSourceUrl(minioUtil.getPreSignedUrl(toProductImage.getUrl(), 24 * 60));
         }
+        Long parentId = getParentIdByElementIdAndElementType(toProductImageResult.getId(), toProductImageResult.getResultType());
+        magicToolResultVO.setParentId(parentId);
         return magicToolResultVO;
     }