From dfe99ef080aa6b11d1a8c82b6d9e4bcf64c966a0 Mon Sep 17 00:00:00 2001
From: shahaibo <1023316923@qq.com>
Date: Mon, 21 Apr 2025 22:13:53 +0800
Subject: [PATCH] =?UTF-8?q?TASK:=E6=A8=A1=E5=9D=97=E5=8C=96=EF=BC=9B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../com/ai/da/model/vo/CloudTaskResultVO.java |  3 +-
 .../ai/da/service/impl/DesignServiceImpl.java | 37 ++++++++++++-------
 2 files changed, 25 insertions(+), 15 deletions(-)

diff --git a/src/main/java/com/ai/da/model/vo/CloudTaskResultVO.java b/src/main/java/com/ai/da/model/vo/CloudTaskResultVO.java
index 4478ec5a..8b140a91 100644
--- a/src/main/java/com/ai/da/model/vo/CloudTaskResultVO.java
+++ b/src/main/java/com/ai/da/model/vo/CloudTaskResultVO.java
@@ -1,13 +1,14 @@
 package com.ai.da.model.vo;
 
 import com.ai.da.common.response.PageBaseResponse;
+import com.ai.da.mapper.primary.entity.TDesignPythonOutfit;
 import lombok.Data;
 
 import java.util.List;
 
 @Data
 public class CloudTaskResultVO {
-    private PageBaseResponse<DesignCollectionItemVO> design;
+    private List<DesignCollectionItemVO> design;
     private List<MagicToolResultVO> toProductImage;
     private List<MagicToolResultVO> relight;
 }
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 8ac45771..187c35c0 100644
--- a/src/main/java/com/ai/da/service/impl/DesignServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/DesignServiceImpl.java
@@ -2501,25 +2501,34 @@ public class DesignServiceImpl extends ServiceImpl<DesignMapper, Design> impleme
                 if (null != collectionId) {
                     QueryWrapper<TDesignPythonOutfit> qw = new QueryWrapper<>();
                     qw.lambda().eq(TDesignPythonOutfit::getCollectionId, collectionId);
-                    Page<TDesignPythonOutfit> page = designPythonOutfitMapper.selectPage(new Page<>(query.getPage(), query.getSize()), qw);
-                    IPage<DesignCollectionItemVO> convert = page.convert((Function<TDesignPythonOutfit, DesignCollectionItemVO>) outfit -> {
-                        if (outfit != null) {
-                            DesignCollectionItemVO vo = new DesignCollectionItemVO();
-                            vo.setDesignOutfitId(outfit.getId());
-                            vo.setDesignOutfitUrl(minioUtil.getPreSignedUrl(outfit.getDesignUrl(), 24 * 60));
-                            vo.setDesignItemId(outfit.getDesignItemId());
-                            return vo;
-                        }
-                        return null;
-                    });
-                    result.setDesign(PageBaseResponse.success(convert));
+                    List<TDesignPythonOutfit> list = designPythonOutfitMapper.selectList(qw);
+                    List<DesignCollectionItemVO> voList = new ArrayList<>();
+                    for (TDesignPythonOutfit tDesignPythonOutfit : list) {
+                        DesignCollectionItemVO vo = new DesignCollectionItemVO();
+                        vo.setDesignOutfitId(tDesignPythonOutfit.getId());
+                        vo.setDesignOutfitUrl(minioUtil.getPreSignedUrl(tDesignPythonOutfit.getDesignUrl(), 24 * 60));
+                        vo.setDesignItemId(tDesignPythonOutfit.getDesignItemId());
+                        voList.add(vo);
+                    }
+                    result.setDesign(voList);
+//                    IPage<DesignCollectionItemVO> convert = page.convert((Function<TDesignPythonOutfit, DesignCollectionItemVO>) outfit -> {
+//                        if (outfit != null) {
+//                            DesignCollectionItemVO vo = new DesignCollectionItemVO();
+//                            vo.setDesignOutfitId(outfit.getId());
+//                            vo.setDesignOutfitUrl(minioUtil.getPreSignedUrl(outfit.getDesignUrl(), 24 * 60));
+//                            vo.setDesignItemId(outfit.getDesignItemId());
+//                            return vo;
+//                        }
+//                        return null;
+//                    });
+//                    result.setDesign(PageBaseResponse.success(convert));
                     return result;
                 }else {
-                    result.setDesign(PageBaseResponse.success(new Page<>()));
+                    result.setDesign(new ArrayList<>());
                     return result;
                 }
             }
-            result.setDesign(PageBaseResponse.success(new Page<>()));
+            result.setDesign(new ArrayList<>());
             return result;
         } else if (query.getBuildType().equals(BuildType.TO_PRODUCT_IMAGE.getValue())) {
             List<MagicToolResultVO> results = new ArrayList<>();