From 1637db2fe300626e390b1bc0ab4f6d102a116426 Mon Sep 17 00:00:00 2001
From: shahaibo <1023316923@qq.com>
Date: Thu, 13 Mar 2025 17:57:27 +0800
Subject: [PATCH] =?UTF-8?q?TASK:AiDA=E6=A8=A1=E5=9D=97=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/com/ai/da/model/vo/CollectionElementVO.java     | 2 ++
 .../com/ai/da/service/impl/CollectionServiceImpl.java    | 9 ++++++---
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/ai/da/model/vo/CollectionElementVO.java b/src/main/java/com/ai/da/model/vo/CollectionElementVO.java
index ff4e5b88..53a1888a 100644
--- a/src/main/java/com/ai/da/model/vo/CollectionElementVO.java
+++ b/src/main/java/com/ai/da/model/vo/CollectionElementVO.java
@@ -40,4 +40,6 @@ public class CollectionElementVO {
 
     private String urlWithWhiteSide;
 
+    private String originalUrl;
+
 }
diff --git a/src/main/java/com/ai/da/service/impl/CollectionServiceImpl.java b/src/main/java/com/ai/da/service/impl/CollectionServiceImpl.java
index bf598dcd..a90ba255 100644
--- a/src/main/java/com/ai/da/service/impl/CollectionServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/CollectionServiceImpl.java
@@ -126,12 +126,13 @@ public class CollectionServiceImpl extends ServiceImpl<CollectionMapper, Collect
                     List<CollectionElementVO> moodBoards = CopyUtil.copyList(v, CollectionElementVO.class, (o, d) -> {
                         d.setDesignType(DesignTypeEnum.COLLECTION.getRealName());
                         String url = o.getUrl();
+                        d.setOriginalUrl(url);
                         if (minioUtil.doesObjectExist(url)) {
                             d.setUrl(minioUtil.getPreSignedUrl(url, 24 * 60));
                         }
                     });
                     response.setMoodBoards(moodBoards.stream()
-                            .filter(d -> minioUtil.doesObjectExist(d.getUrl()))
+                            .filter(d -> minioUtil.doesObjectExist(d.getOriginalUrl()))
                             .collect(Collectors.toList()));
                     break;
 
@@ -140,12 +141,13 @@ public class CollectionServiceImpl extends ServiceImpl<CollectionMapper, Collect
                         d.setIsPin(o.getHasPin());
                         d.setDesignType(DesignTypeEnum.COLLECTION.getRealName());
                         String url = o.getUrl();
+                        d.setOriginalUrl(url);
                         if (minioUtil.doesObjectExist(url)) {
                             d.setUrl(minioUtil.getPreSignedUrl(url, 24 * 60));
                         }
                     });
                     response.setPrintBoards(printBoards.stream()
-                            .filter(d -> minioUtil.doesObjectExist(d.getUrl()))
+                            .filter(d -> minioUtil.doesObjectExist(d.getOriginalUrl()))
                             .collect(Collectors.toList()));
                     break;
 
@@ -154,6 +156,7 @@ public class CollectionServiceImpl extends ServiceImpl<CollectionMapper, Collect
                         d.setIsPin(o.getHasPin());
                         d.setDesignType(DesignTypeEnum.COLLECTION.getRealName());
                         String url = o.getUrl();
+                        d.setOriginalUrl(url);
                         if (minioUtil.doesObjectExist(url)) {
                             d.setUrl(minioUtil.getPreSignedUrl(url, 24 * 60));
                             if (url.contains(".")) {
@@ -165,7 +168,7 @@ public class CollectionServiceImpl extends ServiceImpl<CollectionMapper, Collect
                         }
                     });
                     response.setSketchBoards(sketchBoards.stream()
-                            .filter(d -> minioUtil.doesObjectExist(d.getUrl()))
+                            .filter(d -> minioUtil.doesObjectExist(d.getOriginalUrl()))
                             .collect(Collectors.toList()));
                     break;