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 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 minioUtil.doesObjectExist(d.getUrl())) + .filter(d -> minioUtil.doesObjectExist(d.getOriginalUrl())) .collect(Collectors.toList())); break; @@ -154,6 +156,7 @@ public class CollectionServiceImpl extends ServiceImpl minioUtil.doesObjectExist(d.getUrl())) + .filter(d -> minioUtil.doesObjectExist(d.getOriginalUrl())) .collect(Collectors.toList())); break;