diff --git a/src/main/java/com/ai/da/model/enums/Module.java b/src/main/java/com/ai/da/model/enums/Module.java index 78c24bc5..8f9a2eed 100644 --- a/src/main/java/com/ai/da/model/enums/Module.java +++ b/src/main/java/com/ai/da/model/enums/Module.java @@ -14,7 +14,9 @@ public enum Module implements IEnumDisplay { poseTransfer("Pose Transfer"), canvas("Canvas"), patternMaking3D("3D Pattern Making"), - deReconstruction("De/Reconstruction"); + deReconstruction("De/Reconstruction"), + uploadElement("UploadElement"), + ; private final String value; diff --git a/src/main/java/com/ai/da/model/vo/ModuleChooseVO.java b/src/main/java/com/ai/da/model/vo/ModuleChooseVO.java index 0d899a9f..ae73e6f7 100644 --- a/src/main/java/com/ai/da/model/vo/ModuleChooseVO.java +++ b/src/main/java/com/ai/da/model/vo/ModuleChooseVO.java @@ -1,5 +1,6 @@ package com.ai.da.model.vo; +import com.ai.da.mapper.primary.entity.ToProductElement; import com.ai.da.model.dto.PortfolioDTO; import com.ai.da.model.dto.ToProductImageDTO; import io.swagger.annotations.ApiModel; @@ -37,4 +38,6 @@ public class ModuleChooseVO { private ThreeDModuleVO patternMaking3D; private List mannequin; + + private List uploadElement; } diff --git a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java index fcffafa2..3518f410 100644 --- a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java @@ -1747,6 +1747,14 @@ public class UserLikeGroupServiceImpl extends ServiceImpl qw = new QueryWrapper<>(); + qw.lambda().eq(ToProductElement::getProjectId, projectDTO.getId()); + List toProductElements = toProductElementMapper.selectList(qw); + for (ToProductElement toProductElement : toProductElements) { + toProductElement.setUrl(minioUtil.getPreSignedUrl(toProductElement.getUrl(), 24 * 60)); + } + moduleChooseVO.setUploadElement(toProductElements); } } return moduleChooseVO;