From 1c32aaa46c75a2f1077f7b3da4246d79715689f8 Mon Sep 17 00:00:00 2001
From: litianxiang <litianxiangxtt@163.com>
Date: Wed, 15 Oct 2025 14:47:04 +0800
Subject: [PATCH] =?UTF-8?q?=E9=98=B2=E6=AD=A2=E5=9B=A0=E4=B8=BA=E4=BA=8C?=
 =?UTF-8?q?=E5=88=9B=E5=AF=BC=E8=87=B4=E7=9A=84=E6=95=B0=E6=8D=AE=E5=AD=98?=
 =?UTF-8?q?=E5=9C=A8taskId=E7=9B=B8=E5=90=8C=E5=AF=BC=E8=87=B4=E6=8A=A5?=
 =?UTF-8?q?=E9=94=99=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../ai/da/service/impl/GenerateServiceImpl.java  | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

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 b4b33323..770aa0e0 100644
--- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
@@ -2685,9 +2685,19 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
             }
         }
 
-        // 处理父ID逻辑
-        processParentId(vo, dbItem != null ? dbItem :
-                poseTransformationMapper.selectOne(new QueryWrapper<PoseTransformation>().eq("unique_id", taskId)));
+        AuthPrincipalVo userHolder = UserContext.getUserHolder();
+        Long userId = userHolder.getId();
+        if (userId != null){
+            // 处理父ID逻辑
+            processParentId(vo, dbItem != null ? dbItem :
+                    poseTransformationMapper.selectOne(new QueryWrapper<PoseTransformation>().eq("unique_id", taskId).eq("account_id", userId)));
+
+        }else {
+            // 处理父ID逻辑
+            processParentId(vo, dbItem != null ? dbItem :
+                    poseTransformationMapper.selectList(new QueryWrapper<PoseTransformation>().eq("unique_id", taskId)).get(0));
+
+        }
 
         return vo;
     }