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 6d70169b..e00404d7 100644 --- a/src/main/java/com/ai/da/model/vo/ModuleChooseVO.java +++ b/src/main/java/com/ai/da/model/vo/ModuleChooseVO.java @@ -25,8 +25,8 @@ public class ModuleChooseVO { private List sketchBoard; // private moodBoardModuleChooseVO mannequin; private DesignModuleChooseVO design; - private List toProduct; - private List relight; + private ToProductOrRelightVO toProduct; + private ToProductOrRelightVO relight; private CavasModuleChooseVO canvas; diff --git a/src/main/java/com/ai/da/model/vo/ToProductOrRelightVO.java b/src/main/java/com/ai/da/model/vo/ToProductOrRelightVO.java new file mode 100644 index 00000000..a67fb1ce --- /dev/null +++ b/src/main/java/com/ai/da/model/vo/ToProductOrRelightVO.java @@ -0,0 +1,11 @@ +package com.ai.da.model.vo; + +import lombok.Data; + +import java.util.List; + +@Data +public class ToProductOrRelightVO { + private List likedList; + private List list; +} diff --git a/src/main/java/com/ai/da/model/vo/ToProductVO.java b/src/main/java/com/ai/da/model/vo/ToProductVO.java deleted file mode 100644 index d44ec250..00000000 --- a/src/main/java/com/ai/da/model/vo/ToProductVO.java +++ /dev/null @@ -1,8 +0,0 @@ -package com.ai.da.model.vo; - -import com.ai.da.mapper.primary.entity.UserLike; -import lombok.Data; - -@Data -public class ToProductVO extends UserLike { -} diff --git a/src/main/java/com/ai/da/service/impl/DesignItemServiceImpl.java b/src/main/java/com/ai/da/service/impl/DesignItemServiceImpl.java index abd1a102..41565f5b 100644 --- a/src/main/java/com/ai/da/service/impl/DesignItemServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/DesignItemServiceImpl.java @@ -596,7 +596,11 @@ public class DesignItemServiceImpl extends ServiceImpl i times = 1; } } else if (generateThroughImageTextDTO.getLevel1Type().equals(SKETCH_BOARD.getRealName())) { - creditsEventsEnum = CreditsEventsEnum.LOCAL_TEXT2IMG; + creditsEventsEnum = CreditsEventsEnum.SKETCH_BOARD; if (!StringUtil.isNullOrEmpty(generateThroughImageTextDTO.getModelName()) && generateThroughImageTextDTO.getModelName().equals("high")){ creditsEventsEnum = CreditsEventsEnum.LOCAL_TEXT2IMG_HIGH; times = 1; 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 3836eb79..b8b1a1d6 100644 --- a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java @@ -1932,60 +1932,49 @@ public class UserLikeGroupServiceImpl extends ServiceImpl qw = new QueryWrapper<>(); -// qw.lambda().eq(ToProductImageResult::getIsLike, 1); - qw.lambda().eq(ToProductImageResult::getProjectId, projectId); - qw.lambda().eq(ToProductImageResult::getResultType, "ToProductImage"); - List toProductImageResults = toProductImageResultMapper.selectList(qw); - for (ToProductImageResult toProductImageResult : toProductImageResults) { - toProductImageResult.setUrl(getMinioUrl(toProductImageResult.getUrl())); - } - List toProductImageResultVOS = CopyUtil.copyList(toProductImageResults, ToProductImageResultVO.class); - for (ToProductImageResultVO toProductImageResultVO : toProductImageResultVOS) { - if (toProductImageResultVO.getElementType().equals("ProductElement")) { - ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageResultVO.getElementId()); - toProductImageResultVO.setSourceUrl(getMinioUrl(toProductElement.getUrl())); - }else if ((toProductImageResultVO.getElementType().equals("DesignOutfit"))) { - TDesignPythonOutfit tDesignPythonOutfit = designPythonOutfitMapper.selectById(toProductImageResultVO.getElementId()); - toProductImageResultVO.setSourceUrl(getMinioUrl(tDesignPythonOutfit.getDesignUrl())); - }else { - ToProductImageResult toProductImageResult1 = toProductImageResultMapper.selectById(toProductImageResultVO.getElementId()); - toProductImageResultVO.setSourceUrl(getMinioUrl(toProductImageResult1.getUrl())); + qw.lambda() + .eq(ToProductImageResult::getProjectId, projectId) + .eq(ToProductImageResult::getResultType, resultType); + + List allResults = toProductImageResultMapper.selectList(qw); + + // 准备返回对象 + ToProductOrRelightVO resultVO = new ToProductOrRelightVO(); + + // 处理URL转换和分类 + List likedList = new ArrayList<>(); + List normalList = new ArrayList<>(); + + for (ToProductImageResult result : allResults) { + // 转换URL + result.setUrl(getMinioUrl(result.getUrl())); + // 复制到VO对象 + ToProductImageResultVO vo = CopyUtil.copyObject(result, ToProductImageResultVO.class); + // 设置sourceUrl + setSourceUrl(vo); + // 按isLike分类 + if (result.getIsLike() != null && result.getIsLike() == 1) { + likedList.add(vo); + } else { + normalList.add(vo); } } - moduleChooseVO.setToProduct(toProductImageResultVOS); - }else if (module.equals(Module.relight.name())) { - Long projectId = projectDTO.getId(); -// UserLikeGroup userLikeGroup = getByProjectId(projectDTO.getId()); -// Long userLikeGroupId = userLikeGroup.getId(); + // 设置结果 + resultVO.setLikedList(likedList); + resultVO.setList(normalList); - QueryWrapper qw = new QueryWrapper<>(); -// qw.lambda().eq(ToProductImageResult::getIsLike, 1); - qw.lambda().eq(ToProductImageResult::getProjectId, projectId); - qw.lambda().eq(ToProductImageResult::getResultType, "Relight"); - List toProductImageResults = toProductImageResultMapper.selectList(qw); - for (ToProductImageResult toProductImageResult : toProductImageResults) { - toProductImageResult.setUrl(getMinioUrl(toProductImageResult.getUrl())); + // 根据模块设置返回结果 + if (module.equals(Module.toProduct.name())) { + moduleChooseVO.setToProduct(resultVO); + } else { + moduleChooseVO.setRelight(resultVO); } - List toProductImageResultVOS = CopyUtil.copyList(toProductImageResults, ToProductImageResultVO.class); - for (ToProductImageResultVO toProductImageResultVO : toProductImageResultVOS) { - if (toProductImageResultVO.getElementType().equals("ProductElement")) { - ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageResultVO.getElementId()); - toProductImageResultVO.setSourceUrl(getMinioUrl(toProductElement.getUrl())); - }else if ((toProductImageResultVO.getElementType().equals("DesignOutfit"))) { - TDesignPythonOutfit tDesignPythonOutfit = designPythonOutfitMapper.selectById(toProductImageResultVO.getElementId()); - toProductImageResultVO.setSourceUrl(getMinioUrl(tDesignPythonOutfit.getDesignUrl())); - }else { - ToProductImageResult toProductImageResult1 = toProductImageResultMapper.selectById(toProductImageResultVO.getElementId()); - toProductImageResultVO.setSourceUrl(getMinioUrl(toProductImageResult1.getUrl())); - } - } - moduleChooseVO.setRelight(toProductImageResultVOS); }else if (module.equals(Module.deReconstruction.name())){ SketchReconstructionVO sketchReconstruction = generateService.getSketchReconstruction(projectDTO.getId()); moduleChooseVO.setDeReconstruction(sketchReconstruction); @@ -2061,6 +2050,20 @@ public class UserLikeGroupServiceImpl extends ServiceImpl qw = new QueryWrapper<>(); qw.lambda().eq(CollectionElementRelModel::getCollectionElementId, id);