From 73bab2338e9725361043b6f1baa8a1991971be9e Mon Sep 17 00:00:00 2001
From: shahaibo <1023316923@qq.com>
Date: Fri, 30 May 2025 14:04:18 +0800
Subject: [PATCH] =?UTF-8?q?TASK:collection=20sort=EF=BC=9B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main/java/com/ai/da/model/vo/DesignCollectionVO.java     | 2 ++
 src/main/java/com/ai/da/model/vo/PoseTransformationVO.java   | 2 ++
 src/main/java/com/ai/da/service/impl/DesignServiceImpl.java  | 5 +++--
 .../java/com/ai/da/service/impl/GenerateServiceImpl.java     | 1 +
 4 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/ai/da/model/vo/DesignCollectionVO.java b/src/main/java/com/ai/da/model/vo/DesignCollectionVO.java
index 807f364c..53b5626a 100644
--- a/src/main/java/com/ai/da/model/vo/DesignCollectionVO.java
+++ b/src/main/java/com/ai/da/model/vo/DesignCollectionVO.java
@@ -25,6 +25,8 @@ public class DesignCollectionVO {
 
     private List<String> UnfinishedList;
 
+    private String resultType;
+
     public DesignCollectionVO() {
     }
 }
diff --git a/src/main/java/com/ai/da/model/vo/PoseTransformationVO.java b/src/main/java/com/ai/da/model/vo/PoseTransformationVO.java
index 05e37e11..fc99dce3 100644
--- a/src/main/java/com/ai/da/model/vo/PoseTransformationVO.java
+++ b/src/main/java/com/ai/da/model/vo/PoseTransformationVO.java
@@ -26,6 +26,8 @@ public class PoseTransformationVO implements AllCollectionVO{
     private String status;
     private String collectionType;
 
+    private String resultType;
+
     public PoseTransformationVO(Long id, String taskId, String gifUrl, String videoUrl, String firstFrameUrl, byte isLiked, String status) {
         this.id = id;
         this.taskId = taskId;
diff --git a/src/main/java/com/ai/da/service/impl/DesignServiceImpl.java b/src/main/java/com/ai/da/service/impl/DesignServiceImpl.java
index 915da6ee..85a96133 100644
--- a/src/main/java/com/ai/da/service/impl/DesignServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/DesignServiceImpl.java
@@ -1131,9 +1131,9 @@ public class DesignServiceImpl extends ServiceImpl<DesignMapper, Design> impleme
         }
         List<DesignItem> designItems = designItemService.getByDesignId(designId);
         if (CollectionUtils.isEmpty(designItems)) {
-            return new DesignCollectionVO(designId, design.getCollectionId(), null, null, null);
+            return new DesignCollectionVO(designId, design.getCollectionId(), null, null, null, null);
         }
-        return new DesignCollectionVO(designId, design.getCollectionId(), coverDesignItemToVO(designItems), null, null);
+        return new DesignCollectionVO(designId, design.getCollectionId(), coverDesignItemToVO(designItems), null, null, null);
     }
 
     private List<DesignCollectionItemVO> coverDesignItemToVO(List<DesignItem> designItems) {
@@ -1813,6 +1813,7 @@ public class DesignServiceImpl extends ServiceImpl<DesignMapper, Design> impleme
             objectSignList.remove(objectSign);
         }
         result.setUnfinishedList(objectSignList);
+        result.setResultType(CollectionType.DESIGN.getValue());
         return result;
     }
 
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 419d5ebd..b0f364ee 100644
--- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
@@ -1024,6 +1024,7 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
                     poseTransformationVO.setFirstFrameUrl(minioUtil.getPreSignedUrl(poseTransformationVO.getFirstFrameUrl(), CommonConstant.MINIO_IMAGE_EXPIRE_TIME));
                 }
             }
+            poseTransformationVO.setResultType(CollectionType.POSE_TRANSFORM.getValue());
             return poseTransformationVO;
         }else {
             return new PoseTransformationVO();