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 i } } - // 处理父ID逻辑 - processParentId(vo, dbItem != null ? dbItem : - poseTransformationMapper.selectOne(new QueryWrapper().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().eq("unique_id", taskId).eq("account_id", userId))); + + }else { + // 处理父ID逻辑 + processParentId(vo, dbItem != null ? dbItem : + poseTransformationMapper.selectList(new QueryWrapper().eq("unique_id", taskId)).get(0)); + + } return vo; }