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 060f6132..d7bbee87 100644 --- a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java @@ -1,3089 +1,3089 @@ -package com.ai.da.service.impl; - -import cn.hutool.core.collection.CollectionUtil; -import com.ai.da.common.config.exception.BusinessException; -import com.ai.da.common.constant.CommonConstant; -import com.ai.da.common.context.UserContext; -import com.ai.da.common.enums.CollectionLevel1TypeEnum; -import com.ai.da.common.enums.CreditsEventsEnum; -import com.ai.da.common.enums.DesignTypeEnum; -import com.ai.da.common.enums.LibraryLevel1TypeEnum; -import com.ai.da.common.response.PageBaseResponse; -import com.ai.da.common.response.ResultEnum; -import com.ai.da.common.utils.*; -import com.ai.da.mapper.primary.*; -import com.ai.da.mapper.primary.entity.*; -import com.ai.da.mapper.primary.entity.Collection; -import com.ai.da.mapper.secondary.AttributeRetrievalMapper; -import com.ai.da.mapper.secondary.entity.AttributeRecognitionJSON; -import com.ai.da.model.dto.*; -import com.ai.da.model.enums.CollectionType; -import com.ai.da.model.enums.Module; -import com.ai.da.model.vo.*; -import com.ai.da.python.PythonService; -import com.ai.da.service.*; -import com.alibaba.fastjson.JSON; -import com.alibaba.fastjson.JSONArray; -import com.alibaba.fastjson.JSONObject; -import com.alibaba.fastjson.serializer.SerializerFeature; -import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; -import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper; -import com.baomidou.mybatisplus.core.metadata.IPage; -import com.baomidou.mybatisplus.extension.plugins.pagination.Page; -import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; -import com.google.common.base.Function; -import com.google.gson.Gson; -import io.minio.errors.MinioException; -import io.netty.util.internal.StringUtil; -import lombok.extern.slf4j.Slf4j; -import org.springframework.beans.factory.annotation.Value; -import org.springframework.stereotype.Service; -import org.springframework.transaction.annotation.Isolation; -import org.springframework.transaction.annotation.Transactional; -import org.springframework.util.CollectionUtils; -import org.springframework.util.StringUtils; -import org.springframework.web.multipart.MultipartFile; - -import javax.annotation.Resource; -import javax.servlet.http.HttpServletResponse; -import java.io.*; -import java.net.URLEncoder; -import java.nio.charset.StandardCharsets; -import java.time.Duration; -import java.time.LocalDateTime; -import java.time.ZoneId; -import java.util.*; -import java.util.stream.Collectors; - -/** - * 服务实现类 - * - * @author yanglei - * @since 2022-09-30 - */ -@Slf4j -@Service -public class UserLikeGroupServiceImpl extends ServiceImpl implements UserLikeGroupService { - @Resource - private UserLikeGroupMapper userLikeGroupMapper; - @Resource - private AccountService accountService; - @Resource - private CollectionService collectionService; - @Resource - private UserLikeService userLikeService; - @Resource - private WorkspaceService workspaceService; - @Resource - private UserLikeMapper userLikeMapper; - @Resource - private MinioUtil minioUtil; - @Resource - private TDesignPythonOutfitMapper designPythonOutfitMapper; - @Resource - private DesignMapper designMapper; - @Resource - private DesignService designService; - @Resource - private SysFileMapper sysFileMapper; - @Resource - private LibraryMapper libraryMapper; - @Resource - private GenerateDetailMapper generateDetailMapper; - @Resource - private PortfolioMapper portfolioMapper; - @Resource - private TagsMapper tagsMapper; - @Resource - private DesignItemDetailService designItemDetailService; - @Resource - private CollectionElementMapper collectionElementMapper; - @Resource - private AttributeRetrievalMapper attributeRetrievalMapper; - @Resource - private ProductImageAttributeMapper productImageAttributeMapper; - @Resource - private CollectionSortMapper collectionSortMapper; - @Resource - private ClassificationService classificationService; - @Resource -// private ProjectMapper projectMapper; - private ProjectService projectService; - @Resource - private PortfolioService portfolioService; - @Resource - private LibraryModelPointMapper libraryModelPointMapper; - @Resource - private LibraryService libraryService; - @Resource - private BrandDNAMapper brandDNAMapper; - @Resource - private BrandRelLibraryMapper brandRelLibraryMapper; - @Resource - private ThreeDLayoutMapper threeDLayoutMapper; - @Resource - private ThreeDPatternLayoutMapper threeDPatternLayoutMapper; - @Resource - private ThreeDDetailMapper threeDDetailMapper; - @Resource - private ThreeDSimpleMapper threeDSimpleMapper; - @Resource - private ThreeDModuleMapper threeDModuleMapper; - @Resource - private ProductImageService productImageService; - @Resource - private CollectionElementRelModelMapper collectionElementRelModelMapper; - @Resource - private CollectionSortService collectionSortService; - @Resource - private GenerateService generateService; - @Resource - private ToProductElementMapper toProductElementMapper; - @Resource - private ToProductImageRecordMapper toProductImageRecordMapper; - @Resource - private ToProductImageResultMapper toProductImageResultMapper; - @Resource - private CloudTaskMapper cloudTaskMapper; - @Resource - private PythonService pythonService; - @Resource - private CreditsService creditsService; - @Resource - private PoseTransformationMapper poseTransformationMapper; - @Resource - private ExportFileMapper exportFileMapper; - @Resource - private DesignItemDetailCanvasMapper designItemDetailCanvasMapper; - - @Override - public void deleteUserGroup(Long userGroupId) { - UserLikeGroup group = getById(userGroupId); - if (Objects.isNull(group)) { - throw new BusinessException("history.not.found"); - } - userLikeGroupMapper.deleteById(userGroupId); - } - - @Override - public HistoryUpdateVO updateUserGroupName(Long userGroupId, String userGroupName, String timeZone) { - UserLikeGroup group = getById(userGroupId); - if (Objects.isNull(group)) { - throw new BusinessException("history.not.found"); - } - QueryWrapper queryWrapper = new QueryWrapper<>(); - queryWrapper.eq("id", userGroupId); - UserLikeGroup userLikeGroup = new UserLikeGroup(); - userLikeGroup.setName(userGroupName); - userLikeGroup.setUpdateDate(DateUtil.getByTimeZone(timeZone)); - log.info("跟新group名字###{}###时间是###{}", userLikeGroup.getUpdateDate(), TimeZone.getDefault()); - userLikeGroupMapper.update(userLikeGroup, queryWrapper); - return new HistoryUpdateVO(userLikeGroup.getName(), userLikeGroup.getUpdateDate().getTime()); - } - -// public static void main(String[] args) { -// System.out.println(TimeZone.getDefault()); -// Date kolkata =DateUtil.getByTimeZone("Asia/Kolkata"); -// Date shanghai =DateUtil.getByTimeZone("Asia/Shanghai"); -// System.out.println(kolkata); -// System.out.println(shanghai); -// System.out.println(kolkata.getTime()); -// System.out.println(shanghai.getTime()); -// System.out.println(kolkata.getTime() == shanghai.getTime()); -// } - - private List selectByUserId(Long userId) { - QueryWrapper queryWrapper = new QueryWrapper<>(); - queryWrapper.eq("accountId", userId); - return userLikeGroupMapper.selectList(queryWrapper); - } - - @Override - public Long insertUserGroup(Long userId, Long collectionId, String timeZone, Long projectId) { - UserLikeGroup userLikeGroup = new UserLikeGroup(); - //默认当前时间 - userLikeGroup.setName(DateUtil.dateToStr(new Date(), DateUtil.YYYY_MM_DD_HH_MM_SS)); - userLikeGroup.setCreateDate(DateUtil.getByTimeZone(timeZone)); - userLikeGroup.setUpdateDate(DateUtil.getByTimeZone(timeZone)); - userLikeGroup.setAccountId(userId); - userLikeGroup.setCollectionId(collectionId); - userLikeGroup.setOriginal(1); - userLikeGroup.setProjectId(projectId); - userLikeGroupMapper.insertUserLikeGroup(userLikeGroup); - return userLikeGroup.getId(); - } - - @Override - public UserLikeChooseVO choose(Long userGroupId) { - UserLikeGroup group = getById(userGroupId); - if (Objects.isNull(group)) { - throw new BusinessException("history.not.found"); - } - List userLikeVOS = userLikeService.getGroupDetail(userGroupId); - String sex = null; - - QueryWrapper userLikeSortQw = new QueryWrapper<>(); - userLikeSortQw.lambda().eq(CollectionSort::getUserLikeGroupId, userGroupId); - List userLikeSortList = collectionSortMapper.selectList(userLikeSortQw); - if (CollectionUtil.isEmpty(userLikeSortList)) { - Integer sort = 1; - for (UserLikeVO userLikeVO : userLikeVOS) { - CollectionSort userLikeSort = new CollectionSort(); - userLikeSort.setUserLikeId(userLikeVO.getId()); - userLikeSort.setUserLikeGroupId(userGroupId); - userLikeSort.setSort(sort); - collectionSortMapper.insert(userLikeSort); - sort ++; - } - } - - userLikeVOS.forEach(o -> { - TDesignPythonOutfit tDesignPythonOutfit1 = designPythonOutfitMapper.selectById(o.getDesignOutfitId()); - o.setUrl(tDesignPythonOutfit1.getDesignUrl()); - if (o.getUrl().contains("/")) { - int index = o.getUrl().lastIndexOf("/"); - o.setPictureName(o.getUrl().substring(index + 1)); - } - o.setDesignOutfitUrl(minioUtil.getPreSignedUrl(o.getUrl(), 24 * 60)); - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(TDesignPythonOutfit::getDesignItemId, o.getDesignItemId()); - List tDesignPythonOutfits = designPythonOutfitMapper.selectList(qw); - if (CollectionUtil.isNotEmpty(tDesignPythonOutfits)) { - TDesignPythonOutfit tDesignPythonOutfit = tDesignPythonOutfits.get(0); - o.setDesignOutfitId(tDesignPythonOutfit.getId()); - } - - QueryWrapper userLikeSortQueryWrapper = new QueryWrapper<>(); - userLikeSortQueryWrapper.lambda().eq(CollectionSort::getUserLikeId, o.getId()); - List userLikeSorts = collectionSortMapper.selectList(userLikeSortQueryWrapper); - if (CollectionUtil.isNotEmpty(userLikeSorts)) { - CollectionSort userLikeSort = userLikeSorts.get(0); - o.setSort(userLikeSort.getSort()); - o.setUserLikeSortId(userLikeSort.getId()); - } - }); - UserLikeCollectionVO userLikeCollection = collectionService.chooseCollection(group.getCollectionId()); - Integer beenPublished = 0; - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(Portfolio::getUserLikeGroupSourceId, userGroupId); - List portfolios = portfolioMapper.selectList(qw); -// Portfolio portfolio = new Portfolio(); - PortfolioDTO portfolioDTO = new PortfolioDTO(); - if (CollectionUtil.isNotEmpty(portfolios)) { -// portfolio = portfolios.get(0); - portfolioDTO = CopyUtil.copyObject(portfolios.get(0), PortfolioDTO.class); - beenPublished = 1; - portfolioDTO.setTagsDTO(tagsMapper.getTagByPortfolioId(portfolioDTO.getId())); - } - return new UserLikeChooseVO(userGroupId, userLikeVOS, userLikeCollection, sex, beenPublished, portfolioDTO); - } - - @Override - public ProjectChooseVO choose(ProjectDTO projectDTO) { - - Project project = projectService.getById(projectDTO.getId()); - - Long workspaceId = workspaceService.getByProjectId(project.getId()); - WorkspaceVO workspaceVO = new WorkspaceVO(); - if (null != workspaceId) { - workspaceVO = workspaceService.getByIdForProject(workspaceId); - } - - String process = project.getProcess(); - String name = project.getName(); - - UserLikeGroup userLikeGroup = userLikeService.getUserLikeGroupByProjectId(project.getId()); - Long collectionId = null; - if (Objects.nonNull(userLikeGroup)) { - Collection collection = collectionService.getById(userLikeGroup.getCollectionId()); - if (Objects.nonNull(collection)) { - collectionId = collection.getId(); - } - } - - Integer beenPublished = 0; - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(Portfolio::getProjectId, projectDTO.getId()); - List portfolios = portfolioMapper.selectList(qw); -// Portfolio portfolio = new Portfolio(); - PortfolioDTO portfolioDTO = new PortfolioDTO(); - if (CollectionUtil.isNotEmpty(portfolios)) { -// portfolio = portfolios.get(0); - portfolioDTO = CopyUtil.copyObject(portfolios.get(0), PortfolioDTO.class); - beenPublished = 1; - portfolioDTO.setTagsDTO(tagsMapper.getTagByPortfolioId(portfolioDTO.getId())); - } - - return new ProjectChooseVO(projectDTO.getId(), name, workspaceVO, process, collectionId, beenPublished, portfolioDTO); - } - - @Override - public UserLikeGroup getByProjectId(Long projectId) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(UserLikeGroup::getProjectId, projectId); - List userLikeGroupList = userLikeGroupMapper.selectList(qw); - if (CollectionUtil.isEmpty(userLikeGroupList)) { - return null; - } - return userLikeGroupList.get(0); - } - - @Override - public void deleteTrialData(Long userId) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(UserLikeGroup::getAccountId, userId); - List userLikeGroupIdList = userLikeGroupMapper.selectList(qw).stream() - .map(UserLikeGroup::getId) - .collect(Collectors.toList()); - userLikeGroupMapper.delete(qw); - if (CollectionUtil.isNotEmpty(userLikeGroupIdList)) { - QueryWrapper userLikeQueryWrapper = new QueryWrapper<>(); - userLikeQueryWrapper.lambda().in(UserLike::getUserLikeGroupId, userLikeGroupIdList); - userLikeMapper.delete(userLikeQueryWrapper); - } - } - - @Override - @Transactional(rollbackFor = Exception.class) - public void updateDate(Long id,String timeZone){ - UpdateWrapper uw = new UpdateWrapper<>(); - uw.eq("id",id); - uw.set("update_date",DateUtil.getByTimeZone(timeZone)); - - baseMapper.update(null,uw); - } - - @Override - public Long exportSave(MultipartFile file, Long projectId, String module, Long designItemDetailId) { - AuthPrincipalVo userHolder = UserContext.getUserHolder(); - if (Objects.isNull(projectId) && !module.equals("designItemDetail")){ - log.warn("用户 {} 保存 {} 模块的画布时, projectId为空", userHolder.getId(), module); - throw new BusinessException("project id cannot be empty", ResultEnum.PROMPT.getCode()); - } - - QueryWrapper qw = new QueryWrapper<>(); - List exportFiles = null; - if (Objects.nonNull(projectId)){ - qw.lambda().eq(ExportFile::getProjectId, projectId); - qw.lambda().eq(ExportFile::getModule, module); - exportFiles = exportFileMapper.selectList(qw); - } else if (Objects.nonNull(designItemDetailId)){ - DesignItemDetailCanvas designItemDetailCanvas = designItemDetailService.getDIDCByDesignItemDetailId(designItemDetailId); - if (Objects.nonNull(designItemDetailCanvas) && Objects.nonNull(designItemDetailCanvas.getExportFileId())){ - qw.lambda().eq(ExportFile::getId, designItemDetailCanvas.getExportFileId()); - qw.lambda().eq(ExportFile::getModule, module); - exportFiles = exportFileMapper.selectList(qw); - } - } -// String upload = minioUtil.upload("aida-users", userHolder.getId() + "/exportFile", file); - - ExportFile exportFile = null; - if (CollectionUtil.isNotEmpty(exportFiles)) { - exportFile = exportFiles.get(0); - // 更新画布信息时，只重新上传图片，不用修改数据库url - String url = exportFile.getUrl(); - String path = url.substring(url.indexOf("/") + 1); - minioUtil.upload("aida-users", path, file, null); -// exportFile.setUrl(upload); -// exportFileMapper.updateById(exportFile); - }else { - String upload = minioUtil.upload("aida-users", userHolder.getId() + "/exportFile", file); - exportFile = new ExportFile(); - exportFile.setProjectId(projectId); - exportFile.setModule(module); - exportFile.setUrl(upload); - exportFileMapper.insert(exportFile); - if (Objects.nonNull(designItemDetailId)){ - designItemDetailService.saveDesignItemDetailCanvas(designItemDetailId, exportFile.getId()); - } - } - return exportFile.getId(); - } - - @Override - @Transactional(rollbackFor = Exception.class) - public List toProduct(ToProductImageDTO toProductImageDTO) { - // 判断用户当前积分是否够本次生成消耗 - boolean fluxTask = !StringUtil.isNullOrEmpty(toProductImageDTO.getModelName()) - && toProductImageDTO.getModelName().equals("flux"); - CreditsEventsEnum creditsEventsEnum = fluxTask ? CreditsEventsEnum.TO_PRODUCT_IMAGE_FLUX : CreditsEventsEnum.TO_PRODUCT_IMAGE; - Boolean preDeduction = creditsService.creditsPreDeduction(creditsEventsEnum, toProductImageDTO.getToProductImageVOList().size()); - if (!preDeduction) { - throw new BusinessException("Your remaining credits are insufficient for this generation. Please recharge.", ResultEnum.WARNING.getCode()); - } - - AuthPrincipalVo userHolder = UserContext.getUserHolder(); - Long projectId = toProductImageDTO.getProjectId(); - UserLikeGroup userLikeGroup = getByProjectId(projectId); - Long userLikeGroupId = null; - ToProductImageRecord toProductImageRecord = new ToProductImageRecord(); - toProductImageRecord.setProjectId(projectId); - if (Objects.nonNull(userLikeGroup)) { - userLikeGroupId = userLikeGroup.getId(); - toProductImageRecord.setUserLikeGroupId(userLikeGroupId); - } - - // 翻译 - String prompt = toProductImageDTO.getPrompt(); - StringBuilder sb = new StringBuilder("The best quality, masterpiece, real image."); - if (!StringUtil.isNullOrEmpty(prompt)) { - prompt = pythonService.promptTranslate(prompt); - - String[] words = prompt.split("\\s+"); - if (words.length > 180) { - throw new BusinessException("Please keep your input text under 200 words. Thanks!"); - } - } - - toProductImageRecord.setCreateTime(LocalDateTime.now()); - if (!StringUtils.isEmpty(toProductImageDTO.getPrompt())) { - toProductImageRecord.setPrompt(toProductImageDTO.getPrompt()); - } - toProductImageRecordMapper.insert(toProductImageRecord); - - List result = new ArrayList<>(); - boolean childFlag = !StringUtil.isNullOrEmpty(toProductImageDTO.getAgeGroup()) - && toProductImageDTO.getAgeGroup().equals("Child"); - int i = 0; - -// else { -// s = "best quality, masterpiece. detailed, high-res, simple background, studio photography, extremely detailed, updo, detailed face, face, close-up, HDR, UHD, 8K realistic, Highly detailed, simple background, Studio lighting"; -// } - for (ToProductImageVO toProductImageVO : toProductImageDTO.getToProductImageVOList()) { - String taskId; - if (toProductImageVO.getElementType().equals("DesignOutfit")) { - taskId = UUID.randomUUID() + "-" + i + "-" + userHolder.getId(); - TDesignPythonOutfit tDesignPythonOutfit = designPythonOutfitMapper.selectById(toProductImageVO.getElementId()); - - Long designItemId = tDesignPythonOutfit.getDesignItemId(); - QueryWrapper designItemDetailQueryWrapper = new QueryWrapper<>(); - designItemDetailQueryWrapper.lambda().eq(DesignItemDetail::getDesignItemId, designItemId); - designItemDetailQueryWrapper.lambda().ne(DesignItemDetail::getType, "Body"); - List designItemDetails = designItemDetailService.list(designItemDetailQueryWrapper); - String collect = designItemDetails.stream().map(DesignItemDetail::getType).collect(Collectors.joining(",")); - - Long designId = tDesignPythonOutfit.getDesignId(); - Design design = designMapper.selectById(designId); - String productType = "overall"; - if (design.getSingleOverall().equals("single")) { - productType = "single"; - sb.append(collect); - }else { - if (collect.contains("Tops") && childFlag) { - sb.append("a handsome boy,"); - }else if (collect.contains("Tops") && !childFlag){ - sb.append("a handsome man,"); - }else if (childFlag) { - sb.append("a beautiful girl,"); - }else { - sb.append("a beautiful woman,"); - } - sb.append("wearing ").append(collect); - } - if (StringUtils.isEmpty(prompt)) { - sb.append(",high quality clothing details,8K realistic,HDR"); - }else { - sb.append(",high quality clothing details,").append(prompt).append(",8K realistic,HDR"); - } - ToProductImageResultVO toProductImageResult = new ToProductImageResultVO(); - if (fluxTask){ - if (childFlag){ - sb.append(", Children's face"); - } - taskId = generateService.flux(CreditsEventsEnum.TO_PRODUCT_IMAGE_FLUX, sb.toString(), tDesignPythonOutfit.getDesignUrl(), childFlag); - toProductImageResult.setModelName("flux"); - toProductImageResult.setResultType(CollectionType.TO_PRODUCT_IMAGE.getValue()); - } else { - // 走模型 - if (childFlag){ - sb.append(", (Children's face:1.3)"); - } - pythonService.toProductImage(tDesignPythonOutfit.getDesignUrl(), taskId, sb.toString(), toProductImageDTO.getImageStrength(), productType); -// toProductImageResult.setModelName("local"); - } - - toProductImageResult.setElementId(tDesignPythonOutfit.getId()); - toProductImageResult.setElementType("DesignOutfit"); - toProductImageResult.setCreateTime(LocalDateTime.now()); - toProductImageResult.setToProductImageRecordId(toProductImageRecord.getId()); -// toProductImageResult.setUrl(productImageUrl); - toProductImageResult.setIsLike(0); - toProductImageResult.setTaskId(taskId); - toProductImageResult.setProjectId(projectId); - if (userLikeGroupId != null) { - toProductImageResult.setUserLikeGroupId(userLikeGroupId); - } - toProductImageResult.setImageStrength(toProductImageDTO.getImageStrength()); - toProductImageResult.setResultType(CollectionType.TO_PRODUCT_IMAGE.getValue()); - toProductImageResult.setStatus("Pending"); - toProductImageResultMapper.insert(toProductImageResult); -// toProductImageResult.setUrl(minioUtil.getPresignedUrl(toProductImageResult.getUrl(), 24 * 60)); - // 先判断是否需要默认like - if (Objects.nonNull(toProductImageDTO.getIsDefaultLike()) && toProductImageDTO.getIsDefaultLike()){ - // 满足条件情况下默认添加到like - CollectionSort collectionSort = addToProductLike(toProductImageVO.getParentId(), toProductImageResult.getId(), toProductImageDTO.getProjectId()); - // 重新排序 - Integer reSort = collectionSortService.rearrangeChildSort(toProductImageResult.getId(), CollectionType.TO_PRODUCT_IMAGE.getValue(), - toProductImageVO.getParentId(), toProductImageVO.getUserLikeSortId()); - // 将生成结果的排序返回 - toProductImageResult.setSort(Objects.isNull(reSort) ? Objects.isNull(collectionSort) ? null : collectionSort.getSort() : reSort); - toProductImageResult.setParentId(toProductImageVO.getParentId()); - toProductImageResult.setUserLikeSortId(Objects.isNull(collectionSort) ? null : collectionSort.getId()); - }else if (Objects.nonNull(toProductImageDTO.getIsDefaultLike()) && Objects.nonNull(toProductImageVO.getParentId())){ - toProductImageResult.setParentId(toProductImageVO.getParentId()); - // 默认不添加到like,但是需要有parentId,所以这里添加到collectionSort表中 -// designService.addCollectionSort(toProductImageResult.getId(), CollectionType.TO_PRODUCT_IMAGE.getValue(), toProductImageDTO.getProjectId(), toProductImageVO.getParentId()); - } - result.add(toProductImageResult); - }else { - if (StringUtils.isEmpty(prompt)) { - sb.append(",high quality clothing details,8K realistic,HDR"); - }else { - sb.append(",high quality clothing details,").append(prompt).append(",8K realistic,HDR"); - } - - taskId = UUID.randomUUID() + "-" + i + "-" + userHolder.getId(); - ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageVO.getElementId()); - ToProductImageResultVO toProductImageResult = new ToProductImageResultVO(); - - if (fluxTask){ - taskId = generateService.flux(creditsEventsEnum, sb.toString(), toProductElement.getUrl(), childFlag); - toProductImageResult.setModelName("flux"); - toProductImageResult.setResultType(CollectionType.TO_PRODUCT_IMAGE.getValue()); - } else { - // 走模型 - if (childFlag){ - sb.append(", (Children's face:1.3)"); - } - // 走模型 - pythonService.toProductImage(toProductElement.getUrl(), taskId, sb.toString(), toProductImageDTO.getImageStrength(), "overall"); -// toProductImageResult.setModelName("local"); - } - toProductImageResult.setElementId(toProductElement.getId()); - toProductImageResult.setElementType("ProductElement"); - toProductImageResult.setCreateTime(LocalDateTime.now()); - toProductImageResult.setToProductImageRecordId(toProductImageRecord.getId()); -// toProductImageResult.setUrl(productImageUrl); - toProductImageResult.setIsLike(0); - toProductImageResult.setTaskId(taskId); - toProductImageResult.setProjectId(projectId); - if (userLikeGroupId != null) { - toProductImageResult.setUserLikeGroupId(userLikeGroupId); - } - toProductImageResult.setImageStrength(toProductImageDTO.getImageStrength()); - toProductImageResult.setResultType(CollectionType.TO_PRODUCT_IMAGE.getValue()); - toProductImageResult.setStatus("Pending"); - toProductImageResultMapper.insert(toProductImageResult); -// toProductImageResult.setUrl(minioUtil.getPresignedUrl(toProductImageResult.getUrl(), 24 * 60)); - // 先判断是否需要默认like - if (Objects.nonNull(toProductImageDTO.getIsDefaultLike()) && toProductImageDTO.getIsDefaultLike()){ - // 满足条件情况下默认添加到like - CollectionSort collectionSort = addToProductLike(toProductImageVO.getParentId(), toProductImageResult.getId(), toProductImageDTO.getProjectId()); - // 重新排序 - Integer reSort = collectionSortService.rearrangeChildSort(toProductImageResult.getId(), CollectionType.TO_PRODUCT_IMAGE.getValue(), - toProductImageVO.getParentId(), toProductImageVO.getUserLikeSortId()); - // 将生成结果的排序返回 - toProductImageResult.setSort(Objects.isNull(reSort) ? Objects.isNull(collectionSort) ? null : collectionSort.getSort() : reSort); - toProductImageResult.setParentId(toProductImageVO.getParentId()); - toProductImageResult.setUserLikeSortId(Objects.isNull(collectionSort) ? null : collectionSort.getId()); - }else if (Objects.nonNull(toProductImageDTO.getIsDefaultLike()) && Objects.nonNull(toProductImageVO.getParentId())){ - toProductImageResult.setParentId(toProductImageVO.getParentId()); - // 默认不添加到like,但是需要有parentId,所以这里添加到collectionSort表中 -// designService.addCollectionSort(toProductImageResult.getId(), CollectionType.TO_PRODUCT_IMAGE.getValue(), toProductImageDTO.getProjectId(), toProductImageVO.getParentId()); - } - result.add(toProductImageResult); - } - i ++; - sb = new StringBuilder("The best quality, masterpiece, real image."); - // 添加需要扣除的积分到预扣除区 - creditsService.addRecordToCreditsDeduction(userHolder.getId(), taskId, creditsEventsEnum); - // 添加积分扣除记录到db - creditsService.preInsert(userHolder.getId(), creditsEventsEnum.getName(), taskId, Boolean.TRUE, null); - } - // 更新项目更新时间 - projectService.modifyProjectUpdateTime(toProductImageDTO.getProjectId()); - return result; - } - - private CollectionSort addToProductLike(Long parentId, Long toProductImageResultId, Long projectId){ - if (Objects.nonNull(parentId) && !parentId.equals(0L)){ - return productImageLike(new ProductImageLikeDTO(Collections.singletonList(toProductImageResultId), - projectId, parentId)); - } - return null; - } - - @Override - @Transactional(rollbackFor = Exception.class) - public void toProduct(String taskId) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(ToProductImageResult::getTaskId, taskId); - List toProductImageResults = toProductImageResultMapper.selectList(qw); - if (CollectionUtil.isNotEmpty(toProductImageResults)) { - ToProductImageResult toProductImageResultOne = toProductImageResults.get(0); - ToProductImageRecord toProductImageRecord = toProductImageRecordMapper.selectById(toProductImageResultOne.getToProductImageRecordId()); - String prompt = toProductImageRecord.getPrompt(); - StringBuilder sb = new StringBuilder("The best quality, masterpiece, real image."); - if (!StringUtil.isNullOrEmpty(prompt)) { - prompt = pythonService.promptTranslate(prompt); - } - String elementType = toProductImageResultOne.getElementType(); - if (elementType.equals("DesignOutfit")) { - TDesignPythonOutfit tDesignPythonOutfit = designPythonOutfitMapper.selectById(toProductImageResultOne.getElementId()); - Long designItemId = tDesignPythonOutfit.getDesignItemId(); - QueryWrapper designItemDetailQueryWrapper = new QueryWrapper<>(); - designItemDetailQueryWrapper.lambda().eq(DesignItemDetail::getDesignItemId, designItemId); - designItemDetailQueryWrapper.lambda().ne(DesignItemDetail::getType, "Body"); - List designItemDetails = designItemDetailService.list(designItemDetailQueryWrapper); - String collect = designItemDetails.stream().map(DesignItemDetail::getType).collect(Collectors.joining(",")); - - Long designId = tDesignPythonOutfit.getDesignId(); - Design design = designMapper.selectById(designId); - String productType = "overall"; - if (design.getSingleOverall().equals("single")) { - productType = "single"; - sb.append(collect); - }else { - if (collect.contains("Tops")) { - sb.append("a handsome man,"); - }else { - sb.append("a beautiful women,"); - } - sb.append("wearing ").append(collect); - } - if (StringUtils.isEmpty(prompt)) { - sb.append(",high quality clothing details,8K realistic,HDR"); - }else { - sb.append(",high quality clothing details,").append(prompt).append(",8K realistic,HDR"); - } - // 走模型 - pythonService.toProductImage(tDesignPythonOutfit.getDesignUrl(), taskId, sb.toString(), toProductImageResultOne.getImageStrength(), "single"); - }else { - if (StringUtils.isEmpty(prompt)) { - sb.append(",high quality clothing details,8K realistic,HDR"); - }else { - sb.append(",high quality clothing details,").append(prompt).append(",8K realistic,HDR"); - } - - ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageResultOne.getElementId()); - // 走模型 - pythonService.toProductImage(toProductElement.getUrl(), taskId, sb.toString(), toProductImageResultOne.getImageStrength(), "single"); - } - - } - } - - @Override - @Transactional(rollbackFor = Exception.class) - public ToProductElementVO toProductImageElementUpload(MultipartFile file, Long projectId) { - if (null == file || StringUtils.isEmpty(file.getOriginalFilename())) { - throw new BusinessException("file.cannot.be.empty"); - } -// if (projectId == null) { -// throw new BusinessException("projectId.cannot.be.empty"); -// } - AuthPrincipalVo userHolder = UserContext.getUserHolder(); - String bucketName = "aida-users"; - String minioUrl = minioUtil.upload(bucketName, userHolder.getId() + "/toProductImageElement", file); - ToProductElement toProductElement = new ToProductElement(); - toProductElement.setUrl(minioUrl); -// toProductElement.setUserLikeGroupId(userLikeGroupId); - toProductElement.setProjectId(projectId); - toProductElement.setCreateTime(LocalDateTime.now()); -// toProductElement.setType(type); - toProductElementMapper.insert(toProductElement); - ToProductElementVO toProductElementVO = CopyUtil.copyObject(toProductElement, ToProductElementVO.class); - toProductElementVO.setMinioUrl(toProductElementVO.getUrl()); - toProductElementVO.setUrl(minioUtil.getPreSignedUrl(toProductElementVO.getUrl(), 24 * 60)); - return toProductElementVO; - } - - @Override - public CollectionSort productImageLike(ProductImageLikeDTO productImageLikeDTO) { - List toProductImageResultId = productImageLikeDTO.getToProductImageResultId(); - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().in(ToProductImageResult::getId, toProductImageResultId); - ToProductImageResult toProductImageResult = new ToProductImageResult(); - toProductImageResult.setIsLike(1); - toProductImageResultMapper.update(toProductImageResult, qw); - ToProductImageResult toProductImageResult1 = toProductImageResultMapper.selectById(toProductImageResultId.get(0)); - CollectionSort collectionSort = null; - if (toProductImageResult1.getResultType().equals("Relight")) { - if (null != productImageLikeDTO.getCollectionSortParentId()) { - collectionSort = collectionSortService.addCollectionSort(toProductImageResult1.getId(), CollectionType.RELIGHT.getValue(), productImageLikeDTO.getProjectId(), productImageLikeDTO.getCollectionSortParentId()); - } - }else { - if (null != productImageLikeDTO.getCollectionSortParentId()) { - collectionSort = collectionSortService.addCollectionSort(toProductImageResult1.getId(), CollectionType.TO_PRODUCT_IMAGE.getValue(), productImageLikeDTO.getProjectId(), productImageLikeDTO.getCollectionSortParentId()); - } - } - // 更新项目更新时间 - projectService.modifyProjectUpdateTime(productImageLikeDTO.getProjectId()); - return collectionSort; - } - @Resource - private RedisUtil redisUtil; - @Value("${redis.key.toProductImageResultKey}") - private String toProductImageResultKey; - @Value("${redis.key.relightResultKey}") - private String relightResultKey; - @Override - public List getToProductImageResultList(List taskIdList) { - List results = new ArrayList<>(); - Set collect = new HashSet<>(); - taskIdList.forEach(taskId -> { - // 查记录 - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(ToProductImageResult::getTaskId, taskId); - ToProductImageResult toProductImageResult = toProductImageResultMapper.selectOne(qw); - if (Objects.isNull(toProductImageResult)) { - throw new BusinessException("The source image does not exist."); - } - ToProductImageRecord toProductImageRecord = toProductImageRecordMapper.selectById(toProductImageResult.getToProductImageRecordId()); - if (Objects.isNull(toProductImageRecord)) { - throw new BusinessException("This task does not exist."); - } - // 判断当任务从哪个模型获取结果 - if (!StringUtil.isNullOrEmpty(toProductImageResult.getModelName()) && toProductImageResult.getModelName().equals("flux")){ - Project project = projectService.getById(toProductImageResult.getProjectId()); - if (Objects.isNull(project)){ - throw new BusinessException("unknown project"); - } - String fluxResult; - if (toProductImageResult.getStatus().equals("Success") && !StringUtil.isNullOrEmpty(toProductImageResult.getUrl())){ - fluxResult = toProductImageResult.getUrl(); - }else { - String objectName = project.getAccountId() + "/product_image/" + taskId + ".png"; - fluxResult = generateService.getFluxResult(taskId, objectName); - } - - if (StringUtil.isNullOrEmpty(fluxResult)){ - toProductImageResult.setStatus("Fail"); - toProductImageResultMapper.updateById(toProductImageResult); - results.add(new MagicToolResultVO(taskId, "Fail")); - } else if (fluxResult.equals("Fail") || fluxResult.equals("Pending")) { - toProductImageResult.setStatus(fluxResult); - toProductImageResultMapper.updateById(toProductImageResult); - results.add(new MagicToolResultVO(taskId, fluxResult)); - } else { - results.add(processFluxResult(fluxResult, toProductImageResult, taskId, toProductImageRecord.getPrompt())); - // 扣积分 - Boolean flag = creditsService.taskCreditsDeduction(project.getAccountId(), taskId); - if (flag) creditsService.updateChangedCredits(String.valueOf(project.getAccountId()), taskId); - } - // 在这个stream中不再继续往后执行 - return; - } - - String key = toProductImageResultKey + ":" + taskId; - MagicToolResultVO magicToolResultVO = new Gson().fromJson(redisUtil.getFromString(key), MagicToolResultVO.class); - if (!Objects.isNull(magicToolResultVO) && !StringUtil.isNullOrEmpty(magicToolResultVO.getUrl())) { - String url = magicToolResultVO.getUrl(); - if (url.substring(url.lastIndexOf("/") + 1).equals("white_image.jpg")) { - magicToolResultVO.setStatus("Invalid"); - } else { - magicToolResultVO.setUrl(minioUtil.getPreSignedUrl(url, CommonConstant.MINIO_IMAGE_EXPIRE_TIME)); - magicToolResultVO.setResultType(toProductImageResult.getResultType()); - magicToolResultVO.setElementId(toProductImageResult.getElementId()); - magicToolResultVO.setElementType(toProductImageResult.getElementType()); - magicToolResultVO.setDirection(toProductImageResult.getDirection()); - magicToolResultVO.setBrightenValue(toProductImageResult.getBrightenValue()); - magicToolResultVO.setImageStrength(toProductImageResult.getImageStrength()); - magicToolResultVO.setPrompt(toProductImageRecord.getPrompt()); - if (toProductImageResult.getElementType().equals("ProductElement")) { - ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageResult.getElementId()); - magicToolResultVO.setSourceUrl(minioUtil.getPreSignedUrl(toProductElement.getUrl(), 24 * 60)); - - Long parentId = collectionSortService.getParentIdByElementIdAndElementType(toProductImageResult.getElementId(), toProductImageResult.getElementType()); - magicToolResultVO.setParentId(parentId); - }else { - TDesignPythonOutfit tDesignPythonOutfit = designPythonOutfitMapper.selectById(toProductImageResult.getElementId()); - magicToolResultVO.setSourceUrl(minioUtil.getPreSignedUrl(tDesignPythonOutfit.getDesignUrl(), 24 * 60)); - - QueryWrapper userLikeQueryWrapper = new QueryWrapper<>(); - userLikeQueryWrapper.lambda().eq(UserLike::getDesignOutfitId, tDesignPythonOutfit.getId()); - List userLikeList = userLikeMapper.selectList(userLikeQueryWrapper); - if (!CollectionUtils.isEmpty(userLikeList)) { - UserLike userLike = userLikeList.get(0); - Long parentId = collectionSortService.getParentIdByElementIdAndElementType(userLike.getId(), toProductImageResult.getElementType()); - magicToolResultVO.setParentId(parentId); - } - } - } - } else if (Objects.isNull(magicToolResultVO)) { - magicToolResultVO = new MagicToolResultVO(taskId, "Executing"); - } - if (!StringUtil.isNullOrEmpty(magicToolResultVO.getStatus())) collect.add(magicToolResultVO.getStatus()); - results.add(magicToolResultVO); - }); - return results; - } - - private MagicToolResultVO processFluxResult(String fluxImgMinioPath, ToProductImageResult toProductImageResult, String taskId, String prompt){ - if (!toProductImageResult.getStatus().equals("Success") - && StringUtil.isNullOrEmpty(toProductImageResult.getUrl())){ - toProductImageResult.setStatus("Success"); - toProductImageResult.setUrl(fluxImgMinioPath); - toProductImageResultMapper.updateById(toProductImageResult); - } - - MagicToolResultVO magicToolResultVO = CopyUtil.copyObject(toProductImageResult, MagicToolResultVO.class); - magicToolResultVO.setTaskId(taskId); - magicToolResultVO.setId(toProductImageResult.getId()); - magicToolResultVO.setUrl(minioUtil.getPreSignedUrl(fluxImgMinioPath, CommonConstant.MINIO_IMAGE_EXPIRE_TIME)); - magicToolResultVO.setStatus("Success"); - magicToolResultVO.setResultType(toProductImageResult.getResultType()); - magicToolResultVO.setElementId(toProductImageResult.getElementId()); - magicToolResultVO.setElementType(toProductImageResult.getElementType()); - magicToolResultVO.setPrompt(prompt); - - if (toProductImageResult.getElementType().equals("ProductElement")) { - ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageResult.getElementId()); - magicToolResultVO.setSourceUrl(minioUtil.getPreSignedUrl(toProductElement.getUrl(), 24 * 60)); - - Long parentId = collectionSortService.getParentIdByElementIdAndElementType(toProductImageResult.getElementId(), toProductImageResult.getElementType()); - magicToolResultVO.setParentId(parentId); - } else if (toProductImageResult.getElementType().equals("DesignOutfit")){ - TDesignPythonOutfit tDesignPythonOutfit = designPythonOutfitMapper.selectById(toProductImageResult.getElementId()); - magicToolResultVO.setSourceUrl(minioUtil.getPreSignedUrl(tDesignPythonOutfit.getDesignUrl(), 24 * 60)); - - QueryWrapper userLikeQueryWrapper = new QueryWrapper<>(); - userLikeQueryWrapper.lambda().eq(UserLike::getDesignOutfitId, tDesignPythonOutfit.getId()); - List userLikeList = userLikeMapper.selectList(userLikeQueryWrapper); - if (!CollectionUtils.isEmpty(userLikeList)) { - UserLike userLike = userLikeList.get(0); - Long parentId = collectionSortService.getParentIdByElementIdAndElementType(userLike.getId(), toProductImageResult.getElementType()); - magicToolResultVO.setParentId(parentId); - } - } else if (toProductImageResult.getElementType().equals("ToProductImage")){ - ToProductImageResult toProductImage = toProductImageResultMapper.selectById(toProductImageResult.getElementId()); - magicToolResultVO.setSourceUrl(minioUtil.getPreSignedUrl(toProductImage.getUrl(), 24 * 60)); - - Long parentId = collectionSortService.getParentIdByElementIdAndElementType(toProductImageResult.getElementId(), toProductImageResult.getElementType()); - magicToolResultVO.setParentId(parentId); - } - -// Long parentId = collectionSortService.getParentIdByElementIdAndElementType(toProductImageResult.getId(), toProductImageResult.getResultType()); -// magicToolResultVO.setParentId(parentId); - return magicToolResultVO; - } - - @Override - public JSONObject exportSearch(ExportSearchDTO exportSearchDTO) { - QueryWrapper qw = new QueryWrapper<>(); - if (Objects.nonNull(exportSearchDTO.getProjectId())){ - qw.lambda().eq(ExportFile::getProjectId, exportSearchDTO.getProjectId()); - } else if (Objects.nonNull(exportSearchDTO.getId())){ - qw.lambda().eq(ExportFile::getId, exportSearchDTO.getId()); - } - qw.lambda().eq(ExportFile::getModule, exportSearchDTO.getModule()); - List exportFiles = exportFileMapper.selectList(qw); - if (CollectionUtil.isNotEmpty(exportFiles)) { - try { - InputStream download = minioUtil.download(exportFiles.get(0).getUrl()); - String convert = convert(download); - JSONObject jsonObject = JSONObject.parseObject(convert); - - JSONObject canvas = jsonObject.getJSONObject("canvas"); - if (Objects.nonNull(canvas)) { - JSONArray objectJSONArrayLevel1 = canvas.getJSONArray("objects"); - for (int i = 0; i < objectJSONArrayLevel1.size(); i++) { - JSONObject jsonObjectLevel1 = objectJSONArrayLevel1.getJSONObject(i); - String type = jsonObjectLevel1.getString("type"); - if (type.equals("image")) { - String minioUrl = jsonObjectLevel1.getString("minioUrl"); - if (!StringUtil.isNullOrEmpty(minioUrl)) jsonObjectLevel1.put("src", minioUtil.getPreSignedUrl(minioUrl, 24 * 60)); - } - if (type.equals("group")) { - JSONArray objectJSONArrayLevel2 = jsonObjectLevel1.getJSONArray("objects"); - for (int j = 0; j < objectJSONArrayLevel2.size(); j++) { - JSONObject jsonObjectLevel2 = objectJSONArrayLevel2.getJSONObject(j); - String typeGroup = jsonObjectLevel2.getString("type"); - if (typeGroup.equals("image")) { - String minioUrl = jsonObjectLevel2.getString("minioUrl"); - jsonObjectLevel2.put("src", minioUtil.getPreSignedUrl(minioUrl, 24 * 60)); - } - objectJSONArrayLevel2.set(j, jsonObjectLevel2); - } - jsonObjectLevel1.put("objects", objectJSONArrayLevel2); - } - /*if (type.equals("rect")) { - JSONObject custom = jsonObjectLevel1.getJSONObject("custom"); - if (custom.getString("type").equals("init")) { - String minioUrl = jsonObjectLevel1.getString("minioUrl"); - if (!StringUtils.isEmpty(minioUrl)) { - JSONObject fill = jsonObjectLevel1.getJSONObject("fill"); - fill.put("source", minioUtil.getPreSignedUrl(minioUrl, 24 * 60)); - jsonObjectLevel1.put("fill", fill); - } - } - }*/ - objectJSONArrayLevel1.set(i, jsonObjectLevel1); - } - canvas.put("objects", objectJSONArrayLevel1); - jsonObject.put("canvas", canvas); - }else { - JSONArray objectJSONArrayLevel1 = jsonObject.getJSONArray("objects"); - for (int i = 0; i < objectJSONArrayLevel1.size(); i++) { - JSONObject jsonObjectLevel1 = objectJSONArrayLevel1.getJSONObject(i); - String type = jsonObjectLevel1.getString("type"); - if (type.equals("image")) { - String minioUrl = jsonObjectLevel1.getString("minioUrl"); - if (!StringUtil.isNullOrEmpty(minioUrl)) jsonObjectLevel1.put("src", minioUtil.getPreSignedUrl(minioUrl, 24 * 60)); - } - if (type.equals("group")) { - JSONArray objectJSONArrayLevel2 = jsonObjectLevel1.getJSONArray("objects"); - for (int j = 0; j < objectJSONArrayLevel2.size(); j++) { - JSONObject jsonObjectLevel2 = objectJSONArrayLevel2.getJSONObject(j); - String typeGroup = jsonObjectLevel2.getString("type"); - if (typeGroup.equals("image")) { - String minioUrl = jsonObjectLevel2.getString("minioUrl"); - jsonObjectLevel2.put("src", minioUtil.getPreSignedUrl(minioUrl, 24 * 60)); - } - objectJSONArrayLevel2.set(j, jsonObjectLevel2); - } - jsonObjectLevel1.put("objects", objectJSONArrayLevel2); - } - /*if (type.equals("rect")) { - JSONObject custom = jsonObjectLevel1.getJSONObject("custom"); - if (custom.getString("type").equals("init")) { - String minioUrl = jsonObjectLevel1.getString("minioUrl"); - if (!StringUtils.isEmpty(minioUrl)) { - JSONObject fill = jsonObjectLevel1.getJSONObject("fill"); - fill.put("source", minioUtil.getPreSignedUrl(minioUrl, 24 * 60)); - jsonObjectLevel1.put("fill", fill); - } - } - }*/ - objectJSONArrayLevel1.set(i, jsonObjectLevel1); - } - jsonObject.put("objects", objectJSONArrayLevel1); - } - return jsonObject; - }catch (Exception e){ - log.error(e.getMessage()); - return null; - } - }else { - return null; - } - } - - @Override - public CanvasElementUpload canvasElementUpload(MultipartFile file) { - AuthPrincipalVo userHolder = UserContext.getUserHolder(); - CanvasElementUpload canvasElementUpload = new CanvasElementUpload(); - String url = minioUtil.upload("aida-users", userHolder.getId() + "/canvasElementUpload", file); - canvasElementUpload.setUrl(url); - canvasElementUpload.setMinioUrl(minioUtil.getPreSignedUrl(url, 24 * 60)); - return canvasElementUpload; - } - - @Override - public List productImageLikeList(ToProductImageDTO toProductImageDTO) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(ToProductImageResult::getIsLike, 1); - - Long projectId = toProductImageDTO.getProjectId(); -// UserLikeGroup userLikeGroup = getByProjectId(projectId); -// Long userLikeGroupId = null; - - qw.lambda().eq(ToProductImageResult::getProjectId, projectId); - List toProductImageResults = toProductImageResultMapper.selectList(qw); - for (ToProductImageResult toProductImageResult : toProductImageResults) { - toProductImageResult.setUrl(minioUtil.getPreSignedUrl(toProductImageResult.getUrl(), 24 * 60)); - } - List toProductImageResultVOS = CopyUtil.copyList(toProductImageResults, ToProductImageResultVO.class); - for (ToProductImageResultVO toProductImageResultVO : toProductImageResultVOS) { - if (toProductImageResultVO.getElementType().equals("ProductElement")) { - ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageResultVO.getElementId()); - toProductImageResultVO.setSourceUrl(minioUtil.getPreSignedUrl(toProductElement.getUrl(), 24 * 60)); - }else if ((toProductImageResultVO.getElementType().equals("DesignOutfit"))) { - TDesignPythonOutfit tDesignPythonOutfit = designPythonOutfitMapper.selectById(toProductImageResultVO.getElementId()); - toProductImageResultVO.setSourceUrl(minioUtil.getPreSignedUrl(tDesignPythonOutfit.getDesignUrl(), 24 * 60)); - }else { - ToProductImageResult toProductImageResult1 = toProductImageResultMapper.selectById(toProductImageResultVO.getElementId()); - toProductImageResultVO.setSourceUrl(minioUtil.getPreSignedUrl(toProductImageResult1.getUrl(), 24 * 60)); - } - } - // 更新项目更新时间 - projectService.modifyProjectUpdateTime(toProductImageDTO.getProjectId()); - return toProductImageResultVOS; - } - - @Override - public Boolean productImageUnLike(ProductImageLikeDTO productImageLikeDTO) { - List toProductImageResultId = productImageLikeDTO.getToProductImageResultId(); - if (!CollectionUtils.isEmpty(toProductImageResultId)) { - Long toProductImageId = toProductImageResultId.get(0); - ToProductImageResult toProductImageResult = toProductImageResultMapper.selectById(toProductImageId); - toProductImageResult.setIsLike(0); - toProductImageResultMapper.updateById(toProductImageResult); - if (toProductImageResult.getResultType().equals(CollectionType.RELIGHT.getValue())) { - if (null != productImageLikeDTO.getCollectionSortParentId()) { - collectionSortService.deleteCollectionSort(toProductImageResult.getId(), CollectionType.RELIGHT.getValue(), productImageLikeDTO.getProjectId(), productImageLikeDTO.getCollectionSortParentId()); - } - }else { - if (null != productImageLikeDTO.getCollectionSortParentId()) { - collectionSortService.deleteCollectionSort(toProductImageResult.getId(), CollectionType.TO_PRODUCT_IMAGE.getValue(), productImageLikeDTO.getProjectId(), productImageLikeDTO.getCollectionSortParentId()); - } - } - } - // 更新项目更新时间 - projectService.modifyProjectUpdateTime(productImageLikeDTO.getProjectId()); - return Boolean.TRUE; - } - - @Override - @Transactional(rollbackFor = Exception.class) - public void relight(String taskId) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(ToProductImageResult::getTaskId, taskId); - List toProductImageResults = toProductImageResultMapper.selectList(qw); - if (CollectionUtil.isNotEmpty(toProductImageResults)) { - ToProductImageResult toProductImageResultOne = toProductImageResults.get(0); - ToProductImageRecord toProductImageRecord = toProductImageRecordMapper.selectById(toProductImageResultOne.getToProductImageRecordId()); - String prompt = toProductImageRecord.getPrompt(); - - // 翻译 - String s = ""; - if (!StringUtil.isNullOrEmpty(prompt)) { - s = pythonService.promptTranslate(prompt); - }else { - s = "Snow moutain, snowy day, natural light"; - } - String elementType = toProductImageResultOne.getElementType(); - if (elementType.equals("ToProductImage")) { - ToProductImageResult toProductImageResult1 = toProductImageResultMapper.selectById(toProductImageResultOne.getElementId()); - String relightType = "overall"; - if (toProductImageResult1.getElementType().equals("DesignOutfit")) { - TDesignPythonOutfit tDesignPythonOutfit = designPythonOutfitMapper.selectById(toProductImageResult1.getElementId()); - Long designId = tDesignPythonOutfit.getDesignId(); - Design design = designMapper.selectById(designId); - if (design.getSingleOverall().equals("single")) { - relightType = "single"; - } - } - - // 走模型 - pythonService.relight(toProductImageResult1.getUrl(), taskId, s, toProductImageResultOne.getDirection(), "single"); - }else { - ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageResultOne.getElementId()); - // 走模型 - pythonService.relight(toProductElement.getUrl(), taskId, s, toProductImageResultOne.getDirection(), "single"); - } - - } - } - - @Override - @Transactional(rollbackFor = Exception.class) - public List relight(ToProductImageDTO toProductImageDTO) { - // 判断用户当前积分是否够本次生成消耗 - boolean fluxTask = !StringUtil.isNullOrEmpty(toProductImageDTO.getModelName()) - && toProductImageDTO.getModelName().equals("flux"); - CreditsEventsEnum creditsEventsEnum = fluxTask ? CreditsEventsEnum.RELIGHT_FLUX : CreditsEventsEnum.RELIGHT; - Boolean preDeduction = creditsService.creditsPreDeduction(creditsEventsEnum, toProductImageDTO.getToProductImageVOList().size()); - if (!preDeduction) { - throw new BusinessException("Your remaining credits are insufficient for this generation. Please recharge.", ResultEnum.WARNING.getCode()); - } - - AuthPrincipalVo userHolder = UserContext.getUserHolder(); - Long projectId = toProductImageDTO.getProjectId(); - UserLikeGroup userLikeGroup = getByProjectId(projectId); - Long userLikeGroupId = null; - - ToProductImageRecord toProductImageRecord = new ToProductImageRecord(); - toProductImageRecord.setProjectId(projectId); - if (Objects.nonNull(userLikeGroup)) { - userLikeGroupId = userLikeGroup.getId(); - toProductImageRecord.setUserLikeGroupId(userLikeGroupId); - } - - // 翻译 - String prompt = toProductImageDTO.getPrompt(); - String s = ""; - if (!StringUtil.isNullOrEmpty(prompt)) { - s = pythonService.promptTranslate(prompt); - }else { -// s = "Snow moutain, snowy day, natural light"; - s = "Street, natural light"; - } - // 对输入文本进行长度限制 - String[] words = prompt.split("\\s+"); - if (words.length > 200) { - throw new BusinessException("Please keep your input text under 200 words. Thanks!"); - } - - toProductImageRecord.setCreateTime(LocalDateTime.now()); - if (!StringUtils.isEmpty(toProductImageDTO.getPrompt())) { - toProductImageRecord.setPrompt(toProductImageDTO.getPrompt()); - } - toProductImageRecordMapper.insert(toProductImageRecord); - - List result = new ArrayList<>(); - - int i = 0; - - for (ToProductImageVO toProductImageVO : toProductImageDTO.getToProductImageVOList()) { - String taskId = UUID.randomUUID() + "-" + i + "-" + userHolder.getId(); - i ++; - if (toProductImageVO.getElementType().equals("ToProductImage")) { - ToProductImageResult toProductImageResult1 = toProductImageResultMapper.selectById(toProductImageVO.getElementId()); - String relightType = "overall"; - if (toProductImageResult1.getElementType().equals("DesignOutfit")) { - TDesignPythonOutfit tDesignPythonOutfit = designPythonOutfitMapper.selectById(toProductImageResult1.getElementId()); - Long designId = tDesignPythonOutfit.getDesignId(); - Design design = designMapper.selectById(designId); - if (design.getSingleOverall().equals("single")) { - relightType = "single"; - } - } - - ToProductImageResultVO toProductImageResult = new ToProductImageResultVO(); - if (fluxTask){ - taskId = generateService.flux(creditsEventsEnum, s, toProductImageResult1.getUrl(), false); - toProductImageResult.setModelName("flux"); - toProductImageResult.setResultType(CollectionType.RELIGHT.getValue()); - } else { - // 走模型 - pythonService.relight(toProductImageResult1.getUrl(), taskId, s, toProductImageDTO.getDirection(), relightType); - } - toProductImageResult.setElementId(toProductImageResult1.getId()); - toProductImageResult.setElementType("ToProductImage"); - toProductImageResult.setCreateTime(LocalDateTime.now()); - toProductImageResult.setToProductImageRecordId(toProductImageRecord.getId()); -// toProductImageResult.setUrl(productImageUrl); - toProductImageResult.setIsLike(0); - toProductImageResult.setTaskId(taskId); - toProductImageResult.setProjectId(projectId); - if (null != userLikeGroupId) { - toProductImageResult.setUserLikeGroupId(userLikeGroupId); - } - if (toProductImageDTO.getBrightenValue() != null) { - toProductImageResult.setBrightenValue(toProductImageDTO.getBrightenValue()); - } - toProductImageResult.setDirection(toProductImageDTO.getDirection()); - toProductImageResult.setResultType(CollectionType.RELIGHT.getValue()); - toProductImageResult.setStatus("Pending"); - toProductImageResultMapper.insert(toProductImageResult); -// toProductImageResult.setUrl(minioUtil.getPresignedUrl(toProductImageResult.getUrl(), 24 * 60)); - // 先判断是否需要默认like - if (Objects.nonNull(toProductImageDTO.getIsDefaultLike()) && toProductImageDTO.getIsDefaultLike()){ - // 满足条件情况下默认添加到like - CollectionSort collectionSort = addToProductLike(toProductImageVO.getParentId(), toProductImageResult.getId(), toProductImageDTO.getProjectId()); - // 重新排序 - Integer reSort = collectionSortService.rearrangeChildSort(toProductImageResult.getId(), CollectionType.RELIGHT.getValue(), - toProductImageVO.getParentId(), toProductImageVO.getUserLikeSortId()); - // 将生成结果的排序返回 - toProductImageResult.setSort(Objects.isNull(reSort) ? Objects.isNull(collectionSort) ? null : collectionSort.getSort() : reSort); - toProductImageResult.setParentId(toProductImageVO.getParentId()); - toProductImageResult.setUserLikeSortId(Objects.isNull(collectionSort) ? null : collectionSort.getId()); - }else if (Objects.nonNull(toProductImageDTO.getIsDefaultLike()) && Objects.nonNull(toProductImageVO.getParentId())){ - toProductImageResult.setParentId(toProductImageVO.getParentId()); - // 默认不添加到like,但是需要有parentId,所以这里添加到collectionSort表中 -// designService.addCollectionSort(toProductImageResult.getId(), CollectionType.RELIGHT.getValue(), toProductImageDTO.getProjectId(), toProductImageVO.getParentId()); - } - result.add(toProductImageResult); - }else { - ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageVO.getElementId()); - ToProductImageResultVO toProductImageResult = new ToProductImageResultVO(); - if (fluxTask){ - taskId = generateService.flux(CreditsEventsEnum.RELIGHT_FLUX, s, toProductElement.getUrl(), false); - toProductImageResult.setModelName("flux"); - toProductImageResult.setResultType(CollectionType.RELIGHT.getValue()); - } else { - // 走模型 - pythonService.relight(toProductElement.getUrl(), taskId, s, toProductImageDTO.getDirection(), "overall"); - } - toProductImageResult.setElementId(toProductElement.getId()); - toProductImageResult.setElementType("ProductElement"); - toProductImageResult.setCreateTime(LocalDateTime.now()); - toProductImageResult.setToProductImageRecordId(toProductImageRecord.getId()); -// toProductImageResult.setUrl(productImageUrl); - toProductImageResult.setIsLike(0); - toProductImageResult.setTaskId(taskId); - toProductImageResult.setProjectId(projectId); - if (null != userLikeGroupId) { - toProductImageResult.setUserLikeGroupId(userLikeGroupId); - } - if (toProductImageDTO.getBrightenValue() != null) { - toProductImageResult.setBrightenValue(toProductImageDTO.getBrightenValue()); - } - toProductImageResult.setDirection(toProductImageDTO.getDirection()); - toProductImageResult.setResultType(CollectionType.RELIGHT.getValue()); - toProductImageResult.setStatus("Pending"); - toProductImageResultMapper.insert(toProductImageResult); -// toProductImageResult.setUrl(minioUtil.getPresignedUrl(toProductImageResult.getUrl(), 24 * 60)); - // 先判断是否需要默认like - if (Objects.nonNull(toProductImageDTO.getIsDefaultLike()) && toProductImageDTO.getIsDefaultLike()){ - // 满足条件情况下默认添加到like - CollectionSort collectionSort = addToProductLike(toProductImageVO.getParentId(), toProductImageResult.getId(), toProductImageDTO.getProjectId()); - // 重新排序 - Integer reSort = collectionSortService.rearrangeChildSort(toProductImageResult.getId(), CollectionType.RELIGHT.getValue(), - toProductImageVO.getParentId(), toProductImageVO.getUserLikeSortId()); - // 将生成结果的排序返回 - toProductImageResult.setSort(Objects.isNull(reSort) ? Objects.isNull(collectionSort) ? null : collectionSort.getSort() : reSort); - toProductImageResult.setParentId(toProductImageVO.getParentId()); - toProductImageResult.setUserLikeSortId(Objects.isNull(collectionSort) ? null : collectionSort.getId()); - }else if (Objects.nonNull(toProductImageDTO.getIsDefaultLike()) && Objects.nonNull(toProductImageVO.getParentId())){ - toProductImageResult.setParentId(toProductImageVO.getParentId()); - // 默认不添加到like,但是需要有parentId,所以这里添加到collectionSort表中 -// designService.addCollectionSort(toProductImageResult.getId(), CollectionType.RELIGHT.getValue(), toProductImageDTO.getProjectId(), toProductImageVO.getParentId()); - } - result.add(toProductImageResult); - } - // 添加需要扣除的积分到预扣除区 - creditsService.addRecordToCreditsDeduction(userHolder.getId(), taskId, creditsEventsEnum); - // 添加积分扣除记录到db - creditsService.preInsert(userHolder.getId(), creditsEventsEnum.getName(), taskId, Boolean.TRUE, null); - } - // 更新项目更新时间 - projectService.modifyProjectUpdateTime(toProductImageDTO.getProjectId()); - return result; - } - - @Override - public List getRelightResult(List taskIdList) { - List results = new ArrayList<>(); - Set collect = new HashSet<>(); - taskIdList.forEach(taskId -> { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(ToProductImageResult::getTaskId, taskId); - ToProductImageResult toProductImageResult = toProductImageResultMapper.selectOne(qw); - if (Objects.isNull(toProductImageResult)) { - throw new BusinessException("The source image does not exist."); - } - ToProductImageRecord toProductImageRecord = toProductImageRecordMapper.selectById(toProductImageResult.getToProductImageRecordId()); - if (Objects.isNull(toProductImageRecord)) { - throw new BusinessException("This task does not exist."); - } - // 判断当任务从哪个模型获取结果 - if (!StringUtil.isNullOrEmpty(toProductImageResult.getModelName()) - && toProductImageResult.getModelName().equals("flux")){ - Project project = projectService.getById(toProductImageResult.getProjectId()); - if (Objects.isNull(project)){ - throw new BusinessException("unknown project"); - } - String fluxResult; - if (toProductImageResult.getStatus().equals("Success") && !StringUtil.isNullOrEmpty(toProductImageResult.getUrl())){ - fluxResult = toProductImageResult.getUrl(); - }else { - String objectName = project.getAccountId() + "/relight_image/" + taskId + ".png"; - fluxResult = generateService.getFluxResult(taskId, objectName); - } - - if (StringUtil.isNullOrEmpty(fluxResult)){ - toProductImageResult.setStatus("Fail"); - toProductImageResultMapper.updateById(toProductImageResult); - results.add(new MagicToolResultVO(taskId, "Fail")); - } else if (fluxResult.equals("Fail") || fluxResult.equals("Pending")) { - toProductImageResult.setStatus(fluxResult); - toProductImageResultMapper.updateById(toProductImageResult); - results.add(new MagicToolResultVO(taskId, fluxResult)); - } else { - results.add(processFluxResult(fluxResult, toProductImageResult, taskId, toProductImageRecord.getPrompt())); - // 扣积分 - Boolean flag = creditsService.taskCreditsDeduction(project.getAccountId(), taskId); - if (flag) creditsService.updateChangedCredits(String.valueOf(project.getAccountId()), taskId); - } - return; - } - - String key = relightResultKey + ":" + taskId; - MagicToolResultVO magicToolResultVO = new Gson().fromJson(redisUtil.getFromString(key), MagicToolResultVO.class); - if (!Objects.isNull(magicToolResultVO) && !StringUtil.isNullOrEmpty(magicToolResultVO.getUrl())) { - String url = magicToolResultVO.getUrl(); - magicToolResultVO.setDirection(toProductImageResult.getDirection()); - magicToolResultVO.setBrightenValue(toProductImageResult.getBrightenValue()); - magicToolResultVO.setImageStrength(toProductImageResult.getImageStrength()); - magicToolResultVO.setPrompt(toProductImageRecord.getPrompt()); - if (url.substring(url.lastIndexOf("/") + 1).equals("white_image.jpg")) { - magicToolResultVO.setStatus("Invalid"); - } else { - magicToolResultVO.setUrl(minioUtil.getPreSignedUrl(url, CommonConstant.MINIO_IMAGE_EXPIRE_TIME)); - - magicToolResultVO.setResultType(toProductImageResult.getResultType()); - magicToolResultVO.setElementId(toProductImageResult.getElementId()); - magicToolResultVO.setElementType(toProductImageResult.getElementType()); - if (toProductImageResult.getElementType().equals("ProductElement")) { - ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageResult.getElementId()); - magicToolResultVO.setSourceUrl(minioUtil.getPreSignedUrl(toProductElement.getUrl(), 24 * 60)); - }else { - ToProductImageResult toProductImageResult1 = toProductImageResultMapper.selectById(toProductImageResult.getElementId()); - magicToolResultVO.setSourceUrl(minioUtil.getPreSignedUrl(toProductImageResult1.getUrl(), 24 * 60)); - } - Long parentId = collectionSortService.getParentIdByElementIdAndElementType(toProductImageResult.getElementId(), toProductImageResult.getElementType()); - magicToolResultVO.setParentId(parentId); - } - } else if (Objects.isNull(magicToolResultVO)) { - magicToolResultVO = new MagicToolResultVO(); - } - if (!StringUtil.isNullOrEmpty(magicToolResultVO.getStatus())) collect.add(magicToolResultVO.getStatus()); - results.add(magicToolResultVO); - }); - return results; - } - - public static String convert(InputStream inputStream) { - try (BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream))) { - return reader.lines().collect(Collectors.joining("\n")); - } catch (Exception e) { - e.printStackTrace(); - return null; - } - } - - @Transactional(rollbackFor = Exception.class) - public void deleteToProductRelightResult(Long id, Long projectId, String type) { - // 1. 权限校验和删除主数据 - Long accountId = UserContext.getUserHolder().getId(); - Project project = projectService.getById(projectId); - if (!project.getAccountId().equals(accountId)) { - throw new IllegalArgumentException("项目不属于当前账号"); - } - - // 2. 删除主表数据 - UpdateWrapper wrapper = new UpdateWrapper<>(); - wrapper.lambda().eq(ToProductImageResult::getProjectId, projectId) - .eq(ToProductImageResult::getId, id).or().eq(ToProductImageResult::getElementId, id) - .set(ToProductImageResult::getIsDeleted, 1); - - int update = toProductImageResultMapper.update(null, wrapper); - log.info("删除 {} 结果, id={}, 影响行数={}", type, id, update); - - if (update == 0) { - return; // 未删除任何数据，直接返回 - } - - // 3. 查询所有被删除的排序项（按sort升序） - List deletedItems = collectionSortMapper.selectList( - new QueryWrapper() - .eq("project_id", projectId) - .eq("relation_id", id) - .eq("relation_type", type) - .orderByAsc("sort") - ); - - if (deletedItems.isEmpty()) return; - - // 4. 删除这些记录 - int deletedCount = collectionSortMapper.delete( - new QueryWrapper() - .eq("project_id", projectId) - .eq("relation_id", id) - .eq("relation_type", type) - ); - log.info("删除排序记录：id={}, type={}, 影响行数={}", id, type, deletedCount); - - // 5. 对每个被删除的sort值，单独调整后续记录 - for (CollectionSort deletedItem : deletedItems) { - collectionSortMapper.update( - null, - new UpdateWrapper() - .eq("project_id", projectId) - .gt("sort", deletedItem.getSort()) - .setSql("sort = sort - 1") // 每条SQL只减1 - ); - } - } - - @Override - public String likeHistoryRelSketch() { - QueryWrapper qw = new QueryWrapper<>(); - List userLikeGroups = userLikeGroupMapper.selectList(qw); - List>> result = new ArrayList<>(); - - for (UserLikeGroup userLikeGroup : userLikeGroups) { - Long accountId = userLikeGroup.getAccountId(); - Long collectionId = userLikeGroup.getCollectionId(); - - // 提前转换为Set以提高contains的效率 - List collectionElements = getCollectionElementsByCollectionId(collectionId); - Set urlSet = collectionElements.stream() - .map(CollectionElement::getUrl) - .collect(Collectors.toSet()); - - List userLikes = getUserLikesByGroupId(userLikeGroup.getId()); - - for (UserLike userLike : userLikes) { - List> userLikeSketchList = new ArrayList<>(); - List designItemDetails = getDesignItemDetails(userLike.getDesignItemId()); - - for (DesignItemDetail designItemDetail : designItemDetails) { - Map sketch = new HashMap<>(); - String path = designItemDetail.getPath(); - - // 下载路径到本地 -// if (!StringUtils.isEmpty(path)) { -// String bucketName = minioUtil.getBucketNameFromPath(path); -// String objectName = minioUtil.getObjectNameFromPath(path); -// String localBasePath = "C:\\workspace\\fileData\\minio"; -// String localFilePath = Paths.get(localBasePath, path).toString(); -// -// // 检查文件是否已经存在，避免重复下载 -// File localFile = new File(localFilePath); -// if (!localFile.exists()) { -// minioUtil.downloadMinioObjectToLocal(bucketName, objectName, localFilePath); -// } else { -// log.info("File already exists, skipping download: {}", localFilePath); -// } -// } - - Long designId = designItemDetail.getDesignId(); - Design design = designMapper.selectById(designId); - if (null == design) { - continue; - } - if (design.getSingleOverall().equals("single")) { - continue; - } - if (design.getModelType().equals("System")) { - SysFile sysFile = sysFileMapper.selectById(design.getTemplateId()); - if (null == sysFile) { - continue; - } - sketch.put("sex", sysFile.getLevel2Type().equals("Male") ? "Male" : "Female"); - } else { - Library library = libraryMapper.selectById(design.getTemplateId()); - if (null == library) { - continue; - } - sketch.put("sex", library.getLevel2Type().equals("Female") ? "Female" : "Male"); - } - - sketch.put("type", designItemDetail.getType()); - sketch.put("path", path); - sketch.put("beSelected", urlSet.contains(path)); - sketch.put("accountId", accountId); - - if (path.contains("aida-sys-image/images/")) { - String replace = path.replace("aida-sys-image/images/", ""); - String style = getStyleByUrl(replace); - if (!StringUtils.isEmpty(style)) { - sketch.put("style", style); - } - } - userLikeSketchList.add(sketch); - } - if (CollectionUtil.isNotEmpty(userLikeSketchList)) { - result.add(userLikeSketchList); - } - } - } - - // 将结果以美观的JSON形式保存到文件中 - saveResultAsPrettyJson(result, "C:\\Users\\10233\\Desktop\\result.json"); - - return null; - } - - @Override - public String download() { - // 下载 aida-collection-element bucket 下所有 Sketchboard 文件夹内容 -// String sketchboardLocalPath = "C:\\workspace\\fileData\\minio"; -// minioUtil.downloadSketchboardDirectories(sketchboardLocalPath); -// -// // 下载整个 aida-users bucket -// String aidaUsersLocalPath = "C:\\workspace\\fileData\\minio"; -// minioUtil.downloadEntireBucket("aida-users", aidaUsersLocalPath); - return null; - } - - // 提取公共方法 - private List getCollectionElementsByCollectionId(Long collectionId) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(CollectionElement::getCollectionId, collectionId) - .eq(CollectionElement::getLevel1Type, "Sketchboard"); - return collectionElementMapper.selectList(qw); - } - - private List getUserLikesByGroupId(Long groupId) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(UserLike::getUserLikeGroupId, groupId); - return userLikeMapper.selectList(qw); - } - - private List getDesignItemDetails(Long designItemId) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(DesignItemDetail::getDesignItemId, designItemId) - .ne(DesignItemDetail::getType, "Body"); - return designItemDetailService.list(qw); - } - - private void saveResultAsPrettyJson(List>> result, String filePath) { - // 使用一个 Map 将结果存储为字符串键 - Map resultMap = new HashMap<>(); - resultMap.put("data", result); - - try (FileWriter file = new FileWriter(filePath)) { - // 使用 JSONObject 将转换后的 Map 转为 JSON - JSONObject jsonObject = new JSONObject(resultMap); - // 格式化输出为美观的 JSON - file.write(JSON.toJSONString(jsonObject, SerializerFeature.PrettyFormat)); - } catch (IOException e) { - e.printStackTrace(); - } - } - - - - - private String getStyleByUrl(String replace) { - String[] split = replace.split("/"); - String tableName = getTableName(split); - if (StringUtils.isEmpty(tableName)) { - return null; - }else { - replace = split[1] + "/" + split[2]; - return attributeRetrievalMapper.getStyleByUrl(replace, tableName); - } - } - - private String getTableName(String[] split) { - switch (split[0]) { - case "female": { - switch (split[1]) { - case "blouse": { - return "female_top"; - } - case "trousers": { - return "female_pants"; - } - case "skirt": { - return "female_skirt"; - } - case "outwear": { - return "female_outwear"; - } - case "dress": { - return "female_dress"; - } - default: { - return null; - } - } - } - case "male": { - switch (split[1]) { - case "top": { - return "male_top"; - } - case "bottom": { - return "male_bottom"; - } - case "outwear": { - return "male_outwear"; - } - default: { - return null; - } - } - } - default: { - return null; - } - } - } - @Override - public Boolean productImageInitialize(ProductImageInitializeDTO productImageInitializeDTO) { - AuthPrincipalVo userHolder = UserContext.getUserHolder(); - productImageService.asyncInitialize(productImageInitializeDTO.getBrandId(), userHolder); - return true; - } - - @Override - public InitializeProgressVO getInitializeProgress(ProductImageInitializeDTO productImageInitializeDTO) { - InitializeProgressVO vo = new InitializeProgressVO(); - -// QueryWrapper<> - - ProgressDTO dto = redisUtil.getTaskProgressDTO(String.valueOf(productImageInitializeDTO.getBrandId())); - - if (Objects.isNull(dto)) { - vo.setAnalyzed(false); - return vo; - } - vo.setAnalyzed(true); - if (dto.getTotal() == 0 || dto.isError()) { - vo.setPercent(0.0); - return vo; - } - vo.setPercent(dto.getCurrent() * 1.0 / dto.getTotal()); - return vo; - } - - private ProductImageAttribute toAttrDict(AttributeRecognitionJSON attrDictJSON) { - ProductImageAttribute attributeRetrieval = new ProductImageAttribute(); - - // 处理 imgName -// if (CollectionUtil.isNotEmpty(attrDictJSON.getImgName()) && attrDictJSON.getImgName().get(0) != null) { -// attributeRetrieval.setImgName(attrDictJSON.getImgName().get(0)); -// } - // 处理 length - if (CollectionUtil.isNotEmpty(attrDictJSON.getLength()) && attrDictJSON.getLength().get(0) != null) { - attributeRetrieval.setLength(attrDictJSON.getLength().get(0)); - } - // 处理 sleeveLength - if (CollectionUtil.isNotEmpty(attrDictJSON.getSleeveLength()) && attrDictJSON.getSleeveLength().get(0) != null) { - attributeRetrieval.setSleeveLength(attrDictJSON.getSleeveLength().get(0)); - } - // 处理 sleeveShape - if (CollectionUtil.isNotEmpty(attrDictJSON.getSleeveShape()) && attrDictJSON.getSleeveShape().get(0) != null) { - attributeRetrieval.setSleeveShape(attrDictJSON.getSleeveShape().get(0)); - } - // 处理 sleeveShoulder - if (CollectionUtil.isNotEmpty(attrDictJSON.getSleeveShoulder()) && attrDictJSON.getSleeveShoulder().get(0) != null) { - attributeRetrieval.setSleeveShoulder(attrDictJSON.getSleeveShoulder().get(0)); - } - // 处理 neckline - if (CollectionUtil.isNotEmpty(attrDictJSON.getNeckline()) && attrDictJSON.getNeckline().get(0) != null) { - attributeRetrieval.setNeckline(attrDictJSON.getNeckline().get(0)); - } - // 处理 collar - if (CollectionUtil.isNotEmpty(attrDictJSON.getCollar()) && attrDictJSON.getCollar().get(0) != null) { - attributeRetrieval.setCollar(attrDictJSON.getCollar().get(0)); - } - // 处理 design - if (CollectionUtil.isNotEmpty(attrDictJSON.getDesign()) && attrDictJSON.getDesign().get(0) != null) { - attributeRetrieval.setDesign(attrDictJSON.getDesign().get(0)); - } - // 处理 silhouette - if (CollectionUtil.isNotEmpty(attrDictJSON.getSilhouette()) && attrDictJSON.getSilhouette().get(0) != null) { - attributeRetrieval.setSilhouette(attrDictJSON.getSilhouette().get(0)); - } - // 处理 type - if (CollectionUtil.isNotEmpty(attrDictJSON.getType()) && attrDictJSON.getType().get(0) != null) { - attributeRetrieval.setType(attrDictJSON.getType().get(0)); - } - // 处理 openingType - if (CollectionUtil.isNotEmpty(attrDictJSON.getOpeningType()) && attrDictJSON.getOpeningType().get(0) != null) { - attributeRetrieval.setOpeningType(attrDictJSON.getOpeningType().get(0)); - } - // 处理 subtype - if (CollectionUtil.isNotEmpty(attrDictJSON.getSubtype()) && attrDictJSON.getSubtype().get(0) != null) { - attributeRetrieval.setSubtype(attrDictJSON.getSubtype().get(0)); - } - - return attributeRetrieval; - } - -// private ProductImageAttribute toAttrDict(AttributeRecognitionJSON attrDictJSON) { -// ProductImageAttribute attributeRetrieval = new ProductImageAttribute(); -//// attributeRetrieval.setImgName(attrDictJSON.getImgName().get(0)); -//// attributeRetrieval.setLength(attrDictJSON.getLength().get(0)); -//// attributeRetrieval.setSleeveLength(attrDictJSON.getSleeveLength().get(0)); -//// attributeRetrieval.setSleeveShape(attrDictJSON.getSleeveShape().get(0)); -//// attributeRetrieval.setSleeveShoulder(attrDictJSON.getSleeveShoulder().get(0)); -//// attributeRetrieval.setNeckline(attrDictJSON.getNeckline().get(0)); -//// attributeRetrieval.setCollar(attrDictJSON.getCollar().get(0)); -// if (CollectionUtil.isNotEmpty(attrDictJSON.getDesign()) && attrDictJSON.getDesign().get(0) != null) { -// attributeRetrieval.setDesign(attrDictJSON.getDesign().get(0)); -// } -// if (CollectionUtil.isNotEmpty(attrDictJSON.getSilhouette()) && attrDictJSON.getSilhouette().get(0) != null) { -// attributeRetrieval.setSilhouette(attrDictJSON.getSilhouette().get(0)); -// } -// if (CollectionUtil.isNotEmpty(attrDictJSON.getType()) && attrDictJSON.getType().get(0) != null) { -// attributeRetrieval.setType(attrDictJSON.getType().get(0)); -// } -// if (CollectionUtil.isNotEmpty(attrDictJSON.getSubtype()) && attrDictJSON.getSubtype().get(0) != null) { -// attributeRetrieval.setSubtype(attrDictJSON.getSubtype().get(0)); -// } -// if (CollectionUtil.isNotEmpty(attrDictJSON.getOpeningType()) && attrDictJSON.getOpeningType().get(0) != null) { -// attributeRetrieval.setOpeningType(attrDictJSON.getOpeningType().get(0)); -// } -// return attributeRetrieval; -// } - - @Override - public IPage getPage(ProjectQueryDTO query) { - AuthPrincipalVo authPrincipalVo = UserContext.getUserHolder(); - // 分页数据 - QueryWrapper queryWrapper = new QueryWrapper<>(); - - queryWrapper.eq("account_id", authPrincipalVo.getId()); - if (Objects.nonNull(query.getId())){ - queryWrapper.eq("id", query.getId()); - } -// queryWrapper.isNull("parent_id"); - if (!StringUtils.isEmpty(query.getProjectName())) { - queryWrapper.like("name", query.getProjectName()); - } - if (Objects.nonNull(query.getStartDate())) { - queryWrapper.ge("update_time", new Date(query.getStartDate())); - } - if (Objects.nonNull(query.getEndDate())) { - queryWrapper.le("update_time", new Date(query.getEndDate())); - } - if (!StringUtils.isEmpty(query.getProcess())) { - queryWrapper.eq("process", query.getProcess()); - } - // 新增分类过滤 - if (CollectionUtil.isNotEmpty(query.getClassificationIdList())) { - List projectIdList = new ArrayList<>(); - if (query.getIntersection() == 0) { - for (Long classificationId : query.getClassificationIdList()) { - projectIdList.addAll(classificationService.getLibraryIdListByClassificationId(classificationId)); - } - if (CollectionUtil.isNotEmpty(projectIdList)) { - queryWrapper.in("id", projectIdList); - }else { - return new Page<>(); - } - }else { - for (int i = 0; i < query.getClassificationIdList().size(); i++) { - List historyIdListByClassificationId = classificationService.getLibraryIdListByClassificationId(query.getClassificationIdList().get(i)); - if (i == 0) { - projectIdList.addAll(historyIdListByClassificationId); - }else { - projectIdList.retainAll(historyIdListByClassificationId); - } - if (CollectionUtil.isEmpty(projectIdList)) { - return new Page<>(); - } - } - queryWrapper.in("id", projectIdList); - } - } - if (query.getAsc() != null && query.getAsc() == 1) { - queryWrapper.orderByAsc("update_time"); - }else { - queryWrapper.orderByDesc("update_time"); - } - IPage page = projectService.page( - new Page<>(query.getPage(), query.getSize()), queryWrapper); - if (CollectionUtils.isEmpty(page.getRecords())) { - return new Page<>(); - } - Set projectIdSet = page.getRecords().stream().map(Project::getId).collect(Collectors.toSet()); - QueryWrapper userLikeGroupQueryWrapper = new QueryWrapper<>(); - userLikeGroupQueryWrapper.lambda().in(UserLikeGroup::getProjectId, projectIdSet); - List userLikeGroups = userLikeGroupMapper.selectList(userLikeGroupQueryWrapper); - - Map> groupDetailMap = new HashMap<>(); - - if (CollectionUtil.isNotEmpty(userLikeGroups)) { - List groupIds = userLikeGroups.stream().map(UserLikeGroup::getId).collect(Collectors.toList()); - List groupDetails = userLikeService.getGroupDetails(groupIds); - - groupDetailMap = groupDetails.stream() - .collect(Collectors.groupingBy(UserLikeVO::getUserLikeGroupId)); - } - - Account account = accountService.getById(authPrincipalVo.getId()); - Map> finalGroupDetailMap = groupDetailMap; - IPage convert = page.convert((Function) project -> { - if (project != null) { - ProjectVO projectVO = CopyUtil.copyObject(project, ProjectVO.class); - projectVO.setUpdateDate(project.getUpdateTime().atZone(ZoneId.systemDefault()).toInstant().toEpochMilli()); - - QueryWrapper userLikeGroupQueryWrapper1 = new QueryWrapper<>(); - userLikeGroupQueryWrapper1.lambda().eq(UserLikeGroup::getProjectId, projectVO.getId()); - UserLikeGroup userLikeGroup = userLikeGroupMapper.selectOne(userLikeGroupQueryWrapper1); - if (Objects.nonNull(userLikeGroup)) { - UserLikeGroupVO userLikeGroupVO = CopyUtil.copyObject(userLikeGroup, UserLikeGroupVO.class); - userLikeGroupVO.setAuthor(account.getUserName()); - //count 和detail - if (finalGroupDetailMap.keySet().contains(userLikeGroupVO.getId())) { - List details = finalGroupDetailMap.get(userLikeGroupVO.getId()); - for (UserLikeVO detail : details) { - TDesignPythonOutfit tDesignPythonOutfit = designPythonOutfitMapper.selectById(detail.getDesignOutfitId()); - detail.setUrl(minioUtil.getPreSignedUrl(tDesignPythonOutfit.getDesignUrl(), 24 * 60)); - } - userLikeGroupVO.setGroupDetails(details); - userLikeGroupVO.setSketchCount(CollectionUtils.isEmpty(details) ? 0 : details.size()); - }else { - userLikeGroupVO.setSketchCount(0); - } - if (userLikeGroupVO.getOriginal() == 0) { - userLikeGroupVO.setOriginalAccountName(accountService.getById(userLikeGroupVO.getOriginalAccountId()).getUserName()); - Portfolio byId = portfolioService.getByIdAll(userLikeGroupVO.getOriginalPortfolioId()); - if (Objects.nonNull(byId)) { - String portfolioName = byId.getPortfolioName(); - userLikeGroupVO.setOriginalPortfolioName(portfolioName); - } - } - projectVO.setUserLikeGroupVO(userLikeGroupVO); - } - return projectVO; - } - return null; - }); - return convert; - } - - - @Override - public ModuleChooseVO getModuleContent(ProjectDTO projectDTO) { - ModuleChooseVO moduleChooseVO = new ModuleChooseVO(); - moduleChooseVO.setProjectId(projectDTO.getId()); -// UserLikeCollectionVO userLikeCollectionVO = new UserLikeCollectionVO(); - for (String module : projectDTO.getModuleList()) { - if (module.equals(Module.colorBoard.name())) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(CollectionElement::getProjectId, projectDTO.getId()); - qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.COLOR_BOARD.getRealName()); - List collectionElements = collectionElementMapper.selectList(qw); - moduleChooseVO.setColorBoard(collectionService.resolveColorBoard(collectionElements)); - }else if (module.equals(Module.moodBoard.name())) { - MoodBoardModuleChooseVO moodBoardModuleChooseVO = new MoodBoardModuleChooseVO(); - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(CollectionElement::getProjectId, projectDTO.getId()); - qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.MOOD_BOARD.getRealName()); - List collectionElements = collectionElementMapper.selectList(qw); - - List list = new ArrayList<>(); - for (CollectionElement collectionElement : collectionElements) { - if (null != collectionElement.getIsCompositeImage() && collectionElement.getIsCompositeImage() == 1) { - moodBoardModuleChooseVO.setMoodTemplateId(String.valueOf(collectionElement.getId())); - moodBoardModuleChooseVO.setMoodTemplateName(collectionElement.getName()); - moodBoardModuleChooseVO.setMoodTemplateUrl(getMinioUrl(collectionElement.getUrl())); - - String moodboardPositionString = collectionService.getMoodboardPositionString(collectionElement.getId()); - if (!StringUtils.isEmpty(moodboardPositionString)) { - moodBoardModuleChooseVO.setMoodboardPosition(moodboardPositionString); - } - continue; - } - CollectionElementVO collectionElementVO = CopyUtil.copyObject(collectionElement, CollectionElementVO.class); - collectionElementVO.setDesignType(DesignTypeEnum.COLLECTION.getRealName()); - String url = collectionElement.getUrl(); - collectionElementVO.setOriginalUrl(url); - if (minioUtil.doesObjectExist(url)) { - collectionElementVO.setUrl(getMinioUrl(url)); - } - if (minioUtil.doesObjectExist(collectionElementVO.getOriginalUrl())) { - list.add(collectionElementVO); - } - } - moodBoardModuleChooseVO.setMoodBoards(list); - moduleChooseVO.setMoodBoard(moodBoardModuleChooseVO); - }else if (module.equals(Module.printBoard.name())) { - - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(CollectionElement::getProjectId, projectDTO.getId()); - qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.PRINT_BOARD.getRealName()); - List collectionElements = collectionElementMapper.selectList(qw); - List list = new ArrayList<>(); - - for (CollectionElement collectionElement : collectionElements) { - CollectionElementVO collectionElementVO = CopyUtil.copyObject(collectionElement, CollectionElementVO.class); - collectionElementVO.setIsPin(collectionElement.getHasPin()); - collectionElementVO.setDesignType(DesignTypeEnum.COLLECTION.getRealName()); - String url = collectionElement.getUrl(); - collectionElementVO.setOriginalUrl(url); - if (minioUtil.doesObjectExist(url)) { - collectionElementVO.setUrl(getMinioUrl(url)); - } - if (minioUtil.doesObjectExist(collectionElementVO.getOriginalUrl())) { - list.add(collectionElementVO); - } - } - moduleChooseVO.setPrintBoard(list); - }else if (module.equals(Module.sketchBoard.name())) { - - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(CollectionElement::getProjectId, projectDTO.getId()); - qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.SKETCH_BOARD.getRealName()); - List collectionElements = collectionElementMapper.selectList(qw); - List list = new ArrayList<>(); - - for (CollectionElement collectionElement : collectionElements) { - CollectionElementVO collectionElementVO = CopyUtil.copyObject(collectionElement, CollectionElementVO.class); - collectionElementVO.setIsPin(collectionElement.getHasPin()); - collectionElementVO.setDesignType(DesignTypeEnum.COLLECTION.getRealName()); - String url = collectionElement.getUrl(); - collectionElementVO.setOriginalUrl(url); - if (minioUtil.doesObjectExist(url)) { - collectionElementVO.setUrl(getMinioUrl(url)); - if (url.contains(".")) { - String[] split = url.split("\\."); - collectionElementVO.setUrlWithWhiteSide(getMinioUrl(split[0] + "-show." + split[1])); - } else { - collectionElementVO.setUrlWithWhiteSide(getMinioUrl(url + "-show")); - } - } - if (minioUtil.doesObjectExist(collectionElementVO.getOriginalUrl())) { - list.add(collectionElementVO); - } - } - List voList = new ArrayList<>(); - for (CollectionElement collectionElement : collectionElements) { - CollectionSketchVO vo = CopyUtil.copyObject(collectionElement, CollectionSketchVO.class); - String url = collectionElement.getUrl(); - if (url.contains(".")) { - String[] split = url.split("\\."); - vo.setUrlWithWhiteSide(getMinioUrl(split[0] + "-show." + split[1], true)); - }else { - vo.setUrlWithWhiteSide(getMinioUrl(url + "-show", true)); - } - vo.setIsPin(collectionElement.getHasPin()); - vo.setLevel2Type(collectionElement.getLevel2Type()); - vo.setDesignType("Collection"); - vo.setSketchBoardId(collectionElement.getId()); - voList.add(vo); - } - moduleChooseVO.setBoundingBox(voList); - moduleChooseVO.setSketchBoard(list); - }else if (module.equals(Module.design.name())) { - DesignModuleChooseVO vo = new DesignModuleChooseVO(); - QueryWrapper userLikeSortQueryWrapper = new QueryWrapper<>(); - userLikeSortQueryWrapper.lambda().eq(CollectionSort::getProjectId, projectDTO.getId()); - userLikeSortQueryWrapper.lambda().isNull(CollectionSort::getParentId); - userLikeSortQueryWrapper.lambda().eq(CollectionSort::getRelationType, CollectionType.DESIGN.getValue()); - userLikeSortQueryWrapper.lambda().orderByAsc(CollectionSort::getSort); - List userLikeSortList = collectionSortMapper.selectList(userLikeSortQueryWrapper); - - if (CollectionUtils.isEmpty(userLikeSortList)) { - moduleChooseVO.setDesign(vo); - }else { - List list = new ArrayList<>(); - - for (CollectionSort collectionSort : userLikeSortList) { - UserLike userLike = userLikeMapper.selectById(collectionSort.getRelationId()); - UserLikeVO o = CopyUtil.copyObject(userLike, UserLikeVO.class); - TDesignPythonOutfit tDesignPythonOutfit1 = designPythonOutfitMapper.selectById(o.getDesignOutfitId()); - o.setUrl(tDesignPythonOutfit1.getDesignUrl()); - if (o.getUrl().contains("/")) { - int index = o.getUrl().lastIndexOf("/"); - o.setPictureName(o.getUrl().substring(index + 1)); - } - o.setDesignOutfitUrl(getMinioUrl(o.getUrl())); - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(TDesignPythonOutfit::getDesignItemId, o.getDesignItemId()); - List tDesignPythonOutfits = designPythonOutfitMapper.selectList(qw); - if (CollectionUtil.isNotEmpty(tDesignPythonOutfits)) { - TDesignPythonOutfit tDesignPythonOutfit = tDesignPythonOutfits.get(0); - o.setDesignOutfitId(tDesignPythonOutfit.getId()); - } - - o.setSort(collectionSort.getSort()); - o.setUserLikeSortId(collectionSort.getId()); - o.setRelationType(collectionSort.getRelationType()); - o.setCollectionType(CollectionType.DESIGN.getValue()); - o.setResultType(CollectionType.DESIGN.getValue()); - QueryWrapper childCollectionQw = new QueryWrapper<>(); - childCollectionQw.lambda().eq(CollectionSort::getParentId, collectionSort.getId()); - childCollectionQw.lambda().orderByAsc(CollectionSort::getSort); - List childSortList = collectionSortMapper.selectList(childCollectionQw); - List childList = new ArrayList<>(); - for (CollectionSort userLikeSort : childSortList) { - if (userLikeSort.getRelationType().equals(CollectionType.TO_PRODUCT_IMAGE.getValue())) { - ToProductImageResult toProductImageResult = toProductImageResultMapper.selectById(userLikeSort.getRelationId()); - if (isGenerateTaskFailed(toProductImageResult.getStatus(), toProductImageResult.getCreateTime())){ - continue; - } - toProductImageResult.setUrl(getMinioUrl(toProductImageResult.getUrl())); - ToProductImageResultVO toProductImageResultVO = CopyUtil.copyObject(toProductImageResult, ToProductImageResultVO.class); - - ToProductImageRecord toProductImageRecord = toProductImageRecordMapper.selectById(toProductImageResult.getToProductImageRecordId()); - if (Objects.isNull(toProductImageRecord)) { - log.warn("toProductRecord不存在，projectId:{}, toProductImageResultId:{}", projectDTO.getId(), toProductImageResult.getId()); - continue; - } - toProductImageResultVO.setPrompt(toProductImageRecord.getPrompt()); - - 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())); - } - toProductImageResultVO.setCollectionType(CollectionType.TO_PRODUCT_IMAGE.getValue()); - toProductImageResultVO.setSort(userLikeSort.getSort()); - toProductImageResultVO.setUserLikeSortId(userLikeSort.getId()); - toProductImageResultVO.setRelationType(userLikeSort.getRelationType()); - toProductImageResultVO.setParentId(userLikeSort.getParentId()); - childList.add(toProductImageResultVO); - } else if (userLikeSort.getRelationType().equals(CollectionType.RELIGHT.getValue())) { - ToProductImageResult toProductImageResult = toProductImageResultMapper.selectById(userLikeSort.getRelationId()); - if (isGenerateTaskFailed(toProductImageResult.getStatus(), toProductImageResult.getCreateTime())){ - continue; - } - toProductImageResult.setUrl(getMinioUrl(toProductImageResult.getUrl())); - ToProductImageResultVO toProductImageResultVO = CopyUtil.copyObject(toProductImageResult, ToProductImageResultVO.class); - - ToProductImageRecord toProductImageRecord = toProductImageRecordMapper.selectById(toProductImageResult.getToProductImageRecordId()); - if (Objects.isNull(toProductImageRecord)) { - log.warn("toProductRecord不存在，projectId:{}, toProductImageResultId:{}", projectDTO.getId(), toProductImageResult.getId()); - continue; - } - toProductImageResultVO.setPrompt(toProductImageRecord.getPrompt()); - - 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())); - } - toProductImageResultVO.setCollectionType(CollectionType.RELIGHT.getValue()); - toProductImageResultVO.setSort(userLikeSort.getSort()); - toProductImageResultVO.setUserLikeSortId(userLikeSort.getId()); - toProductImageResultVO.setRelationType(userLikeSort.getRelationType()); - toProductImageResultVO.setParentId(userLikeSort.getParentId()); - childList.add(toProductImageResultVO); - } else if (userLikeSort.getRelationType().equals(CollectionType.POSE_TRANSFORM.getValue())) { - PoseTransformation item = poseTransformationMapper.selectById(userLikeSort.getRelationId()); - if (isGenerateTaskFailed(item.getTaskStatus(), item.getCreateTime())){ - continue; - } - PoseTransformationVO poseTransformationVO = new PoseTransformationVO(); - poseTransformationVO.setId(item.getId()); - poseTransformationVO.setTaskId(item.getUniqueId()); - poseTransformationVO.setProductImage(getMinioUrl(item.getProductImage())); - poseTransformationVO.setGifUrl(getMinioUrl(item.getGifUrl())); - poseTransformationVO.setVideoUrl(getMinioUrl(item.getVideoUrl())); - poseTransformationVO.setFirstFrameUrl(getMinioUrl(item.getFirstFrameUrl())); - poseTransformationVO.setIsLiked(item.getIsLiked()); - poseTransformationVO.setCollectionType(CollectionType.POSE_TRANSFORM.getValue()); - poseTransformationVO.setSort(userLikeSort.getSort()); - poseTransformationVO.setUserLikeSortId(userLikeSort.getId()); - poseTransformationVO.setRelationType(userLikeSort.getRelationType()); - poseTransformationVO.setResultType(CollectionType.POSE_TRANSFORM.getValue()); - poseTransformationVO.setParentId(userLikeSort.getParentId()); - poseTransformationVO.setModelName(item.getModelName()); - poseTransformationVO.setPoseId(item.getPoseId()); - poseTransformationVO.setStatus(item.getTaskStatus()); - childList.add(poseTransformationVO); - } - } - o.setChildList(childList); - - list.add(o); - } - vo.setUserLikeDetails(list); - } - - UserLikeGroup group = getByProjectId(projectDTO.getId()); - if (Objects.nonNull(group)) { - Long userGroupId = group.getId(); - vo.setUserGroupId(userGroupId); - } - moduleChooseVO.setDesign(vo); - }else if (module.equals(Module.canvas.name())) { - CavasModuleChooseVO vo = new CavasModuleChooseVO(); - - UserLikeGroup userLikeGroup = getByProjectId(projectDTO.getId()); - - Integer beenPublished = 0; - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(Portfolio::getUserLikeGroupSourceId, userLikeGroup.getId()); - List portfolios = portfolioMapper.selectList(qw); - - PortfolioDTO portfolioDTO = new PortfolioDTO(); - if (CollectionUtil.isNotEmpty(portfolios)) { - - portfolioDTO = CopyUtil.copyObject(portfolios.get(0), PortfolioDTO.class); - beenPublished = 1; - portfolioDTO.setTagsDTO(tagsMapper.getTagByPortfolioId(portfolioDTO.getId())); - } - vo.setBeenPublished(beenPublished); - vo.setPortfolioDTO(portfolioDTO); - moduleChooseVO.setCanvas(vo); - }else if (module.equals(Module.toProduct.name()) || module.equals(Module.relight.name())) { - Long projectId = projectDTO.getId(); - String resultType = module.equals(Module.toProduct.name()) ? "ToProductImage" : "Relight"; - - // 查询所有结果（包括喜欢和不喜欢的） - QueryWrapper qw = new QueryWrapper<>(); - 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); - // 设置prompt - ToProductImageRecord toProductImageRecord = toProductImageRecordMapper.selectById(result.getToProductImageRecordId()); - if (Objects.nonNull(toProductImageRecord)){ - vo.setPrompt(toProductImageRecord.getPrompt()); - } - vo.setParentId(getUnlikedResultParentId(result, null)); - // 按isLike分类 - if (result.getIsLike() != null && result.getIsLike() == 1) { - likedList.add(vo); - } else { - normalList.add(vo); - } - } - // 设置结果 - resultVO.setLikedList(likedList); - resultVO.setList(normalList); - - // 根据模块设置返回结果 - if (module.equals(Module.toProduct.name())) { - moduleChooseVO.setToProduct(resultVO); - } else { - moduleChooseVO.setRelight(resultVO); - } - }else if (module.equals(Module.deReconstruction.name())){ - SketchReconstructionVO sketchReconstruction = generateService.getSketchReconstruction(projectDTO.getId()); - moduleChooseVO.setDeReconstruction(sketchReconstruction); - }else if (module.equals(Module.poseTransfer.name())){ - List likedList = generateService.getPoseTransformationResult(null, projectDTO.getId(), true); - List list = generateService.getPoseTransformationResult(null, projectDTO.getId(), false); - PoseTransferVO vo = new PoseTransferVO(); - vo.setLikedList(likedList); - vo.setList(list); - moduleChooseVO.setPoseTransfer(vo); - }else if (module.equals(Module.patternMaking3D.name())) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(ThreeDModule::getProjectId, projectDTO.getId()); - List threeDModules = threeDModuleMapper.selectList(qw); - if (CollectionUtil.isNotEmpty(threeDModules)) { - ThreeDModule threeDModule = threeDModules.get(0); - if (null != threeDModule.getThreeDSimpleId()){ - ThreeDModuleVO threeDModuleVO = CopyUtil.copyObject(threeDModule, ThreeDModuleVO.class); - if (null != threeDModule.getCollectionElementId()) { - CollectionElement collectionElement = collectionElementMapper.selectById(threeDModule.getCollectionElementId()); - if (Objects.nonNull(collectionElement)) { - threeDModuleVO.setPrintMinioUrl(getMinioUrl(collectionElement.getUrl())); - } - } -// if (!StringUtils.isEmpty(threeDModuleVO.getPrintMinioUrl())) { -// threeDModuleVO.setPrintMinioUrl(minioUtil.getPreSignedUrl(threeDModuleVO.getPrintMinioUrl(), CommonConstant.MINIO_IMAGE_EXPIRE_TIME)); -// } - moduleChooseVO.setPatternMaking3D(threeDModuleVO); - } - } - }else if (module.equals(Module.mannequin.name())) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(CollectionElement::getProjectId, projectDTO.getId()); - qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.MODEL.getRealName()); - List collectionElements = collectionElementMapper.selectList(qw); - List list = new ArrayList<>(); - if (CollectionUtil.isNotEmpty(collectionElements)) { - for (CollectionElement collectionElement : collectionElements) { - CollectionElementVO collectionElementVO = CopyUtil.copyObject(collectionElement, CollectionElementVO.class); - CollectionElementRelModel relModel = getCollectionElementRelModel(collectionElementVO.getId()); - collectionElementVO.setUrl(getMinioUrl(collectionElementVO.getUrl())); - collectionElementVO.setMannequinRelationId(relModel.getRelationId()); - collectionElementVO.setMannequinRelationType(relModel.getRelationType()); - collectionElementVO.setCollectionId(collectionElement.getId()); - list.add(collectionElementVO); - } - } - moduleChooseVO.setMannequin(list); - }else if (module.equals(Module.uploadElement.name())) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(ToProductElement::getProjectId, projectDTO.getId()); - List toProductElements = toProductElementMapper.selectList(qw); - for (ToProductElement toProductElement : toProductElements) { - toProductElement.setUrl(getMinioUrl(toProductElement.getUrl())); - } - moduleChooseVO.setUploadElement(toProductElements); - } - } - return moduleChooseVO; - } - - private String getMinioUrl(String minioPath){ - if (!StringUtil.isNullOrEmpty(minioPath)){ - return minioUtil.getPreSignedUrl(minioPath, CommonConstant.MINIO_IMAGE_EXPIRE_TIME); - } - return null; - } - - private String getMinioUrl(String minioPath, boolean resetCatch){ - if (!StringUtil.isNullOrEmpty(minioPath)){ - return minioUtil.getPreSignedUrl(minioPath, CommonConstant.MINIO_IMAGE_EXPIRE_TIME, resetCatch); - } - return null; - } - - // 提取出的公共方法 - private void setSourceUrl(ToProductImageResultVO vo) { - if (vo.getElementType().equals("ProductElement")) { - ToProductElement element = toProductElementMapper.selectById(vo.getElementId()); - vo.setSourceUrl(getMinioUrl(element.getUrl())); - } else if (vo.getElementType().equals("DesignOutfit")) { - TDesignPythonOutfit outfit = designPythonOutfitMapper.selectById(vo.getElementId()); - vo.setSourceUrl(getMinioUrl(outfit.getDesignUrl())); - } else { - ToProductImageResult result = toProductImageResultMapper.selectById(vo.getElementId()); - vo.setSourceUrl(getMinioUrl(result.getUrl())); - } - } - - private CollectionElementRelModel getCollectionElementRelModel(Long id) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(CollectionElementRelModel::getCollectionElementId, id); - CollectionElementRelModel collectionElementRelModel = collectionElementRelModelMapper.selectOne(qw); - return collectionElementRelModel; - } - - private boolean isGenerateTaskFailed(String taskStatus, LocalDateTime createTime) { - // 1. 先判断TaskStatus字段是否有值；若有值且为Fail，则失败 - if (!StringUtil.isNullOrEmpty(taskStatus)) { - return "Fail".equals(taskStatus); - } - // 2. 若taskStatus无值，判断记录创建时间距离现在是否超过1小时 - if (createTime == null) { - // 如果创建时间为null，视为失败（根据业务需求决定） - return true; - } - // 计算当前时间与创建时间的差值 - Duration duration = Duration.between(createTime, LocalDateTime.now()); - // 判断是否超过1小时（3600秒） - return duration.toHours() >= 1; - } - - // 获取未被like的to product\relight的parentId - public Long getUnlikedResultParentId(ToProductImageResult toProductImageResult, String url) { - // 卫语句处理null情况 - if (toProductImageResult == null && StringUtil.isNullOrEmpty(url)) { - return null; - } - - // 如果需要查询结果对象 - if (toProductImageResult == null) { - toProductImageResult = toProductImageResultMapper.selectOne( - new QueryWrapper().eq("url", url)); - } - - // 卫语句处理查询结果为空的情况 - if (toProductImageResult == null) { - return null; - } - - // 根据不同类型处理 - String elementType = toProductImageResult.getElementType(); - if ("DesignOutfit".equals(elementType)) { - return handleDesignOutfitCase(toProductImageResult); - } else if ("ToProductImage".equals(elementType)) { - // 两种情况 resultType : ToProductImage | Relight - return handleToProductImageCase(toProductImageResult); - } - - return null; - } - - private Long handleDesignOutfitCase(ToProductImageResult result) { - UserLike userLike = userLikeService.getByDesignOutfitId(result.getElementId()); - if (userLike == null) { - return null; - } - - CollectionSort collectionSort = collectionSortService.queryCollectionSortByRelation( - userLike.getId(), "Design", result.getProjectId()); - return collectionSort != null ? collectionSort.getId() : null; - } - - private Long handleToProductImageCase(ToProductImageResult result) { - CollectionSort collectionSort = collectionSortService.queryCollectionSortByRelation( - result.getElementId(), result.getResultType(), result.getProjectId()); - if (Objects.isNull(collectionSort) && result.getElementType().equals("ToProductImage")){ - ToProductImageResult toProductImageResult = toProductImageResultMapper.selectById(result.getElementId()); - return getUnlikedResultParentId(toProductImageResult, null); - } - return collectionSort != null ? collectionSort.getParentId() : null; - } - - @Override - @Transactional - public ModuleChooseVO saveModuleContent(ModuleSaveDTO moduleSaveDTO) { - ModuleChooseVO result = new ModuleChooseVO(); - Long accountId = UserContext.getUserHolder().getId(); - Long projectId = moduleSaveDTO.getProjectId(); - Project project = projectService.getById(projectId); - - boolean boundingBox = false; - if (Objects.nonNull(moduleSaveDTO.getMoodBoard())) { - MoodBoardModuleSaveDTO moodBoard = moduleSaveDTO.getMoodBoard(); - if (moodBoard.getMoodTemplateId() != null) { - // moodboard合成图存储 - CollectionElement compositeImage = collectionElementMapper.selectById(moodBoard.getMoodTemplateId()); - if (compositeImage.getProjectId() == null) { - compositeImage.setProjectId(moduleSaveDTO.getProjectId()); - compositeImage.setIsCompositeImage(1); - collectionElementMapper.updateById(compositeImage); - if (!StringUtils.isEmpty(moodBoard.getMoodboardPosition())) { - // 合成图位置信息通过collectElementId关联（旧逻辑通过collectionId关联） - designService.parseMoodboardPosition(moodBoard.getMoodboardPosition(), compositeImage.getId()); - } - } - } - List moodBoards = moodBoard.getMoodBoards(); - if (CollectionUtil.isNotEmpty(moodBoards)) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(CollectionElement::getProjectId, projectId); - qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.MOOD_BOARD.getRealName()); - if (moodBoard.getMoodTemplateId() != null) { - qw.lambda().ne(CollectionElement::getId, moodBoard.getMoodTemplateId()); - } -// qw.lambda().ne(CollectionElement::getCollectionId, 0); - List collectionElements = collectionElementMapper.selectList(qw); - Set old = collectionElements.stream().map(CollectionElement::getId).collect(Collectors.toSet()); - - for (DesignCollectionElementDTO board : moodBoards) { - if (board.getDesignType().equals(DesignTypeEnum.LIBRARY.getRealName())) { - Library library = libraryMapper.selectById(board.getId()); - CollectionElement collectionElement = new CollectionElement(); - collectionElement.setAccountId(accountId); - collectionElement.setProjectId(projectId); - collectionElement.setLevel1Type(CollectionLevel1TypeEnum.MOOD_BOARD.getRealName()); - collectionElement.setName(library.getName()); - collectionElement.setUrl(library.getUrl()); - collectionElement.setHasPin((byte) 0); - collectionElement.setMd5(library.getMd5()); - collectionElement.setCreateDate(new Date()); - collectionElementMapper.insert(collectionElement); - }else if (board.getDesignType().equals(DesignTypeEnum.GENERATE.getRealName())) { - GenerateDetail generateDetail = generateDetailMapper.selectById(board.getId()); - CollectionElement collectionElement = new CollectionElement(); - collectionElement.setAccountId(accountId); - collectionElement.setProjectId(projectId); - collectionElement.setLevel1Type(CollectionLevel1TypeEnum.MOOD_BOARD.getRealName()); -// collectionElement.setName(generateDetail.get()); - collectionElement.setUrl(generateDetail.getUrl()); - collectionElement.setHasPin((byte) 0); - collectionElement.setMd5(generateDetail.getMd5()); - collectionElement.setCreateDate(new Date()); - collectionElementMapper.insert(collectionElement); - }else { - if (old.contains(board.getId())) { - old.remove(board.getId()); - }else { - CollectionElement collectionElement = collectionElementMapper.selectById(board.getId()); - collectionElement.setProjectId(projectId); - collectionElementMapper.updateById(collectionElement); - } - } - } - if (!CollectionUtils.isEmpty(old)) { - collectionElementMapper.deleteBatchIds(old); - } - } - } - if (!CollectionUtils.isEmpty(moduleSaveDTO.getPrintBoard())){ - List printBoards = moduleSaveDTO.getPrintBoard(); - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(CollectionElement::getProjectId, projectId); - qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.PRINT_BOARD.getRealName()); - List collectionElements = collectionElementMapper.selectList(qw); - Set old = collectionElements.stream().map(CollectionElement::getId).collect(Collectors.toSet()); - - for (DesignCollectionPrintElementDTO board : printBoards) { - if (board.getDesignType().equals(DesignTypeEnum.LIBRARY.getRealName())) { - Library library = libraryMapper.selectById(board.getId()); - CollectionElement collectionElement = new CollectionElement(); - collectionElement.setAccountId(accountId); - collectionElement.setProjectId(projectId); - collectionElement.setLevel1Type(CollectionLevel1TypeEnum.PRINT_BOARD.getRealName()); - collectionElement.setLevel2Type(board.getLevel2Type()); - collectionElement.setName(library.getName()); - collectionElement.setUrl(library.getUrl()); - collectionElement.setHasPin(board.getIsPin()); - collectionElement.setMd5(library.getMd5()); - collectionElement.setCreateDate(new Date()); - collectionElementMapper.insert(collectionElement); - }else if (board.getDesignType().equals(DesignTypeEnum.GENERATE.getRealName())) { - GenerateDetail generateDetail = generateDetailMapper.selectById(board.getId()); - CollectionElement collectionElement = new CollectionElement(); - collectionElement.setAccountId(accountId); - collectionElement.setProjectId(projectId); - collectionElement.setLevel1Type(CollectionLevel1TypeEnum.PRINT_BOARD.getRealName()); -// collectionElement.setLevel2Type(board.getLevel2Type()); -// collectionElement.setName(generateDetail.get()); - collectionElement.setUrl(generateDetail.getUrl()); - collectionElement.setHasPin(board.getIsPin()); - collectionElement.setMd5(generateDetail.getMd5()); - collectionElement.setCreateDate(new Date()); - collectionElementMapper.insert(collectionElement); - }else { - if (old.contains(board.getId())) { - CollectionElement collectionElement = collectionElementMapper.selectById(board.getId()); -// collectionElement.setLevel2Type(board.getLevel2Type()); - collectionElement.setHasPin(board.getIsPin()); - collectionElement.setUpdateDate(new Date()); - collectionElementMapper.updateById(collectionElement); - old.remove(board.getId()); - }else { - CollectionElement collectionElement = collectionElementMapper.selectById(board.getId()); - collectionElement.setProjectId(projectId); - collectionElement.setHasPin(board.getIsPin()); - collectionElement.setUpdateDate(new Date()); - collectionElementMapper.updateById(collectionElement); - } - } - } - if (!CollectionUtils.isEmpty(old)) { - collectionElementMapper.deleteBatchIds(old); - } - } - if (!CollectionUtils.isEmpty(moduleSaveDTO.getColorBoard())){ - List colorBoards = moduleSaveDTO.getColorBoard(); - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(CollectionElement::getProjectId, projectId); - qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.COLOR_BOARD.getRealName()); - List collectionElements = collectionElementMapper.selectList(qw); - Set old = collectionElements.stream().map(CollectionElement::getId).collect(Collectors.toSet()); - - for (CollectionColorDTO board : colorBoards) { - if (Objects.nonNull(board.getId()) && old.contains(Long.valueOf(board.getId()))) { - // ??? - old.remove(Long.valueOf(board.getId())); - }else { - CollectionElement collectionElement = new CollectionElement(); - collectionElement.setAccountId(accountId); - collectionElement.setProjectId(projectId); - collectionElement.setLevel1Type(CollectionLevel1TypeEnum.COLOR_BOARD.getRealName()); - collectionElement.setName(board.getName()); - collectionElement.setHasPin((byte) 0); - collectionElement.setColorRgb(board.getRgbValue()); - collectionElement.setMd5("0"); - // 渐变色 - if (!Objects.isNull(board.getGradient()) && !StringUtil.isNullOrEmpty(board.getGradient().getColorImg())){ - board.getGradient().setColorImg(null); - } - collectionElement.setGradientString(JSON.toJSONString(board.getGradient())); - collectionElement.setCreateDate(new Date()); - collectionElementMapper.insert(collectionElement); - } - } - if (!CollectionUtils.isEmpty(old)) { - collectionElementMapper.deleteBatchIds(old); - } - } - if (!CollectionUtils.isEmpty(moduleSaveDTO.getSketchBoard())){ - boundingBox = true; - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(CollectionElement::getProjectId, projectId); - qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.SKETCH_BOARD.getRealName()); - List collectionElements = collectionElementMapper.selectList(qw); - Set old = collectionElements.stream().map(CollectionElement::getId).collect(Collectors.toSet()); - List sketchBoards = moduleSaveDTO.getSketchBoard(); - for (CollectionSketchDTO board : sketchBoards) { - if (board.getDesignType().equals(DesignTypeEnum.LIBRARY.getRealName())) { - Library library = libraryMapper.selectById(board.getSketchBoardId()); - CollectionElement collectionElement = new CollectionElement(); - collectionElement.setAccountId(accountId); - collectionElement.setProjectId(projectId); - collectionElement.setLevel1Type(CollectionLevel1TypeEnum.SKETCH_BOARD.getRealName()); - collectionElement.setLevel2Type(board.getLevel2Type()); - collectionElement.setName(library.getName()); - collectionElement.setUrl(library.getUrl()); - collectionElement.setHasPin(board.getIsPin()); - collectionElement.setMd5(library.getMd5()); - collectionElement.setCreateDate(new Date()); - collectionElementMapper.insert(collectionElement); - }else if (board.getDesignType().equals(DesignTypeEnum.GENERATE.getRealName())) { - GenerateDetail generateDetail = generateDetailMapper.selectById(board.getSketchBoardId()); - CollectionElement collectionElement = new CollectionElement(); - collectionElement.setAccountId(accountId); - collectionElement.setProjectId(projectId); - collectionElement.setLevel1Type(CollectionLevel1TypeEnum.SKETCH_BOARD.getRealName()); - collectionElement.setLevel2Type(board.getLevel2Type()); -// collectionElement.setName(generateDetail.get()); - collectionElement.setUrl(generateDetail.getUrl()); - collectionElement.setHasPin(board.getIsPin()); - collectionElement.setMd5(generateDetail.getMd5()); - collectionElement.setCreateDate(new Date()); - collectionElementMapper.insert(collectionElement); - }else { - if (old.contains(board.getSketchBoardId())) { - CollectionElement collectionElement = collectionElementMapper.selectById(board.getSketchBoardId()); - collectionElement.setLevel2Type(board.getLevel2Type()); - collectionElement.setHasPin(board.getIsPin()); - collectionElement.setUpdateDate(new Date()); - collectionElementMapper.updateById(collectionElement); - old.remove(board.getSketchBoardId()); - }else { - CollectionElement collectionElement = collectionElementMapper.selectById(board.getSketchBoardId()); - collectionElement.setProjectId(projectId); - collectionElement.setLevel2Type(board.getLevel2Type()); - collectionElement.setHasPin(board.getIsPin()); - collectionElement.setUpdateDate(new Date()); - collectionElementMapper.updateById(collectionElement); - } - } - } - if (!CollectionUtils.isEmpty(old)) { - collectionElementMapper.deleteBatchIds(old); - } - } - if (Objects.nonNull(moduleSaveDTO.getPatternMaking3D())) { - PatternMaking3DDTO patternMaking3D = moduleSaveDTO.getPatternMaking3D(); - - QueryWrapper threeDModuleQueryWrapper = new QueryWrapper<>(); - threeDModuleQueryWrapper.lambda().eq(ThreeDModule::getProjectId, projectId); - List threeDModuleList = threeDModuleMapper.selectList(threeDModuleQueryWrapper); - - if (CollectionUtils.isEmpty(threeDModuleList)) { - ThreeDModule threeDModule = new ThreeDModule(); - threeDModule.setProjectId(projectId); - threeDModule.setThreeDSimpleId(patternMaking3D.getThreeDSimpleId()); - threeDModule.setX(patternMaking3D.getX()); - threeDModule.setY(patternMaking3D.getY()); -// threeDModule.setPrintMinioUrl(patternMaking3D.getPrintMinioUrl()); - threeDModule.setCollectionElementId(patternMaking3D.getCollectionElementId()); - threeDModuleMapper.insert(threeDModule); - }else { - ThreeDModule threeDModule = threeDModuleList.get(0); -// if (!Objects.equals(patternMaking3D.getThreeDSimpleId(), threeDModule.getThreeDSimpleId())) { -// -// } - threeDModule.setThreeDSimpleId(patternMaking3D.getThreeDSimpleId()); - threeDModule.setX(patternMaking3D.getX()); - threeDModule.setY(patternMaking3D.getY()); -// threeDModule.setPrintMinioUrl(patternMaking3D.getPrintMinioUrl()); - threeDModule.setCollectionElementId(patternMaking3D.getCollectionElementId()); - threeDModuleMapper.updateById(threeDModule); - } - } - if (CollectionUtil.isNotEmpty(moduleSaveDTO.getMannequin())) { - List collectionElementRelModels = collectionElementMapper.selectByProject(projectId); - List mannequin = moduleSaveDTO.getMannequin(); - List inputMannequinId = mannequin.stream().map(MannequinDTO::getId).collect(Collectors.toList()); - - // 创建relationId到Model的映射 - Map relationIdToModelMap = collectionElementRelModels.stream() - .collect(Collectors.toMap( - CollectionElementRelModel::getRelationId, - java.util.function.Function.identity(), - (existing, replacement) -> existing - )); - List newReletionIds = new ArrayList<>(); - List toBeDeletedRelationIds = new ArrayList<>(); - - if (collectionElementRelModels.isEmpty()) { - // 如果原本没有记录，所有输入的Mannequin都是新的 - newReletionIds = inputMannequinId; - } else { - // 获取所有已有的relationId集合 - Set relationIds = relationIdToModelMap.keySet(); - - // 1. 找出需要删除的：relationIds中有但inputMannequinId中没有的 - toBeDeletedRelationIds = relationIds.stream() - .filter(id -> !inputMannequinId.contains(id)) - .collect(Collectors.toList()); - - // 2. 找出新增的：inputMannequinId中有但relationIds中没有的 - newReletionIds = inputMannequinId.stream() - .filter(id -> !relationIds.contains(id)) - .collect(Collectors.toList()); - } - - if (!toBeDeletedRelationIds.isEmpty()){ - Set toBeDeletedElementIdList = toBeDeletedRelationIds.stream() - .map(relationIdToModelMap::get) // 获取对应的Model - .filter(Objects::nonNull) // 过滤掉可能为null的情况 - .map(CollectionElementRelModel::getCollectionElementId) // 提取collectionElementId - .collect(Collectors.toSet()); - collectionElementMapper.deleteBatchIds(toBeDeletedElementIdList); - // 删除关联关系 - deleteByCollectionElementIdList(toBeDeletedElementIdList); - } - - for (MannequinDTO dto : mannequin) { - if (newReletionIds.isEmpty()){ - break; - }else if (!newReletionIds.contains(dto.getId())){ - continue; - } - if (dto.getType().equals("System")) { - SysFile sysFile = sysFileMapper.selectById(dto.getId()); - - CollectionElement collectionElement = new CollectionElement(); - collectionElement.setAccountId(accountId); - collectionElement.setProjectId(projectId); - collectionElement.setLevel1Type(CollectionLevel1TypeEnum.MODEL.getRealName()); -// collectionElement.setLevel2Type(board.getLevel2Type()); - collectionElement.setName(sysFile.getName()); - collectionElement.setUrl(sysFile.getUrl()); -// collectionElement.setHasPin(board.getIsPin()); - collectionElement.setMd5(sysFile.getMd5()); - collectionElement.setCreateDate(new Date()); - collectionElement.setHasPin((byte) 0); - collectionElementMapper.insert(collectionElement); - CollectionElementRelModel collectionElementRelModel = new CollectionElementRelModel(); - collectionElementRelModel.setCollectionElementId(collectionElement.getId()); - collectionElementRelModel.setRelationId(dto.getId()); - collectionElementRelModel.setRelationType("System"); - collectionElementRelModelMapper.insert(collectionElementRelModel); - }else if (dto.getType().equals("Library")) { - Library library = libraryMapper.selectById(dto.getId()); - - CollectionElement collectionElement = new CollectionElement(); - collectionElement.setAccountId(accountId); - collectionElement.setProjectId(projectId); - collectionElement.setLevel1Type(CollectionLevel1TypeEnum.MODEL.getRealName()); -// collectionElement.setLevel2Type(board.getLevel2Type()); - collectionElement.setName(library.getName()); - collectionElement.setUrl(library.getUrl()); -// collectionElement.setHasPin(board.getIsPin()); - collectionElement.setMd5(library.getMd5()); - collectionElement.setCreateDate(new Date()); - collectionElement.setHasPin((byte) 0); - collectionElementMapper.insert(collectionElement); - CollectionElementRelModel collectionElementRelModel = new CollectionElementRelModel(); - collectionElementRelModel.setCollectionElementId(collectionElement.getId()); - collectionElementRelModel.setRelationId(dto.getId()); - collectionElementRelModel.setRelationType("System"); - collectionElementRelModelMapper.insert(collectionElementRelModel); - - } - } - } - if (boundingBox) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.SKETCH_BOARD.getRealName()); - qw.lambda().eq(CollectionElement::getProjectId, projectId); - List collectionElements = collectionElementMapper.selectList(qw); - - List voList = new ArrayList<>(); - for (CollectionElement collectionElement : collectionElements) { - CollectionSketchVO vo = CopyUtil.copyObject(collectionElement, CollectionSketchVO.class); - String url = collectionElement.getUrl(); - libraryService.processSketchBoards(url, collectionElement.getLevel2Type()); - if (url.contains(".")) { - String[] split = url.split("\\."); - vo.setUrlWithWhiteSide(minioUtil.getPreSignedUrl(split[0] + "-show." + split[1], 24 * 60, true)); - }else { - vo.setUrlWithWhiteSide(minioUtil.getPreSignedUrl(url + "-show", 24 * 60, true)); - } - vo.setIsPin(collectionElement.getHasPin()); - vo.setLevel2Type(collectionElement.getLevel2Type()); - vo.setDesignType("Collection"); - vo.setSketchBoardId(collectionElement.getId()); - voList.add(vo); - } - result.setBoundingBox(voList); - } - project.setUpdateTime(LocalDateTime.now()); - projectService.updateById(project); - return result; - } - - private void deleteByCollectionElementIdList(Set old) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().in(CollectionElementRelModel::getCollectionElementId, old); - collectionElementRelModelMapper.delete(qw); - } - - @Override - public QueryLibraryPageVO getMannequinDetail(MannequinDTO mannequinDTO) { - QueryLibraryPageVO vo = new QueryLibraryPageVO(); - if (mannequinDTO.getType().equals("System")) { - SysFile sysFile = sysFileMapper.selectById(mannequinDTO.getId()); - vo.setDesignType("System"); - vo.setLevel1Type(sysFile.getLevel1Type()); - vo.setLevel2Type(sysFile.getLevel2Type()); - vo.setName(sysFile.getName()); - vo.setMinIOPath(minioUtil.getPreSignedUrl(sysFile.getUrl(), 24 * 60)); - }else { - Library library = libraryMapper.selectById(mannequinDTO.getId()); - vo.setDesignType("Library"); - vo.setLevel1Type(library.getLevel1Type()); - vo.setLevel2Type(library.getLevel2Type()); - vo.setName(library.getName()); - vo.setMinIOPath(minioUtil.getPreSignedUrl(library.getUrl(), 24 * 60)); - } - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(LibraryModelPoint::getRelationId, mannequinDTO.getId()); - qw.lambda().eq(LibraryModelPoint::getModelType, mannequinDTO.getType()); - List libraryModelPoints = libraryModelPointMapper.selectList(qw); - if (CollectionUtil.isEmpty(libraryModelPoints)) { - throw new BusinessException("No model doting information available."); - } - LibraryModelPoint libraryModelPoint = libraryModelPoints.get(0); - LibraryModelPointVO libraryModelPointVO = CopyUtil.copyObject(libraryModelPoint, LibraryModelPointVO.class); - libraryModelPointVO.setTemplateId(libraryModelPoint.getId()); - libraryModelPointVO.setHandLeft(JSON.parseObject(libraryModelPoint.getHandLeft(), List.class)); - libraryModelPointVO.setHandRight(JSON.parseObject(libraryModelPoint.getHandRight(), List.class)); - libraryModelPointVO.setShoulderRight(JSON.parseObject(libraryModelPoint.getShoulderRight(), List.class)); - libraryModelPointVO.setShoulderLeft(JSON.parseObject(libraryModelPoint.getShoulderLeft(), List.class)); - libraryModelPointVO.setWaistbandRight(JSON.parseObject(libraryModelPoint.getWaistbandRight(), List.class)); - libraryModelPointVO.setWaistbandLeft(JSON.parseObject(libraryModelPoint.getWaistbandLeft(), List.class)); - - vo.setLibraryModelPoint(libraryModelPointVO); - return vo; - } - - @Override - public BrandLogoUploadVO brandLogoUpload(MultipartFile file) { - BrandLogoUploadVO vo = new BrandLogoUploadVO(); - AuthPrincipalVo userHolder = UserContext.getUserHolder(); - String path = userHolder.getId().toString() + "/brandLogo"; - vo.setBrandLogo(minioUtil.upload("aida-users",path,file)); - vo.setMinioUrl(minioUtil.getPreSignedUrl(vo.getBrandLogo(), 24 * 60)); - return vo; -// return null; - } - - @Override - public Boolean brandDNASaveOrUpdate(BrandDNADTO brandDNADTO) { - if (brandDNADTO.getId() != null) { - BrandDNA brandDNA = CopyUtil.copyObject(brandDNADTO, BrandDNA.class); - brandDNA.setUpdateTime(LocalDateTime.now()); - brandDNAMapper.updateById(brandDNA); - }else { - BrandDNA brandDNA = CopyUtil.copyObject(brandDNADTO, BrandDNA.class); - AuthPrincipalVo userHolder = UserContext.getUserHolder(); - brandDNA.setAccountId(userHolder.getId()); - brandDNA.setCreateTime(LocalDateTime.now()); - brandDNAMapper.insert(brandDNA); - } - return Boolean.TRUE; - } - - @Override - public LibraryUpdateVo brandDNAUpload(MultipartFile file, Long brandId) throws IOException { - AuthPrincipalVo userHolder = UserContext.getUserHolder(); - String path = userHolder.getId().toString() + "/brandLogo"; - String upload = minioUtil.upload("aida-users", path, file); - Library library = new Library(); - library.setAccountId(userHolder.getId()); - library.setLevel1Type(LibraryLevel1TypeEnum.BRAND_DNA.getRealName()); - library.setName(file.getOriginalFilename()); - library.setUrl(upload); - library.setMd5(MD5Utils.encryptFile(file.getInputStream())); - library.setCreateDate(new Date()); - libraryMapper.insert(library); - - BrandRelLibrary brandRelLibrary = new BrandRelLibrary(); - brandRelLibrary.setLibraryId(library.getId()); - brandRelLibrary.setBrandId(brandId); - brandRelLibraryMapper.insert(brandRelLibrary); - - LibraryUpdateVo vo = CopyUtil.copyObject(library, LibraryUpdateVo.class); - vo.setMinIOPath(minioUtil.getPreSignedUrl(vo.getUrl(), 24 * 60)); - - return vo; - } - - @Override - public PageBaseResponse brandDNAPage(BrandDNAQueryDTO brandDNAQueryDTO) { - AuthPrincipalVo userHolder = UserContext.getUserHolder(); - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(BrandDNA::getAccountId, userHolder.getId()); - IPage page = brandDNAMapper.selectPage(new Page<>(brandDNAQueryDTO.getPage(), brandDNAQueryDTO.getSize()), qw); - IPage convert = page.convert((Function) brand -> { - if (Objects.nonNull(brand)) { - BrandDNAVO brandDNAVO = CopyUtil.copyObject(brand, BrandDNAVO.class); - brandDNAVO.setMinioUrl(minioUtil.getPreSignedUrl(brandDNAVO.getBrandLogo(), 24 * 60)); - return brandDNAVO; - } - return null; - }); - return PageBaseResponse.success(convert); - } - - @Override - public BrandDNAGenerateVO brandDNAGenerate(String prompt) { - BrandDNAGenerateVO vo = new BrandDNAGenerateVO(); - JSONObject jsonObject = pythonService.brandDNAGenerate(prompt); - JSONObject data = jsonObject.getJSONObject("data"); - vo.setBrandSlogan(data.getString("brand_slogan")); - vo.setBrandName(data.getString("brand_name")); - vo.setBrandLogo(data.getString("brand_logo")); - vo.setMinioUrl(minioUtil.getPreSignedUrl(vo.getBrandLogo(), 24 * 60)); - return vo; - } - - @Override - public IPage getThreeDLayoutPage(ThreeDLayoutQueryDTO query) { -// AuthPrincipalVo authPrincipalVo = UserContext.getUserHolder(); -// Long workspaceServiceByProjectId = workspaceService.getByProjectId(query.getProjectId()); -// Workspace byId = workspaceService.getById(workspaceServiceByProjectId); - // 分页数据 - QueryWrapper queryWrapper = new QueryWrapper<>(); - queryWrapper.lambda().eq(ThreeDLayout::getType, "front"); - queryWrapper.lambda().eq(ThreeDLayout::getGender, query.getSex().toLowerCase()); - IPage page = threeDLayoutMapper.selectPage( - new Page<>(query.getPage(), query.getSize()), queryWrapper); - if (CollectionUtils.isEmpty(page.getRecords())) { - return new Page<>(); - } - - IPage convert = page.convert((Function) threeDLayout -> { - if (threeDLayout != null) { - ThreeDLayoutVO threeDLayoutVO = CopyUtil.copyObject(threeDLayout, ThreeDLayoutVO.class); - threeDLayoutVO.setUrl(minioUtil.getPreSignedUrl(threeDLayoutVO.getUrl(), 24 * 60)); - return threeDLayoutVO; - } - return null; - }); - return convert; - } - - @Override - public ThreeDVO getLayoutDetail(Long threeDSimpleId) { - ThreeDVO result = new ThreeDVO(); - - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(ThreeDLayout::getThreeDSimpleId, threeDSimpleId); - List threeDLayoutList = threeDLayoutMapper.selectList(qw); - for (ThreeDLayout threeDLayout : threeDLayoutList) { - threeDLayout.setUrl(minioUtil.getPreSignedUrl(threeDLayout.getUrl(), 24* 60)); - } - result.setThreeDLayoutList(threeDLayoutList); - - QueryWrapper threeDPatternLayoutQueryWrapper = new QueryWrapper<>(); - threeDPatternLayoutQueryWrapper.lambda().eq(ThreeDPatternLayout::getThreeDSimpleId, threeDSimpleId); - List threeDPatternLayoutList = threeDPatternLayoutMapper.selectList(threeDPatternLayoutQueryWrapper); - if (CollectionUtil.isNotEmpty(threeDPatternLayoutList)) { - ThreeDPatternLayout threeDPatternLayout = threeDPatternLayoutList.get(0); -// threeDPatternLayout.setUrl(minioUtil.getPreSignedUrl(threeDPatternLayout.getUrl(), 24 * 60)); - result.setThreeDPatternLayoutUrl(minioUtil.getPreSignedUrl(threeDPatternLayout.getUrl(), 24 * 60)); - } - ThreeDSimple threeDSimple = threeDSimpleMapper.selectById(threeDSimpleId); - result.setThreeDSimpleUrl(minioUtil.getPreSignedUrl(threeDSimple.getUrl(), 24 * 60)); - return result; - } - - @Override - public ThreeDSizeVO getThreeDSize(Long threeDSimpleId) { - ThreeDSizeVO result = new ThreeDSizeVO(); - QueryWrapper threeDDetailQueryWrapper = new QueryWrapper<>(); - threeDDetailQueryWrapper.lambda().eq(ThreeDDetail::getThreeDSimpleId, threeDSimpleId); - List threeDDetailList = threeDDetailMapper.selectList(threeDDetailQueryWrapper); - if (CollectionUtil.isNotEmpty(threeDDetailList)) { - // 以 sizeType 分组 - Map> sizeTypeMap = new HashMap<>(); - for (ThreeDDetail detail : threeDDetailList) { - sizeTypeMap - .computeIfAbsent(detail.getSizeType(), k -> new ArrayList<>()) - .add(detail.getSize()); - } - result.setSizeTypeMap(sizeTypeMap); - } - return result; - } - - @Override - public void getThreeDGlb(Long threeDSimpleId, HttpServletResponse response) throws MinioException, IOException { - ThreeDSimple threeDSimple = threeDSimpleMapper.selectById(threeDSimpleId); - if (ObjectUtils.isAllFieldNull(threeDSimple)) { - throw new BusinessException("3D file is not exist"); - } - InputStream inputStream = minioUtil.download(threeDSimple.getUrl()); - - // 设置响应头 - response.setContentType("model/gltf-binary"); // 确保 MIME 类型正确 - response.setHeader("Access-Control-Expose-Headers", "Content-Disposition"); - String fileName = threeDSimple.getName(); - if (!fileName.toLowerCase().endsWith(".glb")) { - fileName += ".glb"; - } - response.setHeader("Content-Disposition", "attachment; filename*=UTF-8''" + URLEncoder.encode(fileName, StandardCharsets.UTF_8.toString())); - // 将文件内容写入响应输出流 - try { - byte[] buffer = new byte[8192]; - int bytesRead; - while ((bytesRead = inputStream.read(buffer)) != -1) { - response.getOutputStream().write(buffer, 0, bytesRead); - } - inputStream.close(); - response.getOutputStream().flush(); - } catch (Exception e) { - throw new RuntimeException("Failed to get 3D .glb file", e); - } - } - - @Override - public String downloadZip(Long threeDSimpleId, String sizeType, String size, HttpServletResponse response) throws MinioException, IOException { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(ThreeDDetail::getThreeDSimpleId, threeDSimpleId); - qw.lambda().eq(ThreeDDetail::getSizeType, sizeType); - qw.lambda().eq(ThreeDDetail::getSize, size); - ThreeDDetail threeDDetail = threeDDetailMapper.selectOne(qw); - if (ObjectUtils.isAllFieldNull(threeDDetail)) { - throw new BusinessException("3D file is not exist"); - } - return minioUtil.getPreSignedUrl(threeDDetail.getUrl(), 24 * 60); - -// // 设置响应头 -// response.setContentType("application/zip"); // 确保 ZIP 格式 -// response.setHeader("Access-Control-Expose-Headers", "Content-Disposition"); -// String fileName = threeDDetail.getName(); -// if (!fileName.toLowerCase().endsWith(".zip")) { -// fileName += ".zip"; -// } -// response.setHeader("Content-Disposition", "attachment; filename*=UTF-8''" + URLEncoder.encode(fileName, StandardCharsets.UTF_8.toString())); -// -// // 将文件内容写入响应输出流 -// try { -// byte[] buffer = new byte[8192]; -// int bytesRead; -// while ((bytesRead = inputStream.read(buffer)) != -1) { -// response.getOutputStream().write(buffer, 0, bytesRead); -// } -// inputStream.close(); -// response.getOutputStream().flush(); -// } catch (Exception e) { -// throw new RuntimeException("Failed to download ZIP file", e); -// } - } - - @Override - public Boolean delete(Long projectId) { - projectService.removeById(projectId); - return Boolean.TRUE; - } - - @Override - public Boolean brandDNADelete(BrandDNADTO brandDNADTO) { - brandDNAMapper.deleteById(brandDNADTO.getId()); - return Boolean.TRUE; - } - - @Override - @Transactional(isolation = Isolation.SERIALIZABLE) - public void toProductBatch(String taskId, String url, String progress) throws InterruptedException { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(ToProductImageResult::getTaskId, taskId); - ToProductImageResult toProductImageResult = toProductImageResultMapper.selectOne(qw); - if (Objects.nonNull(toProductImageResult)) { - toProductImageResult.setUrl(url); - toProductImageResult.setStatus("Success"); - toProductImageResultMapper.updateById(toProductImageResult); - String taskIdBatch = toProductImageResult.getTaskIdBatch(); - log.info("toProductImage云生成 batchTaskId:" + taskIdBatch + " 完成数+1"); - - CloudTask task = cloudTaskMapper.selectForUpdate(taskIdBatch); - log.info("排查完成数量" + task.getCompletedNum()); - task.setCompletedNum(task.getCompletedNum() + 1); - cloudTaskMapper.updateById(task); - - String key = toProductImageResultKey + ":" + taskId; - String imageName = url.substring(url.lastIndexOf("/") + 1); - String status = imageName.equals("white_image.jpg") ? "Invalid" : "Success"; - GenerateResultVO generateResultVO = new GenerateResultVO(taskId, toProductImageResult.getId(), url, status, null); - redisUtil.addToString(key, new Gson().toJson(generateResultVO), CommonConstant.GENERATE_RESULT_EXPIRE_TIME); - - /*Long accountId = Long.parseLong(taskId.substring(taskId.lastIndexOf("-") + 1)); - if (!status.equals("Invalid")) { - // 4、扣除积分 - Boolean b = creditsService.taskCreditsDeduction(accountId, taskId); - // 3、记录积分变更 - if (b) creditsService.updateChangedCredits(String.valueOf(accountId), taskId); - }*/ - } - - } - - @Override - @Transactional - public void relightBatch(String taskId, String url, String progress) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(ToProductImageResult::getTaskId, taskId); - ToProductImageResult toProductImageResult = toProductImageResultMapper.selectOne(qw); - if (Objects.nonNull(toProductImageResult)) { - if (toProductImageResult.getBrightenValue() != null && toProductImageResult.getBrightenValue() != 1.0) { - pythonService.bright(url, toProductImageResult.getBrightenValue()); - } - toProductImageResult.setUrl(url); -// toProductImageResult.setResultType("Relight"); - toProductImageResult.setStatus("Success"); - toProductImageResultMapper.updateById(toProductImageResult); - - String taskIdBatch = toProductImageResult.getTaskIdBatch(); -// QueryWrapper cloudTaskQueryWrapper = new QueryWrapper<>(); -// cloudTaskQueryWrapper.lambda().eq(CloudTask::getTaskId, taskIdBatch); -// CloudTask cloudTask = cloudTaskMapper.selectOne(cloudTaskQueryWrapper); -// if (Objects.nonNull(cloudTask)) { -// if (cloudTask.getCompletedNum() == null) { -// cloudTask.setCompletedNum(1); -// }else { -// cloudTask.setCompletedNum(cloudTask.getCompletedNum() + 1); -// } -// if (progress.equals("OK")) { -// cloudTask.setStatus(1); -// cloudTask.setCompletedNum(cloudTask.getNums()); -// } -// cloudTaskMapper.updateById(cloudTask); -// } - cloudTaskMapper.increaseCompletedNum(taskIdBatch); - - String key = relightResultKey + ":" + taskId; - String imageName = url.substring(url.lastIndexOf("/") + 1); - String status = imageName.equals("white_image.jpg") ? "Invalid" : "Success"; - GenerateResultVO generateResultVO = new GenerateResultVO(taskId, toProductImageResult.getId(), url, status, null); - redisUtil.addToString(key, new Gson().toJson(generateResultVO), CommonConstant.GENERATE_RESULT_EXPIRE_TIME); - - /*Long accountId = Long.parseLong(taskId.substring(taskId.lastIndexOf("-") + 1)); - if (!status.equals("Invalid")) { - // 4、扣除积分 - Boolean b = creditsService.taskCreditsDeduction(accountId, taskId); - // 3、记录积分变更 - if (b) creditsService.updateChangedCredits(String.valueOf(accountId), taskId); - }*/ - } - } - - @Override - @Transactional - public Boolean collectionLikeUpdate(CollectionLikeUpdateDTO collectionLikeUpdateDTO) { - QueryWrapper qw = new QueryWrapper<>(); - qw.lambda().eq(CollectionSort::getRelationId, collectionLikeUpdateDTO.getOldRelationId()); - qw.lambda().eq(CollectionSort::getRelationType, collectionLikeUpdateDTO.getRelationType()); - CollectionSort collectionSort = collectionSortMapper.selectOne(qw); -// Long relationIdOld = collectionSort.getRelationId(); - collectionSort.setRelationId(collectionLikeUpdateDTO.getRelationId()); - collectionSortMapper.updateById(collectionSort); - if (collectionLikeUpdateDTO.getRelationType().equals(CollectionType.TO_PRODUCT_IMAGE.getValue()) || collectionLikeUpdateDTO.getRelationType().equals(CollectionType.RELIGHT.getValue())) { - ToProductImageResult toProductImageResult = toProductImageResultMapper.selectById(collectionLikeUpdateDTO.getRelationId()); - toProductImageResult.setIsLike(1); - toProductImageResultMapper.updateById(toProductImageResult); - ToProductImageResult toProductImageResultOld = toProductImageResultMapper.selectById(collectionLikeUpdateDTO.getOldRelationId()); - toProductImageResultOld.setIsLike(0); - toProductImageResultMapper.updateById(toProductImageResultOld); - }else if (collectionLikeUpdateDTO.getRelationType().equals(CollectionType.POSE_TRANSFORM.getValue())) { - PoseTransformation poseTransformation = poseTransformationMapper.selectById(collectionLikeUpdateDTO.getRelationId()); - poseTransformation.setIsLiked((byte) 1); - poseTransformationMapper.updateById(poseTransformation); - PoseTransformation poseTransformationOld = poseTransformationMapper.selectById(collectionLikeUpdateDTO.getOldRelationId()); - poseTransformationOld.setIsLiked((byte) 0); - poseTransformationMapper.updateById(poseTransformationOld); - } - return Boolean.TRUE; - } - - @Override - public Boolean toProductImageElementDelete(Long id) { - UpdateWrapper updateWrapper = new UpdateWrapper<>(); - updateWrapper - .set("project_id", null) // 显式设置 NULL - .eq("id", id); // 指定条件 - toProductElementMapper.update(null, updateWrapper); - return Boolean.TRUE; - } - - @Override - public ToProductElementVO convertRelightElement(Long id) { - ToProductImageResult toProductImageResult = toProductImageResultMapper.selectById(id); - - ToProductElement toProductElement = new ToProductElement(); - toProductElement.setUrl(toProductImageResult.getUrl()); -// toProductElement.setUserLikeGroupId(userLikeGroupId); - toProductElement.setProjectId(toProductImageResult.getProjectId()); - toProductElement.setCreateTime(LocalDateTime.now()); - toProductElementMapper.insert(toProductElement); - ToProductElementVO toProductElementVO = CopyUtil.copyObject(toProductElement, ToProductElementVO.class); - toProductElementVO.setMinioUrl(toProductElementVO.getUrl()); - toProductElementVO.setUrl(minioUtil.getPreSignedUrl(toProductElementVO.getUrl(), 24 * 60)); - return toProductElementVO; - } -} +package com.ai.da.service.impl; + +import cn.hutool.core.collection.CollectionUtil; +import com.ai.da.common.config.exception.BusinessException; +import com.ai.da.common.constant.CommonConstant; +import com.ai.da.common.context.UserContext; +import com.ai.da.common.enums.CollectionLevel1TypeEnum; +import com.ai.da.common.enums.CreditsEventsEnum; +import com.ai.da.common.enums.DesignTypeEnum; +import com.ai.da.common.enums.LibraryLevel1TypeEnum; +import com.ai.da.common.response.PageBaseResponse; +import com.ai.da.common.response.ResultEnum; +import com.ai.da.common.utils.*; +import com.ai.da.mapper.primary.*; +import com.ai.da.mapper.primary.entity.*; +import com.ai.da.mapper.primary.entity.Collection; +import com.ai.da.mapper.secondary.AttributeRetrievalMapper; +import com.ai.da.mapper.secondary.entity.AttributeRecognitionJSON; +import com.ai.da.model.dto.*; +import com.ai.da.model.enums.CollectionType; +import com.ai.da.model.enums.Module; +import com.ai.da.model.vo.*; +import com.ai.da.python.PythonService; +import com.ai.da.service.*; +import com.alibaba.fastjson.JSON; +import com.alibaba.fastjson.JSONArray; +import com.alibaba.fastjson.JSONObject; +import com.alibaba.fastjson.serializer.SerializerFeature; +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.google.common.base.Function; +import com.google.gson.Gson; +import io.minio.errors.MinioException; +import io.netty.util.internal.StringUtil; +import lombok.extern.slf4j.Slf4j; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Isolation; +import org.springframework.transaction.annotation.Transactional; +import org.springframework.util.CollectionUtils; +import org.springframework.util.StringUtils; +import org.springframework.web.multipart.MultipartFile; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletResponse; +import java.io.*; +import java.net.URLEncoder; +import java.nio.charset.StandardCharsets; +import java.time.Duration; +import java.time.LocalDateTime; +import java.time.ZoneId; +import java.util.*; +import java.util.stream.Collectors; + +/** + * 服务实现类 + * + * @author yanglei + * @since 2022-09-30 + */ +@Slf4j +@Service +public class UserLikeGroupServiceImpl extends ServiceImpl implements UserLikeGroupService { + @Resource + private UserLikeGroupMapper userLikeGroupMapper; + @Resource + private AccountService accountService; + @Resource + private CollectionService collectionService; + @Resource + private UserLikeService userLikeService; + @Resource + private WorkspaceService workspaceService; + @Resource + private UserLikeMapper userLikeMapper; + @Resource + private MinioUtil minioUtil; + @Resource + private TDesignPythonOutfitMapper designPythonOutfitMapper; + @Resource + private DesignMapper designMapper; + @Resource + private DesignService designService; + @Resource + private SysFileMapper sysFileMapper; + @Resource + private LibraryMapper libraryMapper; + @Resource + private GenerateDetailMapper generateDetailMapper; + @Resource + private PortfolioMapper portfolioMapper; + @Resource + private TagsMapper tagsMapper; + @Resource + private DesignItemDetailService designItemDetailService; + @Resource + private CollectionElementMapper collectionElementMapper; + @Resource + private AttributeRetrievalMapper attributeRetrievalMapper; + @Resource + private ProductImageAttributeMapper productImageAttributeMapper; + @Resource + private CollectionSortMapper collectionSortMapper; + @Resource + private ClassificationService classificationService; + @Resource +// private ProjectMapper projectMapper; + private ProjectService projectService; + @Resource + private PortfolioService portfolioService; + @Resource + private LibraryModelPointMapper libraryModelPointMapper; + @Resource + private LibraryService libraryService; + @Resource + private BrandDNAMapper brandDNAMapper; + @Resource + private BrandRelLibraryMapper brandRelLibraryMapper; + @Resource + private ThreeDLayoutMapper threeDLayoutMapper; + @Resource + private ThreeDPatternLayoutMapper threeDPatternLayoutMapper; + @Resource + private ThreeDDetailMapper threeDDetailMapper; + @Resource + private ThreeDSimpleMapper threeDSimpleMapper; + @Resource + private ThreeDModuleMapper threeDModuleMapper; + @Resource + private ProductImageService productImageService; + @Resource + private CollectionElementRelModelMapper collectionElementRelModelMapper; + @Resource + private CollectionSortService collectionSortService; + @Resource + private GenerateService generateService; + @Resource + private ToProductElementMapper toProductElementMapper; + @Resource + private ToProductImageRecordMapper toProductImageRecordMapper; + @Resource + private ToProductImageResultMapper toProductImageResultMapper; + @Resource + private CloudTaskMapper cloudTaskMapper; + @Resource + private PythonService pythonService; + @Resource + private CreditsService creditsService; + @Resource + private PoseTransformationMapper poseTransformationMapper; + @Resource + private ExportFileMapper exportFileMapper; + @Resource + private DesignItemDetailCanvasMapper designItemDetailCanvasMapper; + + @Override + public void deleteUserGroup(Long userGroupId) { + UserLikeGroup group = getById(userGroupId); + if (Objects.isNull(group)) { + throw new BusinessException("history.not.found"); + } + userLikeGroupMapper.deleteById(userGroupId); + } + + @Override + public HistoryUpdateVO updateUserGroupName(Long userGroupId, String userGroupName, String timeZone) { + UserLikeGroup group = getById(userGroupId); + if (Objects.isNull(group)) { + throw new BusinessException("history.not.found"); + } + QueryWrapper queryWrapper = new QueryWrapper<>(); + queryWrapper.eq("id", userGroupId); + UserLikeGroup userLikeGroup = new UserLikeGroup(); + userLikeGroup.setName(userGroupName); + userLikeGroup.setUpdateDate(DateUtil.getByTimeZone(timeZone)); + log.info("跟新group名字###{}###时间是###{}", userLikeGroup.getUpdateDate(), TimeZone.getDefault()); + userLikeGroupMapper.update(userLikeGroup, queryWrapper); + return new HistoryUpdateVO(userLikeGroup.getName(), userLikeGroup.getUpdateDate().getTime()); + } + +// public static void main(String[] args) { +// System.out.println(TimeZone.getDefault()); +// Date kolkata =DateUtil.getByTimeZone("Asia/Kolkata"); +// Date shanghai =DateUtil.getByTimeZone("Asia/Shanghai"); +// System.out.println(kolkata); +// System.out.println(shanghai); +// System.out.println(kolkata.getTime()); +// System.out.println(shanghai.getTime()); +// System.out.println(kolkata.getTime() == shanghai.getTime()); +// } + + private List selectByUserId(Long userId) { + QueryWrapper queryWrapper = new QueryWrapper<>(); + queryWrapper.eq("accountId", userId); + return userLikeGroupMapper.selectList(queryWrapper); + } + + @Override + public Long insertUserGroup(Long userId, Long collectionId, String timeZone, Long projectId) { + UserLikeGroup userLikeGroup = new UserLikeGroup(); + //默认当前时间 + userLikeGroup.setName(DateUtil.dateToStr(new Date(), DateUtil.YYYY_MM_DD_HH_MM_SS)); + userLikeGroup.setCreateDate(DateUtil.getByTimeZone(timeZone)); + userLikeGroup.setUpdateDate(DateUtil.getByTimeZone(timeZone)); + userLikeGroup.setAccountId(userId); + userLikeGroup.setCollectionId(collectionId); + userLikeGroup.setOriginal(1); + userLikeGroup.setProjectId(projectId); + userLikeGroupMapper.insertUserLikeGroup(userLikeGroup); + return userLikeGroup.getId(); + } + + @Override + public UserLikeChooseVO choose(Long userGroupId) { + UserLikeGroup group = getById(userGroupId); + if (Objects.isNull(group)) { + throw new BusinessException("history.not.found"); + } + List userLikeVOS = userLikeService.getGroupDetail(userGroupId); + String sex = null; + + QueryWrapper userLikeSortQw = new QueryWrapper<>(); + userLikeSortQw.lambda().eq(CollectionSort::getUserLikeGroupId, userGroupId); + List userLikeSortList = collectionSortMapper.selectList(userLikeSortQw); + if (CollectionUtil.isEmpty(userLikeSortList)) { + Integer sort = 1; + for (UserLikeVO userLikeVO : userLikeVOS) { + CollectionSort userLikeSort = new CollectionSort(); + userLikeSort.setUserLikeId(userLikeVO.getId()); + userLikeSort.setUserLikeGroupId(userGroupId); + userLikeSort.setSort(sort); + collectionSortMapper.insert(userLikeSort); + sort ++; + } + } + + userLikeVOS.forEach(o -> { + TDesignPythonOutfit tDesignPythonOutfit1 = designPythonOutfitMapper.selectById(o.getDesignOutfitId()); + o.setUrl(tDesignPythonOutfit1.getDesignUrl()); + if (o.getUrl().contains("/")) { + int index = o.getUrl().lastIndexOf("/"); + o.setPictureName(o.getUrl().substring(index + 1)); + } + o.setDesignOutfitUrl(minioUtil.getPreSignedUrl(o.getUrl(), 24 * 60)); + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(TDesignPythonOutfit::getDesignItemId, o.getDesignItemId()); + List tDesignPythonOutfits = designPythonOutfitMapper.selectList(qw); + if (CollectionUtil.isNotEmpty(tDesignPythonOutfits)) { + TDesignPythonOutfit tDesignPythonOutfit = tDesignPythonOutfits.get(0); + o.setDesignOutfitId(tDesignPythonOutfit.getId()); + } + + QueryWrapper userLikeSortQueryWrapper = new QueryWrapper<>(); + userLikeSortQueryWrapper.lambda().eq(CollectionSort::getUserLikeId, o.getId()); + List userLikeSorts = collectionSortMapper.selectList(userLikeSortQueryWrapper); + if (CollectionUtil.isNotEmpty(userLikeSorts)) { + CollectionSort userLikeSort = userLikeSorts.get(0); + o.setSort(userLikeSort.getSort()); + o.setUserLikeSortId(userLikeSort.getId()); + } + }); + UserLikeCollectionVO userLikeCollection = collectionService.chooseCollection(group.getCollectionId()); + Integer beenPublished = 0; + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(Portfolio::getUserLikeGroupSourceId, userGroupId); + List portfolios = portfolioMapper.selectList(qw); +// Portfolio portfolio = new Portfolio(); + PortfolioDTO portfolioDTO = new PortfolioDTO(); + if (CollectionUtil.isNotEmpty(portfolios)) { +// portfolio = portfolios.get(0); + portfolioDTO = CopyUtil.copyObject(portfolios.get(0), PortfolioDTO.class); + beenPublished = 1; + portfolioDTO.setTagsDTO(tagsMapper.getTagByPortfolioId(portfolioDTO.getId())); + } + return new UserLikeChooseVO(userGroupId, userLikeVOS, userLikeCollection, sex, beenPublished, portfolioDTO); + } + + @Override + public ProjectChooseVO choose(ProjectDTO projectDTO) { + + Project project = projectService.getById(projectDTO.getId()); + + Long workspaceId = workspaceService.getByProjectId(project.getId()); + WorkspaceVO workspaceVO = new WorkspaceVO(); + if (null != workspaceId) { + workspaceVO = workspaceService.getByIdForProject(workspaceId); + } + + String process = project.getProcess(); + String name = project.getName(); + + UserLikeGroup userLikeGroup = userLikeService.getUserLikeGroupByProjectId(project.getId()); + Long collectionId = null; + if (Objects.nonNull(userLikeGroup)) { + Collection collection = collectionService.getById(userLikeGroup.getCollectionId()); + if (Objects.nonNull(collection)) { + collectionId = collection.getId(); + } + } + + Integer beenPublished = 0; + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(Portfolio::getProjectId, projectDTO.getId()); + List portfolios = portfolioMapper.selectList(qw); +// Portfolio portfolio = new Portfolio(); + PortfolioDTO portfolioDTO = new PortfolioDTO(); + if (CollectionUtil.isNotEmpty(portfolios)) { +// portfolio = portfolios.get(0); + portfolioDTO = CopyUtil.copyObject(portfolios.get(0), PortfolioDTO.class); + beenPublished = 1; + portfolioDTO.setTagsDTO(tagsMapper.getTagByPortfolioId(portfolioDTO.getId())); + } + + return new ProjectChooseVO(projectDTO.getId(), name, workspaceVO, process, collectionId, beenPublished, portfolioDTO); + } + + @Override + public UserLikeGroup getByProjectId(Long projectId) { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(UserLikeGroup::getProjectId, projectId); + List userLikeGroupList = userLikeGroupMapper.selectList(qw); + if (CollectionUtil.isEmpty(userLikeGroupList)) { + return null; + } + return userLikeGroupList.get(0); + } + + @Override + public void deleteTrialData(Long userId) { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(UserLikeGroup::getAccountId, userId); + List userLikeGroupIdList = userLikeGroupMapper.selectList(qw).stream() + .map(UserLikeGroup::getId) + .collect(Collectors.toList()); + userLikeGroupMapper.delete(qw); + if (CollectionUtil.isNotEmpty(userLikeGroupIdList)) { + QueryWrapper userLikeQueryWrapper = new QueryWrapper<>(); + userLikeQueryWrapper.lambda().in(UserLike::getUserLikeGroupId, userLikeGroupIdList); + userLikeMapper.delete(userLikeQueryWrapper); + } + } + + @Override + @Transactional(rollbackFor = Exception.class) + public void updateDate(Long id,String timeZone){ + UpdateWrapper uw = new UpdateWrapper<>(); + uw.eq("id",id); + uw.set("update_date",DateUtil.getByTimeZone(timeZone)); + + baseMapper.update(null,uw); + } + + @Override + public Long exportSave(MultipartFile file, Long projectId, String module, Long designItemDetailId) { + AuthPrincipalVo userHolder = UserContext.getUserHolder(); + if (Objects.isNull(projectId) && !module.equals("designItemDetail")){ + log.warn("用户 {} 保存 {} 模块的画布时, projectId为空", userHolder.getId(), module); + throw new BusinessException("project id cannot be empty", ResultEnum.PROMPT.getCode()); + } + + QueryWrapper qw = new QueryWrapper<>(); + List exportFiles = null; + if (Objects.nonNull(projectId)){ + qw.lambda().eq(ExportFile::getProjectId, projectId); + qw.lambda().eq(ExportFile::getModule, module); + exportFiles = exportFileMapper.selectList(qw); + } else if (Objects.nonNull(designItemDetailId)){ + DesignItemDetailCanvas designItemDetailCanvas = designItemDetailService.getDIDCByDesignItemDetailId(designItemDetailId); + if (Objects.nonNull(designItemDetailCanvas) && Objects.nonNull(designItemDetailCanvas.getExportFileId())){ + qw.lambda().eq(ExportFile::getId, designItemDetailCanvas.getExportFileId()); + qw.lambda().eq(ExportFile::getModule, module); + exportFiles = exportFileMapper.selectList(qw); + } + } +// String upload = minioUtil.upload("aida-users", userHolder.getId() + "/exportFile", file); + + ExportFile exportFile = null; + if (CollectionUtil.isNotEmpty(exportFiles)) { + exportFile = exportFiles.get(0); + // 更新画布信息时，只重新上传图片，不用修改数据库url + String url = exportFile.getUrl(); + String path = url.substring(url.indexOf("/") + 1); + minioUtil.upload("aida-users", path, file, null); +// exportFile.setUrl(upload); +// exportFileMapper.updateById(exportFile); + }else { + String upload = minioUtil.upload("aida-users", userHolder.getId() + "/exportFile", file); + exportFile = new ExportFile(); + exportFile.setProjectId(projectId); + exportFile.setModule(module); + exportFile.setUrl(upload); + exportFileMapper.insert(exportFile); + if (Objects.nonNull(designItemDetailId)){ + designItemDetailService.saveDesignItemDetailCanvas(designItemDetailId, exportFile.getId()); + } + } + return exportFile.getId(); + } + + @Override + @Transactional(rollbackFor = Exception.class) + public List toProduct(ToProductImageDTO toProductImageDTO) { + // 判断用户当前积分是否够本次生成消耗 + boolean fluxTask = !StringUtil.isNullOrEmpty(toProductImageDTO.getModelName()) + && toProductImageDTO.getModelName().equals("flux"); + CreditsEventsEnum creditsEventsEnum = fluxTask ? CreditsEventsEnum.TO_PRODUCT_IMAGE_FLUX : CreditsEventsEnum.TO_PRODUCT_IMAGE; + Boolean preDeduction = creditsService.creditsPreDeduction(creditsEventsEnum, toProductImageDTO.getToProductImageVOList().size()); + if (!preDeduction) { + throw new BusinessException("Your remaining credits are insufficient for this generation. Please recharge.", ResultEnum.WARNING.getCode()); + } + + AuthPrincipalVo userHolder = UserContext.getUserHolder(); + Long projectId = toProductImageDTO.getProjectId(); + UserLikeGroup userLikeGroup = getByProjectId(projectId); + Long userLikeGroupId = null; + ToProductImageRecord toProductImageRecord = new ToProductImageRecord(); + toProductImageRecord.setProjectId(projectId); + if (Objects.nonNull(userLikeGroup)) { + userLikeGroupId = userLikeGroup.getId(); + toProductImageRecord.setUserLikeGroupId(userLikeGroupId); + } + + // 翻译 + String prompt = toProductImageDTO.getPrompt(); + StringBuilder sb = new StringBuilder("The best quality, masterpiece, real image."); + if (!StringUtil.isNullOrEmpty(prompt)) { + prompt = pythonService.promptTranslate(prompt); + + String[] words = prompt.split("\\s+"); + if (words.length > 180) { + throw new BusinessException("Please keep your input text under 200 words. Thanks!"); + } + } + + toProductImageRecord.setCreateTime(LocalDateTime.now()); + if (!StringUtils.isEmpty(toProductImageDTO.getPrompt())) { + toProductImageRecord.setPrompt(toProductImageDTO.getPrompt()); + } + toProductImageRecordMapper.insert(toProductImageRecord); + + List result = new ArrayList<>(); + boolean childFlag = !StringUtil.isNullOrEmpty(toProductImageDTO.getAgeGroup()) + && toProductImageDTO.getAgeGroup().equals("Child"); + int i = 0; + +// else { +// s = "best quality, masterpiece. detailed, high-res, simple background, studio photography, extremely detailed, updo, detailed face, face, close-up, HDR, UHD, 8K realistic, Highly detailed, simple background, Studio lighting"; +// } + for (ToProductImageVO toProductImageVO : toProductImageDTO.getToProductImageVOList()) { + String taskId; + if (toProductImageVO.getElementType().equals("DesignOutfit")) { + taskId = UUID.randomUUID() + "-" + i + "-" + userHolder.getId(); + TDesignPythonOutfit tDesignPythonOutfit = designPythonOutfitMapper.selectById(toProductImageVO.getElementId()); + + Long designItemId = tDesignPythonOutfit.getDesignItemId(); + QueryWrapper designItemDetailQueryWrapper = new QueryWrapper<>(); + designItemDetailQueryWrapper.lambda().eq(DesignItemDetail::getDesignItemId, designItemId); + designItemDetailQueryWrapper.lambda().ne(DesignItemDetail::getType, "Body"); + List designItemDetails = designItemDetailService.list(designItemDetailQueryWrapper); + String collect = designItemDetails.stream().map(DesignItemDetail::getType).collect(Collectors.joining(",")); + + Long designId = tDesignPythonOutfit.getDesignId(); + Design design = designMapper.selectById(designId); + String productType = "overall"; + if (design.getSingleOverall().equals("single")) { + productType = "single"; + sb.append(collect); + }else { + if (collect.contains("Tops") && childFlag) { + sb.append("a handsome boy,"); + }else if (collect.contains("Tops") && !childFlag){ + sb.append("a handsome man,"); + }else if (childFlag) { + sb.append("a beautiful girl,"); + }else { + sb.append("a beautiful woman,"); + } + sb.append("wearing ").append(collect); + } + if (StringUtils.isEmpty(prompt)) { + sb.append(",high quality clothing details,8K realistic,HDR"); + }else { + sb.append(",high quality clothing details,").append(prompt).append(",8K realistic,HDR"); + } + ToProductImageResultVO toProductImageResult = new ToProductImageResultVO(); + if (fluxTask){ + if (childFlag){ + sb.append(", Children's face"); + } + taskId = generateService.flux(CreditsEventsEnum.TO_PRODUCT_IMAGE_FLUX, sb.toString(), tDesignPythonOutfit.getDesignUrl(), childFlag); + toProductImageResult.setModelName("flux"); + toProductImageResult.setResultType(CollectionType.TO_PRODUCT_IMAGE.getValue()); + } else { + // 走模型 + if (childFlag){ + sb.append(", (Children's face:1.3)"); + } + pythonService.toProductImage(tDesignPythonOutfit.getDesignUrl(), taskId, sb.toString(), toProductImageDTO.getImageStrength(), productType); +// toProductImageResult.setModelName("local"); + } + + toProductImageResult.setElementId(tDesignPythonOutfit.getId()); + toProductImageResult.setElementType("DesignOutfit"); + toProductImageResult.setCreateTime(LocalDateTime.now()); + toProductImageResult.setToProductImageRecordId(toProductImageRecord.getId()); +// toProductImageResult.setUrl(productImageUrl); + toProductImageResult.setIsLike(0); + toProductImageResult.setTaskId(taskId); + toProductImageResult.setProjectId(projectId); + if (userLikeGroupId != null) { + toProductImageResult.setUserLikeGroupId(userLikeGroupId); + } + toProductImageResult.setImageStrength(toProductImageDTO.getImageStrength()); + toProductImageResult.setResultType(CollectionType.TO_PRODUCT_IMAGE.getValue()); + toProductImageResult.setStatus("Pending"); + toProductImageResultMapper.insert(toProductImageResult); +// toProductImageResult.setUrl(minioUtil.getPresignedUrl(toProductImageResult.getUrl(), 24 * 60)); + // 先判断是否需要默认like + if (Objects.nonNull(toProductImageDTO.getIsDefaultLike()) && toProductImageDTO.getIsDefaultLike()){ + // 满足条件情况下默认添加到like + CollectionSort collectionSort = addToProductLike(toProductImageVO.getParentId(), toProductImageResult.getId(), toProductImageDTO.getProjectId()); + // 重新排序 + Integer reSort = collectionSortService.rearrangeChildSort(toProductImageResult.getId(), CollectionType.TO_PRODUCT_IMAGE.getValue(), + toProductImageVO.getParentId(), toProductImageVO.getUserLikeSortId()); + // 将生成结果的排序返回 + toProductImageResult.setSort(Objects.isNull(reSort) ? Objects.isNull(collectionSort) ? null : collectionSort.getSort() : reSort); + toProductImageResult.setParentId(toProductImageVO.getParentId()); + toProductImageResult.setUserLikeSortId(Objects.isNull(collectionSort) ? null : collectionSort.getId()); + }else if (Objects.nonNull(toProductImageDTO.getIsDefaultLike()) && Objects.nonNull(toProductImageVO.getParentId())){ + toProductImageResult.setParentId(toProductImageVO.getParentId()); + // 默认不添加到like,但是需要有parentId,所以这里添加到collectionSort表中 +// designService.addCollectionSort(toProductImageResult.getId(), CollectionType.TO_PRODUCT_IMAGE.getValue(), toProductImageDTO.getProjectId(), toProductImageVO.getParentId()); + } + result.add(toProductImageResult); + }else { + if (StringUtils.isEmpty(prompt)) { + sb.append(",high quality clothing details,8K realistic,HDR"); + }else { + sb.append(",high quality clothing details,").append(prompt).append(",8K realistic,HDR"); + } + + taskId = UUID.randomUUID() + "-" + i + "-" + userHolder.getId(); + ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageVO.getElementId()); + ToProductImageResultVO toProductImageResult = new ToProductImageResultVO(); + + if (fluxTask){ + taskId = generateService.flux(creditsEventsEnum, sb.toString(), toProductElement.getUrl(), childFlag); + toProductImageResult.setModelName("flux"); + toProductImageResult.setResultType(CollectionType.TO_PRODUCT_IMAGE.getValue()); + } else { + // 走模型 + if (childFlag){ + sb.append(", (Children's face:1.3)"); + } + // 走模型 + pythonService.toProductImage(toProductElement.getUrl(), taskId, sb.toString(), toProductImageDTO.getImageStrength(), "overall"); +// toProductImageResult.setModelName("local"); + } + toProductImageResult.setElementId(toProductElement.getId()); + toProductImageResult.setElementType("ProductElement"); + toProductImageResult.setCreateTime(LocalDateTime.now()); + toProductImageResult.setToProductImageRecordId(toProductImageRecord.getId()); +// toProductImageResult.setUrl(productImageUrl); + toProductImageResult.setIsLike(0); + toProductImageResult.setTaskId(taskId); + toProductImageResult.setProjectId(projectId); + if (userLikeGroupId != null) { + toProductImageResult.setUserLikeGroupId(userLikeGroupId); + } + toProductImageResult.setImageStrength(toProductImageDTO.getImageStrength()); + toProductImageResult.setResultType(CollectionType.TO_PRODUCT_IMAGE.getValue()); + toProductImageResult.setStatus("Pending"); + toProductImageResultMapper.insert(toProductImageResult); +// toProductImageResult.setUrl(minioUtil.getPresignedUrl(toProductImageResult.getUrl(), 24 * 60)); + // 先判断是否需要默认like + if (Objects.nonNull(toProductImageDTO.getIsDefaultLike()) && toProductImageDTO.getIsDefaultLike()){ + // 满足条件情况下默认添加到like + CollectionSort collectionSort = addToProductLike(toProductImageVO.getParentId(), toProductImageResult.getId(), toProductImageDTO.getProjectId()); + // 重新排序 + Integer reSort = collectionSortService.rearrangeChildSort(toProductImageResult.getId(), CollectionType.TO_PRODUCT_IMAGE.getValue(), + toProductImageVO.getParentId(), toProductImageVO.getUserLikeSortId()); + // 将生成结果的排序返回 + toProductImageResult.setSort(Objects.isNull(reSort) ? Objects.isNull(collectionSort) ? null : collectionSort.getSort() : reSort); + toProductImageResult.setParentId(toProductImageVO.getParentId()); + toProductImageResult.setUserLikeSortId(Objects.isNull(collectionSort) ? null : collectionSort.getId()); + }else if (Objects.nonNull(toProductImageDTO.getIsDefaultLike()) && Objects.nonNull(toProductImageVO.getParentId())){ + toProductImageResult.setParentId(toProductImageVO.getParentId()); + // 默认不添加到like,但是需要有parentId,所以这里添加到collectionSort表中 +// designService.addCollectionSort(toProductImageResult.getId(), CollectionType.TO_PRODUCT_IMAGE.getValue(), toProductImageDTO.getProjectId(), toProductImageVO.getParentId()); + } + result.add(toProductImageResult); + } + i ++; + sb = new StringBuilder("The best quality, masterpiece, real image."); + // 添加需要扣除的积分到预扣除区 + creditsService.addRecordToCreditsDeduction(userHolder.getId(), taskId, creditsEventsEnum); + // 添加积分扣除记录到db + creditsService.preInsert(userHolder.getId(), creditsEventsEnum.getName(), taskId, Boolean.TRUE, null); + } + // 更新项目更新时间 + projectService.modifyProjectUpdateTime(toProductImageDTO.getProjectId()); + return result; + } + + private CollectionSort addToProductLike(Long parentId, Long toProductImageResultId, Long projectId){ + if (Objects.nonNull(parentId) && !parentId.equals(0L)){ + return productImageLike(new ProductImageLikeDTO(Collections.singletonList(toProductImageResultId), + projectId, parentId)); + } + return null; + } + + @Override + @Transactional(rollbackFor = Exception.class) + public void toProduct(String taskId) { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(ToProductImageResult::getTaskId, taskId); + List toProductImageResults = toProductImageResultMapper.selectList(qw); + if (CollectionUtil.isNotEmpty(toProductImageResults)) { + ToProductImageResult toProductImageResultOne = toProductImageResults.get(0); + ToProductImageRecord toProductImageRecord = toProductImageRecordMapper.selectById(toProductImageResultOne.getToProductImageRecordId()); + String prompt = toProductImageRecord.getPrompt(); + StringBuilder sb = new StringBuilder("The best quality, masterpiece, real image."); + if (!StringUtil.isNullOrEmpty(prompt)) { + prompt = pythonService.promptTranslate(prompt); + } + String elementType = toProductImageResultOne.getElementType(); + if (elementType.equals("DesignOutfit")) { + TDesignPythonOutfit tDesignPythonOutfit = designPythonOutfitMapper.selectById(toProductImageResultOne.getElementId()); + Long designItemId = tDesignPythonOutfit.getDesignItemId(); + QueryWrapper designItemDetailQueryWrapper = new QueryWrapper<>(); + designItemDetailQueryWrapper.lambda().eq(DesignItemDetail::getDesignItemId, designItemId); + designItemDetailQueryWrapper.lambda().ne(DesignItemDetail::getType, "Body"); + List designItemDetails = designItemDetailService.list(designItemDetailQueryWrapper); + String collect = designItemDetails.stream().map(DesignItemDetail::getType).collect(Collectors.joining(",")); + + Long designId = tDesignPythonOutfit.getDesignId(); + Design design = designMapper.selectById(designId); + String productType = "overall"; + if (design.getSingleOverall().equals("single")) { + productType = "single"; + sb.append(collect); + }else { + if (collect.contains("Tops")) { + sb.append("a handsome man,"); + }else { + sb.append("a beautiful women,"); + } + sb.append("wearing ").append(collect); + } + if (StringUtils.isEmpty(prompt)) { + sb.append(",high quality clothing details,8K realistic,HDR"); + }else { + sb.append(",high quality clothing details,").append(prompt).append(",8K realistic,HDR"); + } + // 走模型 + pythonService.toProductImage(tDesignPythonOutfit.getDesignUrl(), taskId, sb.toString(), toProductImageResultOne.getImageStrength(), "single"); + }else { + if (StringUtils.isEmpty(prompt)) { + sb.append(",high quality clothing details,8K realistic,HDR"); + }else { + sb.append(",high quality clothing details,").append(prompt).append(",8K realistic,HDR"); + } + + ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageResultOne.getElementId()); + // 走模型 + pythonService.toProductImage(toProductElement.getUrl(), taskId, sb.toString(), toProductImageResultOne.getImageStrength(), "single"); + } + + } + } + + @Override + @Transactional(rollbackFor = Exception.class) + public ToProductElementVO toProductImageElementUpload(MultipartFile file, Long projectId) { + if (null == file || StringUtils.isEmpty(file.getOriginalFilename())) { + throw new BusinessException("file.cannot.be.empty"); + } +// if (projectId == null) { +// throw new BusinessException("projectId.cannot.be.empty"); +// } + AuthPrincipalVo userHolder = UserContext.getUserHolder(); + String bucketName = "aida-users"; + String minioUrl = minioUtil.upload(bucketName, userHolder.getId() + "/toProductImageElement", file); + ToProductElement toProductElement = new ToProductElement(); + toProductElement.setUrl(minioUrl); +// toProductElement.setUserLikeGroupId(userLikeGroupId); + toProductElement.setProjectId(projectId); + toProductElement.setCreateTime(LocalDateTime.now()); +// toProductElement.setType(type); + toProductElementMapper.insert(toProductElement); + ToProductElementVO toProductElementVO = CopyUtil.copyObject(toProductElement, ToProductElementVO.class); + toProductElementVO.setMinioUrl(toProductElementVO.getUrl()); + toProductElementVO.setUrl(minioUtil.getPreSignedUrl(toProductElementVO.getUrl(), 24 * 60)); + return toProductElementVO; + } + + @Override + public CollectionSort productImageLike(ProductImageLikeDTO productImageLikeDTO) { + List toProductImageResultId = productImageLikeDTO.getToProductImageResultId(); + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().in(ToProductImageResult::getId, toProductImageResultId); + ToProductImageResult toProductImageResult = new ToProductImageResult(); + toProductImageResult.setIsLike(1); + toProductImageResultMapper.update(toProductImageResult, qw); + ToProductImageResult toProductImageResult1 = toProductImageResultMapper.selectById(toProductImageResultId.get(0)); + CollectionSort collectionSort = null; + if (toProductImageResult1.getResultType().equals("Relight")) { + if (null != productImageLikeDTO.getCollectionSortParentId()) { + collectionSort = collectionSortService.addCollectionSort(toProductImageResult1.getId(), CollectionType.RELIGHT.getValue(), productImageLikeDTO.getProjectId(), productImageLikeDTO.getCollectionSortParentId()); + } + }else { + if (null != productImageLikeDTO.getCollectionSortParentId()) { + collectionSort = collectionSortService.addCollectionSort(toProductImageResult1.getId(), CollectionType.TO_PRODUCT_IMAGE.getValue(), productImageLikeDTO.getProjectId(), productImageLikeDTO.getCollectionSortParentId()); + } + } + // 更新项目更新时间 + projectService.modifyProjectUpdateTime(productImageLikeDTO.getProjectId()); + return collectionSort; + } + @Resource + private RedisUtil redisUtil; + @Value("${redis.key.toProductImageResultKey}") + private String toProductImageResultKey; + @Value("${redis.key.relightResultKey}") + private String relightResultKey; + @Override + public List getToProductImageResultList(List taskIdList) { + List results = new ArrayList<>(); + Set collect = new HashSet<>(); + taskIdList.forEach(taskId -> { + // 查记录 + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(ToProductImageResult::getTaskId, taskId); + ToProductImageResult toProductImageResult = toProductImageResultMapper.selectOne(qw); + if (Objects.isNull(toProductImageResult)) { + throw new BusinessException("The source image does not exist."); + } + ToProductImageRecord toProductImageRecord = toProductImageRecordMapper.selectById(toProductImageResult.getToProductImageRecordId()); + if (Objects.isNull(toProductImageRecord)) { + throw new BusinessException("This task does not exist."); + } + // 判断当任务从哪个模型获取结果 + if (!StringUtil.isNullOrEmpty(toProductImageResult.getModelName()) && toProductImageResult.getModelName().equals("flux")){ + Project project = projectService.getById(toProductImageResult.getProjectId()); + if (Objects.isNull(project)){ + throw new BusinessException("unknown project"); + } + String fluxResult; + if (toProductImageResult.getStatus().equals("Success") && !StringUtil.isNullOrEmpty(toProductImageResult.getUrl())){ + fluxResult = toProductImageResult.getUrl(); + }else { + String objectName = project.getAccountId() + "/product_image/" + taskId + ".png"; + fluxResult = generateService.getFluxResult(taskId, objectName); + } + + if (StringUtil.isNullOrEmpty(fluxResult)){ + toProductImageResult.setStatus("Fail"); + toProductImageResultMapper.updateById(toProductImageResult); + results.add(new MagicToolResultVO(taskId, "Fail")); + } else if (fluxResult.equals("Fail") || fluxResult.equals("Pending")) { + toProductImageResult.setStatus(fluxResult); + toProductImageResultMapper.updateById(toProductImageResult); + results.add(new MagicToolResultVO(taskId, fluxResult)); + } else { + results.add(processFluxResult(fluxResult, toProductImageResult, taskId, toProductImageRecord.getPrompt())); + // 扣积分 + Boolean flag = creditsService.taskCreditsDeduction(project.getAccountId(), taskId); + if (flag) creditsService.updateChangedCredits(String.valueOf(project.getAccountId()), taskId); + } + // 在这个stream中不再继续往后执行 + return; + } + + String key = toProductImageResultKey + ":" + taskId; + MagicToolResultVO magicToolResultVO = new Gson().fromJson(redisUtil.getFromString(key), MagicToolResultVO.class); + if (!Objects.isNull(magicToolResultVO) && !StringUtil.isNullOrEmpty(magicToolResultVO.getUrl())) { + String url = magicToolResultVO.getUrl(); + if (url.substring(url.lastIndexOf("/") + 1).equals("white_image.jpg")) { + magicToolResultVO.setStatus("Invalid"); + } else { + magicToolResultVO.setUrl(minioUtil.getPreSignedUrl(url, CommonConstant.MINIO_IMAGE_EXPIRE_TIME)); + magicToolResultVO.setResultType(toProductImageResult.getResultType()); + magicToolResultVO.setElementId(toProductImageResult.getElementId()); + magicToolResultVO.setElementType(toProductImageResult.getElementType()); + magicToolResultVO.setDirection(toProductImageResult.getDirection()); + magicToolResultVO.setBrightenValue(toProductImageResult.getBrightenValue()); + magicToolResultVO.setImageStrength(toProductImageResult.getImageStrength()); + magicToolResultVO.setPrompt(toProductImageRecord.getPrompt()); + if (toProductImageResult.getElementType().equals("ProductElement")) { + ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageResult.getElementId()); + magicToolResultVO.setSourceUrl(minioUtil.getPreSignedUrl(toProductElement.getUrl(), 24 * 60)); + + Long parentId = collectionSortService.getParentIdByElementIdAndElementType(toProductImageResult.getElementId(), toProductImageResult.getElementType()); + magicToolResultVO.setParentId(parentId); + }else { + TDesignPythonOutfit tDesignPythonOutfit = designPythonOutfitMapper.selectById(toProductImageResult.getElementId()); + magicToolResultVO.setSourceUrl(minioUtil.getPreSignedUrl(tDesignPythonOutfit.getDesignUrl(), 24 * 60)); + + QueryWrapper userLikeQueryWrapper = new QueryWrapper<>(); + userLikeQueryWrapper.lambda().eq(UserLike::getDesignOutfitId, tDesignPythonOutfit.getId()); + List userLikeList = userLikeMapper.selectList(userLikeQueryWrapper); + if (!CollectionUtils.isEmpty(userLikeList)) { + UserLike userLike = userLikeList.get(0); + Long parentId = collectionSortService.getParentIdByElementIdAndElementType(userLike.getId(), toProductImageResult.getElementType()); + magicToolResultVO.setParentId(parentId); + } + } + } + } else if (Objects.isNull(magicToolResultVO)) { + magicToolResultVO = new MagicToolResultVO(taskId, "Executing"); + } + if (!StringUtil.isNullOrEmpty(magicToolResultVO.getStatus())) collect.add(magicToolResultVO.getStatus()); + results.add(magicToolResultVO); + }); + return results; + } + + private MagicToolResultVO processFluxResult(String fluxImgMinioPath, ToProductImageResult toProductImageResult, String taskId, String prompt){ + if (!toProductImageResult.getStatus().equals("Success") + && StringUtil.isNullOrEmpty(toProductImageResult.getUrl())){ + toProductImageResult.setStatus("Success"); + toProductImageResult.setUrl(fluxImgMinioPath); + toProductImageResultMapper.updateById(toProductImageResult); + } + + MagicToolResultVO magicToolResultVO = CopyUtil.copyObject(toProductImageResult, MagicToolResultVO.class); + magicToolResultVO.setTaskId(taskId); + magicToolResultVO.setId(toProductImageResult.getId()); + magicToolResultVO.setUrl(minioUtil.getPreSignedUrl(fluxImgMinioPath, CommonConstant.MINIO_IMAGE_EXPIRE_TIME)); + magicToolResultVO.setStatus("Success"); + magicToolResultVO.setResultType(toProductImageResult.getResultType()); + magicToolResultVO.setElementId(toProductImageResult.getElementId()); + magicToolResultVO.setElementType(toProductImageResult.getElementType()); + magicToolResultVO.setPrompt(prompt); + + if (toProductImageResult.getElementType().equals("ProductElement")) { + ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageResult.getElementId()); + magicToolResultVO.setSourceUrl(minioUtil.getPreSignedUrl(toProductElement.getUrl(), 24 * 60)); + + Long parentId = collectionSortService.getParentIdByElementIdAndElementType(toProductImageResult.getElementId(), toProductImageResult.getElementType()); + magicToolResultVO.setParentId(parentId); + } else if (toProductImageResult.getElementType().equals("DesignOutfit")){ + TDesignPythonOutfit tDesignPythonOutfit = designPythonOutfitMapper.selectById(toProductImageResult.getElementId()); + magicToolResultVO.setSourceUrl(minioUtil.getPreSignedUrl(tDesignPythonOutfit.getDesignUrl(), 24 * 60)); + + QueryWrapper userLikeQueryWrapper = new QueryWrapper<>(); + userLikeQueryWrapper.lambda().eq(UserLike::getDesignOutfitId, tDesignPythonOutfit.getId()); + List userLikeList = userLikeMapper.selectList(userLikeQueryWrapper); + if (!CollectionUtils.isEmpty(userLikeList)) { + UserLike userLike = userLikeList.get(0); + Long parentId = collectionSortService.getParentIdByElementIdAndElementType(userLike.getId(), toProductImageResult.getElementType()); + magicToolResultVO.setParentId(parentId); + } + } else if (toProductImageResult.getElementType().equals("ToProductImage")){ + ToProductImageResult toProductImage = toProductImageResultMapper.selectById(toProductImageResult.getElementId()); + magicToolResultVO.setSourceUrl(minioUtil.getPreSignedUrl(toProductImage.getUrl(), 24 * 60)); + + Long parentId = collectionSortService.getParentIdByElementIdAndElementType(toProductImageResult.getElementId(), toProductImageResult.getElementType()); + magicToolResultVO.setParentId(parentId); + } + +// Long parentId = collectionSortService.getParentIdByElementIdAndElementType(toProductImageResult.getId(), toProductImageResult.getResultType()); +// magicToolResultVO.setParentId(parentId); + return magicToolResultVO; + } + + @Override + public JSONObject exportSearch(ExportSearchDTO exportSearchDTO) { + QueryWrapper qw = new QueryWrapper<>(); + if (Objects.nonNull(exportSearchDTO.getProjectId())){ + qw.lambda().eq(ExportFile::getProjectId, exportSearchDTO.getProjectId()); + } else if (Objects.nonNull(exportSearchDTO.getId())){ + qw.lambda().eq(ExportFile::getId, exportSearchDTO.getId()); + } + qw.lambda().eq(ExportFile::getModule, exportSearchDTO.getModule()); + List exportFiles = exportFileMapper.selectList(qw); + if (CollectionUtil.isNotEmpty(exportFiles)) { + try { + InputStream download = minioUtil.download(exportFiles.get(0).getUrl()); + String convert = convert(download); + JSONObject jsonObject = JSONObject.parseObject(convert); + + JSONObject canvas = jsonObject.getJSONObject("canvas"); + if (Objects.nonNull(canvas)) { + JSONArray objectJSONArrayLevel1 = canvas.getJSONArray("objects"); + for (int i = 0; i < objectJSONArrayLevel1.size(); i++) { + JSONObject jsonObjectLevel1 = objectJSONArrayLevel1.getJSONObject(i); + String type = jsonObjectLevel1.getString("type"); + if (type.equals("image")) { + String minioUrl = jsonObjectLevel1.getString("minioUrl"); + if (!StringUtil.isNullOrEmpty(minioUrl)) jsonObjectLevel1.put("src", minioUtil.getPreSignedUrl(minioUrl, 24 * 60)); + } + if (type.equals("group")) { + JSONArray objectJSONArrayLevel2 = jsonObjectLevel1.getJSONArray("objects"); + for (int j = 0; j < objectJSONArrayLevel2.size(); j++) { + JSONObject jsonObjectLevel2 = objectJSONArrayLevel2.getJSONObject(j); + String typeGroup = jsonObjectLevel2.getString("type"); + if (typeGroup.equals("image") && !StringUtil.isNullOrEmpty(jsonObjectLevel2.getString("minioUrl"))) { + String minioUrl = jsonObjectLevel2.getString("minioUrl"); + jsonObjectLevel2.put("src", minioUtil.getPreSignedUrl(minioUrl, 24 * 60)); + } + objectJSONArrayLevel2.set(j, jsonObjectLevel2); + } + jsonObjectLevel1.put("objects", objectJSONArrayLevel2); + } + /*if (type.equals("rect")) { + JSONObject custom = jsonObjectLevel1.getJSONObject("custom"); + if (custom.getString("type").equals("init")) { + String minioUrl = jsonObjectLevel1.getString("minioUrl"); + if (!StringUtils.isEmpty(minioUrl)) { + JSONObject fill = jsonObjectLevel1.getJSONObject("fill"); + fill.put("source", minioUtil.getPreSignedUrl(minioUrl, 24 * 60)); + jsonObjectLevel1.put("fill", fill); + } + } + }*/ + objectJSONArrayLevel1.set(i, jsonObjectLevel1); + } + canvas.put("objects", objectJSONArrayLevel1); + jsonObject.put("canvas", canvas); + }else { + JSONArray objectJSONArrayLevel1 = jsonObject.getJSONArray("objects"); + for (int i = 0; i < objectJSONArrayLevel1.size(); i++) { + JSONObject jsonObjectLevel1 = objectJSONArrayLevel1.getJSONObject(i); + String type = jsonObjectLevel1.getString("type"); + if (type.equals("image")) { + String minioUrl = jsonObjectLevel1.getString("minioUrl"); + if (!StringUtil.isNullOrEmpty(minioUrl)) jsonObjectLevel1.put("src", minioUtil.getPreSignedUrl(minioUrl, 24 * 60)); + } + if (type.equals("group")) { + JSONArray objectJSONArrayLevel2 = jsonObjectLevel1.getJSONArray("objects"); + for (int j = 0; j < objectJSONArrayLevel2.size(); j++) { + JSONObject jsonObjectLevel2 = objectJSONArrayLevel2.getJSONObject(j); + String typeGroup = jsonObjectLevel2.getString("type"); + if (typeGroup.equals("image") && !StringUtil.isNullOrEmpty(jsonObjectLevel2.getString("minioUrl"))) { + String minioUrl = jsonObjectLevel2.getString("minioUrl"); + jsonObjectLevel2.put("src", minioUtil.getPreSignedUrl(minioUrl, 24 * 60)); + } + objectJSONArrayLevel2.set(j, jsonObjectLevel2); + } + jsonObjectLevel1.put("objects", objectJSONArrayLevel2); + } + /*if (type.equals("rect")) { + JSONObject custom = jsonObjectLevel1.getJSONObject("custom"); + if (custom.getString("type").equals("init")) { + String minioUrl = jsonObjectLevel1.getString("minioUrl"); + if (!StringUtils.isEmpty(minioUrl)) { + JSONObject fill = jsonObjectLevel1.getJSONObject("fill"); + fill.put("source", minioUtil.getPreSignedUrl(minioUrl, 24 * 60)); + jsonObjectLevel1.put("fill", fill); + } + } + }*/ + objectJSONArrayLevel1.set(i, jsonObjectLevel1); + } + jsonObject.put("objects", objectJSONArrayLevel1); + } + return jsonObject; + }catch (Exception e){ + log.error(e.getMessage()); + return null; + } + }else { + return null; + } + } + + @Override + public CanvasElementUpload canvasElementUpload(MultipartFile file) { + AuthPrincipalVo userHolder = UserContext.getUserHolder(); + CanvasElementUpload canvasElementUpload = new CanvasElementUpload(); + String url = minioUtil.upload("aida-users", userHolder.getId() + "/canvasElementUpload", file); + canvasElementUpload.setUrl(url); + canvasElementUpload.setMinioUrl(minioUtil.getPreSignedUrl(url, 24 * 60)); + return canvasElementUpload; + } + + @Override + public List productImageLikeList(ToProductImageDTO toProductImageDTO) { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(ToProductImageResult::getIsLike, 1); + + Long projectId = toProductImageDTO.getProjectId(); +// UserLikeGroup userLikeGroup = getByProjectId(projectId); +// Long userLikeGroupId = null; + + qw.lambda().eq(ToProductImageResult::getProjectId, projectId); + List toProductImageResults = toProductImageResultMapper.selectList(qw); + for (ToProductImageResult toProductImageResult : toProductImageResults) { + toProductImageResult.setUrl(minioUtil.getPreSignedUrl(toProductImageResult.getUrl(), 24 * 60)); + } + List toProductImageResultVOS = CopyUtil.copyList(toProductImageResults, ToProductImageResultVO.class); + for (ToProductImageResultVO toProductImageResultVO : toProductImageResultVOS) { + if (toProductImageResultVO.getElementType().equals("ProductElement")) { + ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageResultVO.getElementId()); + toProductImageResultVO.setSourceUrl(minioUtil.getPreSignedUrl(toProductElement.getUrl(), 24 * 60)); + }else if ((toProductImageResultVO.getElementType().equals("DesignOutfit"))) { + TDesignPythonOutfit tDesignPythonOutfit = designPythonOutfitMapper.selectById(toProductImageResultVO.getElementId()); + toProductImageResultVO.setSourceUrl(minioUtil.getPreSignedUrl(tDesignPythonOutfit.getDesignUrl(), 24 * 60)); + }else { + ToProductImageResult toProductImageResult1 = toProductImageResultMapper.selectById(toProductImageResultVO.getElementId()); + toProductImageResultVO.setSourceUrl(minioUtil.getPreSignedUrl(toProductImageResult1.getUrl(), 24 * 60)); + } + } + // 更新项目更新时间 + projectService.modifyProjectUpdateTime(toProductImageDTO.getProjectId()); + return toProductImageResultVOS; + } + + @Override + public Boolean productImageUnLike(ProductImageLikeDTO productImageLikeDTO) { + List toProductImageResultId = productImageLikeDTO.getToProductImageResultId(); + if (!CollectionUtils.isEmpty(toProductImageResultId)) { + Long toProductImageId = toProductImageResultId.get(0); + ToProductImageResult toProductImageResult = toProductImageResultMapper.selectById(toProductImageId); + toProductImageResult.setIsLike(0); + toProductImageResultMapper.updateById(toProductImageResult); + if (toProductImageResult.getResultType().equals(CollectionType.RELIGHT.getValue())) { + if (null != productImageLikeDTO.getCollectionSortParentId()) { + collectionSortService.deleteCollectionSort(toProductImageResult.getId(), CollectionType.RELIGHT.getValue(), productImageLikeDTO.getProjectId(), productImageLikeDTO.getCollectionSortParentId()); + } + }else { + if (null != productImageLikeDTO.getCollectionSortParentId()) { + collectionSortService.deleteCollectionSort(toProductImageResult.getId(), CollectionType.TO_PRODUCT_IMAGE.getValue(), productImageLikeDTO.getProjectId(), productImageLikeDTO.getCollectionSortParentId()); + } + } + } + // 更新项目更新时间 + projectService.modifyProjectUpdateTime(productImageLikeDTO.getProjectId()); + return Boolean.TRUE; + } + + @Override + @Transactional(rollbackFor = Exception.class) + public void relight(String taskId) { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(ToProductImageResult::getTaskId, taskId); + List toProductImageResults = toProductImageResultMapper.selectList(qw); + if (CollectionUtil.isNotEmpty(toProductImageResults)) { + ToProductImageResult toProductImageResultOne = toProductImageResults.get(0); + ToProductImageRecord toProductImageRecord = toProductImageRecordMapper.selectById(toProductImageResultOne.getToProductImageRecordId()); + String prompt = toProductImageRecord.getPrompt(); + + // 翻译 + String s = ""; + if (!StringUtil.isNullOrEmpty(prompt)) { + s = pythonService.promptTranslate(prompt); + }else { + s = "Snow moutain, snowy day, natural light"; + } + String elementType = toProductImageResultOne.getElementType(); + if (elementType.equals("ToProductImage")) { + ToProductImageResult toProductImageResult1 = toProductImageResultMapper.selectById(toProductImageResultOne.getElementId()); + String relightType = "overall"; + if (toProductImageResult1.getElementType().equals("DesignOutfit")) { + TDesignPythonOutfit tDesignPythonOutfit = designPythonOutfitMapper.selectById(toProductImageResult1.getElementId()); + Long designId = tDesignPythonOutfit.getDesignId(); + Design design = designMapper.selectById(designId); + if (design.getSingleOverall().equals("single")) { + relightType = "single"; + } + } + + // 走模型 + pythonService.relight(toProductImageResult1.getUrl(), taskId, s, toProductImageResultOne.getDirection(), "single"); + }else { + ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageResultOne.getElementId()); + // 走模型 + pythonService.relight(toProductElement.getUrl(), taskId, s, toProductImageResultOne.getDirection(), "single"); + } + + } + } + + @Override + @Transactional(rollbackFor = Exception.class) + public List relight(ToProductImageDTO toProductImageDTO) { + // 判断用户当前积分是否够本次生成消耗 + boolean fluxTask = !StringUtil.isNullOrEmpty(toProductImageDTO.getModelName()) + && toProductImageDTO.getModelName().equals("flux"); + CreditsEventsEnum creditsEventsEnum = fluxTask ? CreditsEventsEnum.RELIGHT_FLUX : CreditsEventsEnum.RELIGHT; + Boolean preDeduction = creditsService.creditsPreDeduction(creditsEventsEnum, toProductImageDTO.getToProductImageVOList().size()); + if (!preDeduction) { + throw new BusinessException("Your remaining credits are insufficient for this generation. Please recharge.", ResultEnum.WARNING.getCode()); + } + + AuthPrincipalVo userHolder = UserContext.getUserHolder(); + Long projectId = toProductImageDTO.getProjectId(); + UserLikeGroup userLikeGroup = getByProjectId(projectId); + Long userLikeGroupId = null; + + ToProductImageRecord toProductImageRecord = new ToProductImageRecord(); + toProductImageRecord.setProjectId(projectId); + if (Objects.nonNull(userLikeGroup)) { + userLikeGroupId = userLikeGroup.getId(); + toProductImageRecord.setUserLikeGroupId(userLikeGroupId); + } + + // 翻译 + String prompt = toProductImageDTO.getPrompt(); + String s = ""; + if (!StringUtil.isNullOrEmpty(prompt)) { + s = pythonService.promptTranslate(prompt); + }else { +// s = "Snow moutain, snowy day, natural light"; + s = "Street, natural light"; + } + // 对输入文本进行长度限制 + String[] words = prompt.split("\\s+"); + if (words.length > 200) { + throw new BusinessException("Please keep your input text under 200 words. Thanks!"); + } + + toProductImageRecord.setCreateTime(LocalDateTime.now()); + if (!StringUtils.isEmpty(toProductImageDTO.getPrompt())) { + toProductImageRecord.setPrompt(toProductImageDTO.getPrompt()); + } + toProductImageRecordMapper.insert(toProductImageRecord); + + List result = new ArrayList<>(); + + int i = 0; + + for (ToProductImageVO toProductImageVO : toProductImageDTO.getToProductImageVOList()) { + String taskId = UUID.randomUUID() + "-" + i + "-" + userHolder.getId(); + i ++; + if (toProductImageVO.getElementType().equals("ToProductImage")) { + ToProductImageResult toProductImageResult1 = toProductImageResultMapper.selectById(toProductImageVO.getElementId()); + String relightType = "overall"; + if (toProductImageResult1.getElementType().equals("DesignOutfit")) { + TDesignPythonOutfit tDesignPythonOutfit = designPythonOutfitMapper.selectById(toProductImageResult1.getElementId()); + Long designId = tDesignPythonOutfit.getDesignId(); + Design design = designMapper.selectById(designId); + if (design.getSingleOverall().equals("single")) { + relightType = "single"; + } + } + + ToProductImageResultVO toProductImageResult = new ToProductImageResultVO(); + if (fluxTask){ + taskId = generateService.flux(creditsEventsEnum, s, toProductImageResult1.getUrl(), false); + toProductImageResult.setModelName("flux"); + toProductImageResult.setResultType(CollectionType.RELIGHT.getValue()); + } else { + // 走模型 + pythonService.relight(toProductImageResult1.getUrl(), taskId, s, toProductImageDTO.getDirection(), relightType); + } + toProductImageResult.setElementId(toProductImageResult1.getId()); + toProductImageResult.setElementType("ToProductImage"); + toProductImageResult.setCreateTime(LocalDateTime.now()); + toProductImageResult.setToProductImageRecordId(toProductImageRecord.getId()); +// toProductImageResult.setUrl(productImageUrl); + toProductImageResult.setIsLike(0); + toProductImageResult.setTaskId(taskId); + toProductImageResult.setProjectId(projectId); + if (null != userLikeGroupId) { + toProductImageResult.setUserLikeGroupId(userLikeGroupId); + } + if (toProductImageDTO.getBrightenValue() != null) { + toProductImageResult.setBrightenValue(toProductImageDTO.getBrightenValue()); + } + toProductImageResult.setDirection(toProductImageDTO.getDirection()); + toProductImageResult.setResultType(CollectionType.RELIGHT.getValue()); + toProductImageResult.setStatus("Pending"); + toProductImageResultMapper.insert(toProductImageResult); +// toProductImageResult.setUrl(minioUtil.getPresignedUrl(toProductImageResult.getUrl(), 24 * 60)); + // 先判断是否需要默认like + if (Objects.nonNull(toProductImageDTO.getIsDefaultLike()) && toProductImageDTO.getIsDefaultLike()){ + // 满足条件情况下默认添加到like + CollectionSort collectionSort = addToProductLike(toProductImageVO.getParentId(), toProductImageResult.getId(), toProductImageDTO.getProjectId()); + // 重新排序 + Integer reSort = collectionSortService.rearrangeChildSort(toProductImageResult.getId(), CollectionType.RELIGHT.getValue(), + toProductImageVO.getParentId(), toProductImageVO.getUserLikeSortId()); + // 将生成结果的排序返回 + toProductImageResult.setSort(Objects.isNull(reSort) ? Objects.isNull(collectionSort) ? null : collectionSort.getSort() : reSort); + toProductImageResult.setParentId(toProductImageVO.getParentId()); + toProductImageResult.setUserLikeSortId(Objects.isNull(collectionSort) ? null : collectionSort.getId()); + }else if (Objects.nonNull(toProductImageDTO.getIsDefaultLike()) && Objects.nonNull(toProductImageVO.getParentId())){ + toProductImageResult.setParentId(toProductImageVO.getParentId()); + // 默认不添加到like,但是需要有parentId,所以这里添加到collectionSort表中 +// designService.addCollectionSort(toProductImageResult.getId(), CollectionType.RELIGHT.getValue(), toProductImageDTO.getProjectId(), toProductImageVO.getParentId()); + } + result.add(toProductImageResult); + }else { + ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageVO.getElementId()); + ToProductImageResultVO toProductImageResult = new ToProductImageResultVO(); + if (fluxTask){ + taskId = generateService.flux(CreditsEventsEnum.RELIGHT_FLUX, s, toProductElement.getUrl(), false); + toProductImageResult.setModelName("flux"); + toProductImageResult.setResultType(CollectionType.RELIGHT.getValue()); + } else { + // 走模型 + pythonService.relight(toProductElement.getUrl(), taskId, s, toProductImageDTO.getDirection(), "overall"); + } + toProductImageResult.setElementId(toProductElement.getId()); + toProductImageResult.setElementType("ProductElement"); + toProductImageResult.setCreateTime(LocalDateTime.now()); + toProductImageResult.setToProductImageRecordId(toProductImageRecord.getId()); +// toProductImageResult.setUrl(productImageUrl); + toProductImageResult.setIsLike(0); + toProductImageResult.setTaskId(taskId); + toProductImageResult.setProjectId(projectId); + if (null != userLikeGroupId) { + toProductImageResult.setUserLikeGroupId(userLikeGroupId); + } + if (toProductImageDTO.getBrightenValue() != null) { + toProductImageResult.setBrightenValue(toProductImageDTO.getBrightenValue()); + } + toProductImageResult.setDirection(toProductImageDTO.getDirection()); + toProductImageResult.setResultType(CollectionType.RELIGHT.getValue()); + toProductImageResult.setStatus("Pending"); + toProductImageResultMapper.insert(toProductImageResult); +// toProductImageResult.setUrl(minioUtil.getPresignedUrl(toProductImageResult.getUrl(), 24 * 60)); + // 先判断是否需要默认like + if (Objects.nonNull(toProductImageDTO.getIsDefaultLike()) && toProductImageDTO.getIsDefaultLike()){ + // 满足条件情况下默认添加到like + CollectionSort collectionSort = addToProductLike(toProductImageVO.getParentId(), toProductImageResult.getId(), toProductImageDTO.getProjectId()); + // 重新排序 + Integer reSort = collectionSortService.rearrangeChildSort(toProductImageResult.getId(), CollectionType.RELIGHT.getValue(), + toProductImageVO.getParentId(), toProductImageVO.getUserLikeSortId()); + // 将生成结果的排序返回 + toProductImageResult.setSort(Objects.isNull(reSort) ? Objects.isNull(collectionSort) ? null : collectionSort.getSort() : reSort); + toProductImageResult.setParentId(toProductImageVO.getParentId()); + toProductImageResult.setUserLikeSortId(Objects.isNull(collectionSort) ? null : collectionSort.getId()); + }else if (Objects.nonNull(toProductImageDTO.getIsDefaultLike()) && Objects.nonNull(toProductImageVO.getParentId())){ + toProductImageResult.setParentId(toProductImageVO.getParentId()); + // 默认不添加到like,但是需要有parentId,所以这里添加到collectionSort表中 +// designService.addCollectionSort(toProductImageResult.getId(), CollectionType.RELIGHT.getValue(), toProductImageDTO.getProjectId(), toProductImageVO.getParentId()); + } + result.add(toProductImageResult); + } + // 添加需要扣除的积分到预扣除区 + creditsService.addRecordToCreditsDeduction(userHolder.getId(), taskId, creditsEventsEnum); + // 添加积分扣除记录到db + creditsService.preInsert(userHolder.getId(), creditsEventsEnum.getName(), taskId, Boolean.TRUE, null); + } + // 更新项目更新时间 + projectService.modifyProjectUpdateTime(toProductImageDTO.getProjectId()); + return result; + } + + @Override + public List getRelightResult(List taskIdList) { + List results = new ArrayList<>(); + Set collect = new HashSet<>(); + taskIdList.forEach(taskId -> { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(ToProductImageResult::getTaskId, taskId); + ToProductImageResult toProductImageResult = toProductImageResultMapper.selectOne(qw); + if (Objects.isNull(toProductImageResult)) { + throw new BusinessException("The source image does not exist."); + } + ToProductImageRecord toProductImageRecord = toProductImageRecordMapper.selectById(toProductImageResult.getToProductImageRecordId()); + if (Objects.isNull(toProductImageRecord)) { + throw new BusinessException("This task does not exist."); + } + // 判断当任务从哪个模型获取结果 + if (!StringUtil.isNullOrEmpty(toProductImageResult.getModelName()) + && toProductImageResult.getModelName().equals("flux")){ + Project project = projectService.getById(toProductImageResult.getProjectId()); + if (Objects.isNull(project)){ + throw new BusinessException("unknown project"); + } + String fluxResult; + if (toProductImageResult.getStatus().equals("Success") && !StringUtil.isNullOrEmpty(toProductImageResult.getUrl())){ + fluxResult = toProductImageResult.getUrl(); + }else { + String objectName = project.getAccountId() + "/relight_image/" + taskId + ".png"; + fluxResult = generateService.getFluxResult(taskId, objectName); + } + + if (StringUtil.isNullOrEmpty(fluxResult)){ + toProductImageResult.setStatus("Fail"); + toProductImageResultMapper.updateById(toProductImageResult); + results.add(new MagicToolResultVO(taskId, "Fail")); + } else if (fluxResult.equals("Fail") || fluxResult.equals("Pending")) { + toProductImageResult.setStatus(fluxResult); + toProductImageResultMapper.updateById(toProductImageResult); + results.add(new MagicToolResultVO(taskId, fluxResult)); + } else { + results.add(processFluxResult(fluxResult, toProductImageResult, taskId, toProductImageRecord.getPrompt())); + // 扣积分 + Boolean flag = creditsService.taskCreditsDeduction(project.getAccountId(), taskId); + if (flag) creditsService.updateChangedCredits(String.valueOf(project.getAccountId()), taskId); + } + return; + } + + String key = relightResultKey + ":" + taskId; + MagicToolResultVO magicToolResultVO = new Gson().fromJson(redisUtil.getFromString(key), MagicToolResultVO.class); + if (!Objects.isNull(magicToolResultVO) && !StringUtil.isNullOrEmpty(magicToolResultVO.getUrl())) { + String url = magicToolResultVO.getUrl(); + magicToolResultVO.setDirection(toProductImageResult.getDirection()); + magicToolResultVO.setBrightenValue(toProductImageResult.getBrightenValue()); + magicToolResultVO.setImageStrength(toProductImageResult.getImageStrength()); + magicToolResultVO.setPrompt(toProductImageRecord.getPrompt()); + if (url.substring(url.lastIndexOf("/") + 1).equals("white_image.jpg")) { + magicToolResultVO.setStatus("Invalid"); + } else { + magicToolResultVO.setUrl(minioUtil.getPreSignedUrl(url, CommonConstant.MINIO_IMAGE_EXPIRE_TIME)); + + magicToolResultVO.setResultType(toProductImageResult.getResultType()); + magicToolResultVO.setElementId(toProductImageResult.getElementId()); + magicToolResultVO.setElementType(toProductImageResult.getElementType()); + if (toProductImageResult.getElementType().equals("ProductElement")) { + ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageResult.getElementId()); + magicToolResultVO.setSourceUrl(minioUtil.getPreSignedUrl(toProductElement.getUrl(), 24 * 60)); + }else { + ToProductImageResult toProductImageResult1 = toProductImageResultMapper.selectById(toProductImageResult.getElementId()); + magicToolResultVO.setSourceUrl(minioUtil.getPreSignedUrl(toProductImageResult1.getUrl(), 24 * 60)); + } + Long parentId = collectionSortService.getParentIdByElementIdAndElementType(toProductImageResult.getElementId(), toProductImageResult.getElementType()); + magicToolResultVO.setParentId(parentId); + } + } else if (Objects.isNull(magicToolResultVO)) { + magicToolResultVO = new MagicToolResultVO(); + } + if (!StringUtil.isNullOrEmpty(magicToolResultVO.getStatus())) collect.add(magicToolResultVO.getStatus()); + results.add(magicToolResultVO); + }); + return results; + } + + public static String convert(InputStream inputStream) { + try (BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream))) { + return reader.lines().collect(Collectors.joining("\n")); + } catch (Exception e) { + e.printStackTrace(); + return null; + } + } + + @Transactional(rollbackFor = Exception.class) + public void deleteToProductRelightResult(Long id, Long projectId, String type) { + // 1. 权限校验和删除主数据 + Long accountId = UserContext.getUserHolder().getId(); + Project project = projectService.getById(projectId); + if (!project.getAccountId().equals(accountId)) { + throw new IllegalArgumentException("项目不属于当前账号"); + } + + // 2. 删除主表数据 + UpdateWrapper wrapper = new UpdateWrapper<>(); + wrapper.lambda().eq(ToProductImageResult::getProjectId, projectId) + .eq(ToProductImageResult::getId, id).or().eq(ToProductImageResult::getElementId, id) + .set(ToProductImageResult::getIsDeleted, 1); + + int update = toProductImageResultMapper.update(null, wrapper); + log.info("删除 {} 结果, id={}, 影响行数={}", type, id, update); + + if (update == 0) { + return; // 未删除任何数据，直接返回 + } + + // 3. 查询所有被删除的排序项（按sort升序） + List deletedItems = collectionSortMapper.selectList( + new QueryWrapper() + .eq("project_id", projectId) + .eq("relation_id", id) + .eq("relation_type", type) + .orderByAsc("sort") + ); + + if (deletedItems.isEmpty()) return; + + // 4. 删除这些记录 + int deletedCount = collectionSortMapper.delete( + new QueryWrapper() + .eq("project_id", projectId) + .eq("relation_id", id) + .eq("relation_type", type) + ); + log.info("删除排序记录：id={}, type={}, 影响行数={}", id, type, deletedCount); + + // 5. 对每个被删除的sort值，单独调整后续记录 + for (CollectionSort deletedItem : deletedItems) { + collectionSortMapper.update( + null, + new UpdateWrapper() + .eq("project_id", projectId) + .gt("sort", deletedItem.getSort()) + .setSql("sort = sort - 1") // 每条SQL只减1 + ); + } + } + + @Override + public String likeHistoryRelSketch() { + QueryWrapper qw = new QueryWrapper<>(); + List userLikeGroups = userLikeGroupMapper.selectList(qw); + List>> result = new ArrayList<>(); + + for (UserLikeGroup userLikeGroup : userLikeGroups) { + Long accountId = userLikeGroup.getAccountId(); + Long collectionId = userLikeGroup.getCollectionId(); + + // 提前转换为Set以提高contains的效率 + List collectionElements = getCollectionElementsByCollectionId(collectionId); + Set urlSet = collectionElements.stream() + .map(CollectionElement::getUrl) + .collect(Collectors.toSet()); + + List userLikes = getUserLikesByGroupId(userLikeGroup.getId()); + + for (UserLike userLike : userLikes) { + List> userLikeSketchList = new ArrayList<>(); + List designItemDetails = getDesignItemDetails(userLike.getDesignItemId()); + + for (DesignItemDetail designItemDetail : designItemDetails) { + Map sketch = new HashMap<>(); + String path = designItemDetail.getPath(); + + // 下载路径到本地 +// if (!StringUtils.isEmpty(path)) { +// String bucketName = minioUtil.getBucketNameFromPath(path); +// String objectName = minioUtil.getObjectNameFromPath(path); +// String localBasePath = "C:\\workspace\\fileData\\minio"; +// String localFilePath = Paths.get(localBasePath, path).toString(); +// +// // 检查文件是否已经存在，避免重复下载 +// File localFile = new File(localFilePath); +// if (!localFile.exists()) { +// minioUtil.downloadMinioObjectToLocal(bucketName, objectName, localFilePath); +// } else { +// log.info("File already exists, skipping download: {}", localFilePath); +// } +// } + + Long designId = designItemDetail.getDesignId(); + Design design = designMapper.selectById(designId); + if (null == design) { + continue; + } + if (design.getSingleOverall().equals("single")) { + continue; + } + if (design.getModelType().equals("System")) { + SysFile sysFile = sysFileMapper.selectById(design.getTemplateId()); + if (null == sysFile) { + continue; + } + sketch.put("sex", sysFile.getLevel2Type().equals("Male") ? "Male" : "Female"); + } else { + Library library = libraryMapper.selectById(design.getTemplateId()); + if (null == library) { + continue; + } + sketch.put("sex", library.getLevel2Type().equals("Female") ? "Female" : "Male"); + } + + sketch.put("type", designItemDetail.getType()); + sketch.put("path", path); + sketch.put("beSelected", urlSet.contains(path)); + sketch.put("accountId", accountId); + + if (path.contains("aida-sys-image/images/")) { + String replace = path.replace("aida-sys-image/images/", ""); + String style = getStyleByUrl(replace); + if (!StringUtils.isEmpty(style)) { + sketch.put("style", style); + } + } + userLikeSketchList.add(sketch); + } + if (CollectionUtil.isNotEmpty(userLikeSketchList)) { + result.add(userLikeSketchList); + } + } + } + + // 将结果以美观的JSON形式保存到文件中 + saveResultAsPrettyJson(result, "C:\\Users\\10233\\Desktop\\result.json"); + + return null; + } + + @Override + public String download() { + // 下载 aida-collection-element bucket 下所有 Sketchboard 文件夹内容 +// String sketchboardLocalPath = "C:\\workspace\\fileData\\minio"; +// minioUtil.downloadSketchboardDirectories(sketchboardLocalPath); +// +// // 下载整个 aida-users bucket +// String aidaUsersLocalPath = "C:\\workspace\\fileData\\minio"; +// minioUtil.downloadEntireBucket("aida-users", aidaUsersLocalPath); + return null; + } + + // 提取公共方法 + private List getCollectionElementsByCollectionId(Long collectionId) { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(CollectionElement::getCollectionId, collectionId) + .eq(CollectionElement::getLevel1Type, "Sketchboard"); + return collectionElementMapper.selectList(qw); + } + + private List getUserLikesByGroupId(Long groupId) { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(UserLike::getUserLikeGroupId, groupId); + return userLikeMapper.selectList(qw); + } + + private List getDesignItemDetails(Long designItemId) { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(DesignItemDetail::getDesignItemId, designItemId) + .ne(DesignItemDetail::getType, "Body"); + return designItemDetailService.list(qw); + } + + private void saveResultAsPrettyJson(List>> result, String filePath) { + // 使用一个 Map 将结果存储为字符串键 + Map resultMap = new HashMap<>(); + resultMap.put("data", result); + + try (FileWriter file = new FileWriter(filePath)) { + // 使用 JSONObject 将转换后的 Map 转为 JSON + JSONObject jsonObject = new JSONObject(resultMap); + // 格式化输出为美观的 JSON + file.write(JSON.toJSONString(jsonObject, SerializerFeature.PrettyFormat)); + } catch (IOException e) { + e.printStackTrace(); + } + } + + + + + private String getStyleByUrl(String replace) { + String[] split = replace.split("/"); + String tableName = getTableName(split); + if (StringUtils.isEmpty(tableName)) { + return null; + }else { + replace = split[1] + "/" + split[2]; + return attributeRetrievalMapper.getStyleByUrl(replace, tableName); + } + } + + private String getTableName(String[] split) { + switch (split[0]) { + case "female": { + switch (split[1]) { + case "blouse": { + return "female_top"; + } + case "trousers": { + return "female_pants"; + } + case "skirt": { + return "female_skirt"; + } + case "outwear": { + return "female_outwear"; + } + case "dress": { + return "female_dress"; + } + default: { + return null; + } + } + } + case "male": { + switch (split[1]) { + case "top": { + return "male_top"; + } + case "bottom": { + return "male_bottom"; + } + case "outwear": { + return "male_outwear"; + } + default: { + return null; + } + } + } + default: { + return null; + } + } + } + @Override + public Boolean productImageInitialize(ProductImageInitializeDTO productImageInitializeDTO) { + AuthPrincipalVo userHolder = UserContext.getUserHolder(); + productImageService.asyncInitialize(productImageInitializeDTO.getBrandId(), userHolder); + return true; + } + + @Override + public InitializeProgressVO getInitializeProgress(ProductImageInitializeDTO productImageInitializeDTO) { + InitializeProgressVO vo = new InitializeProgressVO(); + +// QueryWrapper<> + + ProgressDTO dto = redisUtil.getTaskProgressDTO(String.valueOf(productImageInitializeDTO.getBrandId())); + + if (Objects.isNull(dto)) { + vo.setAnalyzed(false); + return vo; + } + vo.setAnalyzed(true); + if (dto.getTotal() == 0 || dto.isError()) { + vo.setPercent(0.0); + return vo; + } + vo.setPercent(dto.getCurrent() * 1.0 / dto.getTotal()); + return vo; + } + + private ProductImageAttribute toAttrDict(AttributeRecognitionJSON attrDictJSON) { + ProductImageAttribute attributeRetrieval = new ProductImageAttribute(); + + // 处理 imgName +// if (CollectionUtil.isNotEmpty(attrDictJSON.getImgName()) && attrDictJSON.getImgName().get(0) != null) { +// attributeRetrieval.setImgName(attrDictJSON.getImgName().get(0)); +// } + // 处理 length + if (CollectionUtil.isNotEmpty(attrDictJSON.getLength()) && attrDictJSON.getLength().get(0) != null) { + attributeRetrieval.setLength(attrDictJSON.getLength().get(0)); + } + // 处理 sleeveLength + if (CollectionUtil.isNotEmpty(attrDictJSON.getSleeveLength()) && attrDictJSON.getSleeveLength().get(0) != null) { + attributeRetrieval.setSleeveLength(attrDictJSON.getSleeveLength().get(0)); + } + // 处理 sleeveShape + if (CollectionUtil.isNotEmpty(attrDictJSON.getSleeveShape()) && attrDictJSON.getSleeveShape().get(0) != null) { + attributeRetrieval.setSleeveShape(attrDictJSON.getSleeveShape().get(0)); + } + // 处理 sleeveShoulder + if (CollectionUtil.isNotEmpty(attrDictJSON.getSleeveShoulder()) && attrDictJSON.getSleeveShoulder().get(0) != null) { + attributeRetrieval.setSleeveShoulder(attrDictJSON.getSleeveShoulder().get(0)); + } + // 处理 neckline + if (CollectionUtil.isNotEmpty(attrDictJSON.getNeckline()) && attrDictJSON.getNeckline().get(0) != null) { + attributeRetrieval.setNeckline(attrDictJSON.getNeckline().get(0)); + } + // 处理 collar + if (CollectionUtil.isNotEmpty(attrDictJSON.getCollar()) && attrDictJSON.getCollar().get(0) != null) { + attributeRetrieval.setCollar(attrDictJSON.getCollar().get(0)); + } + // 处理 design + if (CollectionUtil.isNotEmpty(attrDictJSON.getDesign()) && attrDictJSON.getDesign().get(0) != null) { + attributeRetrieval.setDesign(attrDictJSON.getDesign().get(0)); + } + // 处理 silhouette + if (CollectionUtil.isNotEmpty(attrDictJSON.getSilhouette()) && attrDictJSON.getSilhouette().get(0) != null) { + attributeRetrieval.setSilhouette(attrDictJSON.getSilhouette().get(0)); + } + // 处理 type + if (CollectionUtil.isNotEmpty(attrDictJSON.getType()) && attrDictJSON.getType().get(0) != null) { + attributeRetrieval.setType(attrDictJSON.getType().get(0)); + } + // 处理 openingType + if (CollectionUtil.isNotEmpty(attrDictJSON.getOpeningType()) && attrDictJSON.getOpeningType().get(0) != null) { + attributeRetrieval.setOpeningType(attrDictJSON.getOpeningType().get(0)); + } + // 处理 subtype + if (CollectionUtil.isNotEmpty(attrDictJSON.getSubtype()) && attrDictJSON.getSubtype().get(0) != null) { + attributeRetrieval.setSubtype(attrDictJSON.getSubtype().get(0)); + } + + return attributeRetrieval; + } + +// private ProductImageAttribute toAttrDict(AttributeRecognitionJSON attrDictJSON) { +// ProductImageAttribute attributeRetrieval = new ProductImageAttribute(); +//// attributeRetrieval.setImgName(attrDictJSON.getImgName().get(0)); +//// attributeRetrieval.setLength(attrDictJSON.getLength().get(0)); +//// attributeRetrieval.setSleeveLength(attrDictJSON.getSleeveLength().get(0)); +//// attributeRetrieval.setSleeveShape(attrDictJSON.getSleeveShape().get(0)); +//// attributeRetrieval.setSleeveShoulder(attrDictJSON.getSleeveShoulder().get(0)); +//// attributeRetrieval.setNeckline(attrDictJSON.getNeckline().get(0)); +//// attributeRetrieval.setCollar(attrDictJSON.getCollar().get(0)); +// if (CollectionUtil.isNotEmpty(attrDictJSON.getDesign()) && attrDictJSON.getDesign().get(0) != null) { +// attributeRetrieval.setDesign(attrDictJSON.getDesign().get(0)); +// } +// if (CollectionUtil.isNotEmpty(attrDictJSON.getSilhouette()) && attrDictJSON.getSilhouette().get(0) != null) { +// attributeRetrieval.setSilhouette(attrDictJSON.getSilhouette().get(0)); +// } +// if (CollectionUtil.isNotEmpty(attrDictJSON.getType()) && attrDictJSON.getType().get(0) != null) { +// attributeRetrieval.setType(attrDictJSON.getType().get(0)); +// } +// if (CollectionUtil.isNotEmpty(attrDictJSON.getSubtype()) && attrDictJSON.getSubtype().get(0) != null) { +// attributeRetrieval.setSubtype(attrDictJSON.getSubtype().get(0)); +// } +// if (CollectionUtil.isNotEmpty(attrDictJSON.getOpeningType()) && attrDictJSON.getOpeningType().get(0) != null) { +// attributeRetrieval.setOpeningType(attrDictJSON.getOpeningType().get(0)); +// } +// return attributeRetrieval; +// } + + @Override + public IPage getPage(ProjectQueryDTO query) { + AuthPrincipalVo authPrincipalVo = UserContext.getUserHolder(); + // 分页数据 + QueryWrapper queryWrapper = new QueryWrapper<>(); + + queryWrapper.eq("account_id", authPrincipalVo.getId()); + if (Objects.nonNull(query.getId())){ + queryWrapper.eq("id", query.getId()); + } +// queryWrapper.isNull("parent_id"); + if (!StringUtils.isEmpty(query.getProjectName())) { + queryWrapper.like("name", query.getProjectName()); + } + if (Objects.nonNull(query.getStartDate())) { + queryWrapper.ge("update_time", new Date(query.getStartDate())); + } + if (Objects.nonNull(query.getEndDate())) { + queryWrapper.le("update_time", new Date(query.getEndDate())); + } + if (!StringUtils.isEmpty(query.getProcess())) { + queryWrapper.eq("process", query.getProcess()); + } + // 新增分类过滤 + if (CollectionUtil.isNotEmpty(query.getClassificationIdList())) { + List projectIdList = new ArrayList<>(); + if (query.getIntersection() == 0) { + for (Long classificationId : query.getClassificationIdList()) { + projectIdList.addAll(classificationService.getLibraryIdListByClassificationId(classificationId)); + } + if (CollectionUtil.isNotEmpty(projectIdList)) { + queryWrapper.in("id", projectIdList); + }else { + return new Page<>(); + } + }else { + for (int i = 0; i < query.getClassificationIdList().size(); i++) { + List historyIdListByClassificationId = classificationService.getLibraryIdListByClassificationId(query.getClassificationIdList().get(i)); + if (i == 0) { + projectIdList.addAll(historyIdListByClassificationId); + }else { + projectIdList.retainAll(historyIdListByClassificationId); + } + if (CollectionUtil.isEmpty(projectIdList)) { + return new Page<>(); + } + } + queryWrapper.in("id", projectIdList); + } + } + if (query.getAsc() != null && query.getAsc() == 1) { + queryWrapper.orderByAsc("update_time"); + }else { + queryWrapper.orderByDesc("update_time"); + } + IPage page = projectService.page( + new Page<>(query.getPage(), query.getSize()), queryWrapper); + if (CollectionUtils.isEmpty(page.getRecords())) { + return new Page<>(); + } + Set projectIdSet = page.getRecords().stream().map(Project::getId).collect(Collectors.toSet()); + QueryWrapper userLikeGroupQueryWrapper = new QueryWrapper<>(); + userLikeGroupQueryWrapper.lambda().in(UserLikeGroup::getProjectId, projectIdSet); + List userLikeGroups = userLikeGroupMapper.selectList(userLikeGroupQueryWrapper); + + Map> groupDetailMap = new HashMap<>(); + + if (CollectionUtil.isNotEmpty(userLikeGroups)) { + List groupIds = userLikeGroups.stream().map(UserLikeGroup::getId).collect(Collectors.toList()); + List groupDetails = userLikeService.getGroupDetails(groupIds); + + groupDetailMap = groupDetails.stream() + .collect(Collectors.groupingBy(UserLikeVO::getUserLikeGroupId)); + } + + Account account = accountService.getById(authPrincipalVo.getId()); + Map> finalGroupDetailMap = groupDetailMap; + IPage convert = page.convert((Function) project -> { + if (project != null) { + ProjectVO projectVO = CopyUtil.copyObject(project, ProjectVO.class); + projectVO.setUpdateDate(project.getUpdateTime().atZone(ZoneId.systemDefault()).toInstant().toEpochMilli()); + + QueryWrapper userLikeGroupQueryWrapper1 = new QueryWrapper<>(); + userLikeGroupQueryWrapper1.lambda().eq(UserLikeGroup::getProjectId, projectVO.getId()); + UserLikeGroup userLikeGroup = userLikeGroupMapper.selectOne(userLikeGroupQueryWrapper1); + if (Objects.nonNull(userLikeGroup)) { + UserLikeGroupVO userLikeGroupVO = CopyUtil.copyObject(userLikeGroup, UserLikeGroupVO.class); + userLikeGroupVO.setAuthor(account.getUserName()); + //count 和detail + if (finalGroupDetailMap.keySet().contains(userLikeGroupVO.getId())) { + List details = finalGroupDetailMap.get(userLikeGroupVO.getId()); + for (UserLikeVO detail : details) { + TDesignPythonOutfit tDesignPythonOutfit = designPythonOutfitMapper.selectById(detail.getDesignOutfitId()); + detail.setUrl(minioUtil.getPreSignedUrl(tDesignPythonOutfit.getDesignUrl(), 24 * 60)); + } + userLikeGroupVO.setGroupDetails(details); + userLikeGroupVO.setSketchCount(CollectionUtils.isEmpty(details) ? 0 : details.size()); + }else { + userLikeGroupVO.setSketchCount(0); + } + if (userLikeGroupVO.getOriginal() == 0) { + userLikeGroupVO.setOriginalAccountName(accountService.getById(userLikeGroupVO.getOriginalAccountId()).getUserName()); + Portfolio byId = portfolioService.getByIdAll(userLikeGroupVO.getOriginalPortfolioId()); + if (Objects.nonNull(byId)) { + String portfolioName = byId.getPortfolioName(); + userLikeGroupVO.setOriginalPortfolioName(portfolioName); + } + } + projectVO.setUserLikeGroupVO(userLikeGroupVO); + } + return projectVO; + } + return null; + }); + return convert; + } + + + @Override + public ModuleChooseVO getModuleContent(ProjectDTO projectDTO) { + ModuleChooseVO moduleChooseVO = new ModuleChooseVO(); + moduleChooseVO.setProjectId(projectDTO.getId()); +// UserLikeCollectionVO userLikeCollectionVO = new UserLikeCollectionVO(); + for (String module : projectDTO.getModuleList()) { + if (module.equals(Module.colorBoard.name())) { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(CollectionElement::getProjectId, projectDTO.getId()); + qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.COLOR_BOARD.getRealName()); + List collectionElements = collectionElementMapper.selectList(qw); + moduleChooseVO.setColorBoard(collectionService.resolveColorBoard(collectionElements)); + }else if (module.equals(Module.moodBoard.name())) { + MoodBoardModuleChooseVO moodBoardModuleChooseVO = new MoodBoardModuleChooseVO(); + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(CollectionElement::getProjectId, projectDTO.getId()); + qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.MOOD_BOARD.getRealName()); + List collectionElements = collectionElementMapper.selectList(qw); + + List list = new ArrayList<>(); + for (CollectionElement collectionElement : collectionElements) { + if (null != collectionElement.getIsCompositeImage() && collectionElement.getIsCompositeImage() == 1) { + moodBoardModuleChooseVO.setMoodTemplateId(String.valueOf(collectionElement.getId())); + moodBoardModuleChooseVO.setMoodTemplateName(collectionElement.getName()); + moodBoardModuleChooseVO.setMoodTemplateUrl(getMinioUrl(collectionElement.getUrl())); + + String moodboardPositionString = collectionService.getMoodboardPositionString(collectionElement.getId()); + if (!StringUtils.isEmpty(moodboardPositionString)) { + moodBoardModuleChooseVO.setMoodboardPosition(moodboardPositionString); + } + continue; + } + CollectionElementVO collectionElementVO = CopyUtil.copyObject(collectionElement, CollectionElementVO.class); + collectionElementVO.setDesignType(DesignTypeEnum.COLLECTION.getRealName()); + String url = collectionElement.getUrl(); + collectionElementVO.setOriginalUrl(url); + if (minioUtil.doesObjectExist(url)) { + collectionElementVO.setUrl(getMinioUrl(url)); + } + if (minioUtil.doesObjectExist(collectionElementVO.getOriginalUrl())) { + list.add(collectionElementVO); + } + } + moodBoardModuleChooseVO.setMoodBoards(list); + moduleChooseVO.setMoodBoard(moodBoardModuleChooseVO); + }else if (module.equals(Module.printBoard.name())) { + + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(CollectionElement::getProjectId, projectDTO.getId()); + qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.PRINT_BOARD.getRealName()); + List collectionElements = collectionElementMapper.selectList(qw); + List list = new ArrayList<>(); + + for (CollectionElement collectionElement : collectionElements) { + CollectionElementVO collectionElementVO = CopyUtil.copyObject(collectionElement, CollectionElementVO.class); + collectionElementVO.setIsPin(collectionElement.getHasPin()); + collectionElementVO.setDesignType(DesignTypeEnum.COLLECTION.getRealName()); + String url = collectionElement.getUrl(); + collectionElementVO.setOriginalUrl(url); + if (minioUtil.doesObjectExist(url)) { + collectionElementVO.setUrl(getMinioUrl(url)); + } + if (minioUtil.doesObjectExist(collectionElementVO.getOriginalUrl())) { + list.add(collectionElementVO); + } + } + moduleChooseVO.setPrintBoard(list); + }else if (module.equals(Module.sketchBoard.name())) { + + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(CollectionElement::getProjectId, projectDTO.getId()); + qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.SKETCH_BOARD.getRealName()); + List collectionElements = collectionElementMapper.selectList(qw); + List list = new ArrayList<>(); + + for (CollectionElement collectionElement : collectionElements) { + CollectionElementVO collectionElementVO = CopyUtil.copyObject(collectionElement, CollectionElementVO.class); + collectionElementVO.setIsPin(collectionElement.getHasPin()); + collectionElementVO.setDesignType(DesignTypeEnum.COLLECTION.getRealName()); + String url = collectionElement.getUrl(); + collectionElementVO.setOriginalUrl(url); + if (minioUtil.doesObjectExist(url)) { + collectionElementVO.setUrl(getMinioUrl(url)); + if (url.contains(".")) { + String[] split = url.split("\\."); + collectionElementVO.setUrlWithWhiteSide(getMinioUrl(split[0] + "-show." + split[1])); + } else { + collectionElementVO.setUrlWithWhiteSide(getMinioUrl(url + "-show")); + } + } + if (minioUtil.doesObjectExist(collectionElementVO.getOriginalUrl())) { + list.add(collectionElementVO); + } + } + List voList = new ArrayList<>(); + for (CollectionElement collectionElement : collectionElements) { + CollectionSketchVO vo = CopyUtil.copyObject(collectionElement, CollectionSketchVO.class); + String url = collectionElement.getUrl(); + if (url.contains(".")) { + String[] split = url.split("\\."); + vo.setUrlWithWhiteSide(getMinioUrl(split[0] + "-show." + split[1], true)); + }else { + vo.setUrlWithWhiteSide(getMinioUrl(url + "-show", true)); + } + vo.setIsPin(collectionElement.getHasPin()); + vo.setLevel2Type(collectionElement.getLevel2Type()); + vo.setDesignType("Collection"); + vo.setSketchBoardId(collectionElement.getId()); + voList.add(vo); + } + moduleChooseVO.setBoundingBox(voList); + moduleChooseVO.setSketchBoard(list); + }else if (module.equals(Module.design.name())) { + DesignModuleChooseVO vo = new DesignModuleChooseVO(); + QueryWrapper userLikeSortQueryWrapper = new QueryWrapper<>(); + userLikeSortQueryWrapper.lambda().eq(CollectionSort::getProjectId, projectDTO.getId()); + userLikeSortQueryWrapper.lambda().isNull(CollectionSort::getParentId); + userLikeSortQueryWrapper.lambda().eq(CollectionSort::getRelationType, CollectionType.DESIGN.getValue()); + userLikeSortQueryWrapper.lambda().orderByAsc(CollectionSort::getSort); + List userLikeSortList = collectionSortMapper.selectList(userLikeSortQueryWrapper); + + if (CollectionUtils.isEmpty(userLikeSortList)) { + moduleChooseVO.setDesign(vo); + }else { + List list = new ArrayList<>(); + + for (CollectionSort collectionSort : userLikeSortList) { + UserLike userLike = userLikeMapper.selectById(collectionSort.getRelationId()); + UserLikeVO o = CopyUtil.copyObject(userLike, UserLikeVO.class); + TDesignPythonOutfit tDesignPythonOutfit1 = designPythonOutfitMapper.selectById(o.getDesignOutfitId()); + o.setUrl(tDesignPythonOutfit1.getDesignUrl()); + if (o.getUrl().contains("/")) { + int index = o.getUrl().lastIndexOf("/"); + o.setPictureName(o.getUrl().substring(index + 1)); + } + o.setDesignOutfitUrl(getMinioUrl(o.getUrl())); + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(TDesignPythonOutfit::getDesignItemId, o.getDesignItemId()); + List tDesignPythonOutfits = designPythonOutfitMapper.selectList(qw); + if (CollectionUtil.isNotEmpty(tDesignPythonOutfits)) { + TDesignPythonOutfit tDesignPythonOutfit = tDesignPythonOutfits.get(0); + o.setDesignOutfitId(tDesignPythonOutfit.getId()); + } + + o.setSort(collectionSort.getSort()); + o.setUserLikeSortId(collectionSort.getId()); + o.setRelationType(collectionSort.getRelationType()); + o.setCollectionType(CollectionType.DESIGN.getValue()); + o.setResultType(CollectionType.DESIGN.getValue()); + QueryWrapper childCollectionQw = new QueryWrapper<>(); + childCollectionQw.lambda().eq(CollectionSort::getParentId, collectionSort.getId()); + childCollectionQw.lambda().orderByAsc(CollectionSort::getSort); + List childSortList = collectionSortMapper.selectList(childCollectionQw); + List childList = new ArrayList<>(); + for (CollectionSort userLikeSort : childSortList) { + if (userLikeSort.getRelationType().equals(CollectionType.TO_PRODUCT_IMAGE.getValue())) { + ToProductImageResult toProductImageResult = toProductImageResultMapper.selectById(userLikeSort.getRelationId()); + if (isGenerateTaskFailed(toProductImageResult.getStatus(), toProductImageResult.getCreateTime())){ + continue; + } + toProductImageResult.setUrl(getMinioUrl(toProductImageResult.getUrl())); + ToProductImageResultVO toProductImageResultVO = CopyUtil.copyObject(toProductImageResult, ToProductImageResultVO.class); + + ToProductImageRecord toProductImageRecord = toProductImageRecordMapper.selectById(toProductImageResult.getToProductImageRecordId()); + if (Objects.isNull(toProductImageRecord)) { + log.warn("toProductRecord不存在，projectId:{}, toProductImageResultId:{}", projectDTO.getId(), toProductImageResult.getId()); + continue; + } + toProductImageResultVO.setPrompt(toProductImageRecord.getPrompt()); + + 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())); + } + toProductImageResultVO.setCollectionType(CollectionType.TO_PRODUCT_IMAGE.getValue()); + toProductImageResultVO.setSort(userLikeSort.getSort()); + toProductImageResultVO.setUserLikeSortId(userLikeSort.getId()); + toProductImageResultVO.setRelationType(userLikeSort.getRelationType()); + toProductImageResultVO.setParentId(userLikeSort.getParentId()); + childList.add(toProductImageResultVO); + } else if (userLikeSort.getRelationType().equals(CollectionType.RELIGHT.getValue())) { + ToProductImageResult toProductImageResult = toProductImageResultMapper.selectById(userLikeSort.getRelationId()); + if (isGenerateTaskFailed(toProductImageResult.getStatus(), toProductImageResult.getCreateTime())){ + continue; + } + toProductImageResult.setUrl(getMinioUrl(toProductImageResult.getUrl())); + ToProductImageResultVO toProductImageResultVO = CopyUtil.copyObject(toProductImageResult, ToProductImageResultVO.class); + + ToProductImageRecord toProductImageRecord = toProductImageRecordMapper.selectById(toProductImageResult.getToProductImageRecordId()); + if (Objects.isNull(toProductImageRecord)) { + log.warn("toProductRecord不存在，projectId:{}, toProductImageResultId:{}", projectDTO.getId(), toProductImageResult.getId()); + continue; + } + toProductImageResultVO.setPrompt(toProductImageRecord.getPrompt()); + + 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())); + } + toProductImageResultVO.setCollectionType(CollectionType.RELIGHT.getValue()); + toProductImageResultVO.setSort(userLikeSort.getSort()); + toProductImageResultVO.setUserLikeSortId(userLikeSort.getId()); + toProductImageResultVO.setRelationType(userLikeSort.getRelationType()); + toProductImageResultVO.setParentId(userLikeSort.getParentId()); + childList.add(toProductImageResultVO); + } else if (userLikeSort.getRelationType().equals(CollectionType.POSE_TRANSFORM.getValue())) { + PoseTransformation item = poseTransformationMapper.selectById(userLikeSort.getRelationId()); + if (isGenerateTaskFailed(item.getTaskStatus(), item.getCreateTime())){ + continue; + } + PoseTransformationVO poseTransformationVO = new PoseTransformationVO(); + poseTransformationVO.setId(item.getId()); + poseTransformationVO.setTaskId(item.getUniqueId()); + poseTransformationVO.setProductImage(getMinioUrl(item.getProductImage())); + poseTransformationVO.setGifUrl(getMinioUrl(item.getGifUrl())); + poseTransformationVO.setVideoUrl(getMinioUrl(item.getVideoUrl())); + poseTransformationVO.setFirstFrameUrl(getMinioUrl(item.getFirstFrameUrl())); + poseTransformationVO.setIsLiked(item.getIsLiked()); + poseTransformationVO.setCollectionType(CollectionType.POSE_TRANSFORM.getValue()); + poseTransformationVO.setSort(userLikeSort.getSort()); + poseTransformationVO.setUserLikeSortId(userLikeSort.getId()); + poseTransformationVO.setRelationType(userLikeSort.getRelationType()); + poseTransformationVO.setResultType(CollectionType.POSE_TRANSFORM.getValue()); + poseTransformationVO.setParentId(userLikeSort.getParentId()); + poseTransformationVO.setModelName(item.getModelName()); + poseTransformationVO.setPoseId(item.getPoseId()); + poseTransformationVO.setStatus(item.getTaskStatus()); + childList.add(poseTransformationVO); + } + } + o.setChildList(childList); + + list.add(o); + } + vo.setUserLikeDetails(list); + } + + UserLikeGroup group = getByProjectId(projectDTO.getId()); + if (Objects.nonNull(group)) { + Long userGroupId = group.getId(); + vo.setUserGroupId(userGroupId); + } + moduleChooseVO.setDesign(vo); + }else if (module.equals(Module.canvas.name())) { + CavasModuleChooseVO vo = new CavasModuleChooseVO(); + + UserLikeGroup userLikeGroup = getByProjectId(projectDTO.getId()); + + Integer beenPublished = 0; + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(Portfolio::getUserLikeGroupSourceId, userLikeGroup.getId()); + List portfolios = portfolioMapper.selectList(qw); + + PortfolioDTO portfolioDTO = new PortfolioDTO(); + if (CollectionUtil.isNotEmpty(portfolios)) { + + portfolioDTO = CopyUtil.copyObject(portfolios.get(0), PortfolioDTO.class); + beenPublished = 1; + portfolioDTO.setTagsDTO(tagsMapper.getTagByPortfolioId(portfolioDTO.getId())); + } + vo.setBeenPublished(beenPublished); + vo.setPortfolioDTO(portfolioDTO); + moduleChooseVO.setCanvas(vo); + }else if (module.equals(Module.toProduct.name()) || module.equals(Module.relight.name())) { + Long projectId = projectDTO.getId(); + String resultType = module.equals(Module.toProduct.name()) ? "ToProductImage" : "Relight"; + + // 查询所有结果（包括喜欢和不喜欢的） + QueryWrapper qw = new QueryWrapper<>(); + 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); + // 设置prompt + ToProductImageRecord toProductImageRecord = toProductImageRecordMapper.selectById(result.getToProductImageRecordId()); + if (Objects.nonNull(toProductImageRecord)){ + vo.setPrompt(toProductImageRecord.getPrompt()); + } + vo.setParentId(getUnlikedResultParentId(result, null)); + // 按isLike分类 + if (result.getIsLike() != null && result.getIsLike() == 1) { + likedList.add(vo); + } else { + normalList.add(vo); + } + } + // 设置结果 + resultVO.setLikedList(likedList); + resultVO.setList(normalList); + + // 根据模块设置返回结果 + if (module.equals(Module.toProduct.name())) { + moduleChooseVO.setToProduct(resultVO); + } else { + moduleChooseVO.setRelight(resultVO); + } + }else if (module.equals(Module.deReconstruction.name())){ + SketchReconstructionVO sketchReconstruction = generateService.getSketchReconstruction(projectDTO.getId()); + moduleChooseVO.setDeReconstruction(sketchReconstruction); + }else if (module.equals(Module.poseTransfer.name())){ + List likedList = generateService.getPoseTransformationResult(null, projectDTO.getId(), true); + List list = generateService.getPoseTransformationResult(null, projectDTO.getId(), false); + PoseTransferVO vo = new PoseTransferVO(); + vo.setLikedList(likedList); + vo.setList(list); + moduleChooseVO.setPoseTransfer(vo); + }else if (module.equals(Module.patternMaking3D.name())) { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(ThreeDModule::getProjectId, projectDTO.getId()); + List threeDModules = threeDModuleMapper.selectList(qw); + if (CollectionUtil.isNotEmpty(threeDModules)) { + ThreeDModule threeDModule = threeDModules.get(0); + if (null != threeDModule.getThreeDSimpleId()){ + ThreeDModuleVO threeDModuleVO = CopyUtil.copyObject(threeDModule, ThreeDModuleVO.class); + if (null != threeDModule.getCollectionElementId()) { + CollectionElement collectionElement = collectionElementMapper.selectById(threeDModule.getCollectionElementId()); + if (Objects.nonNull(collectionElement)) { + threeDModuleVO.setPrintMinioUrl(getMinioUrl(collectionElement.getUrl())); + } + } +// if (!StringUtils.isEmpty(threeDModuleVO.getPrintMinioUrl())) { +// threeDModuleVO.setPrintMinioUrl(minioUtil.getPreSignedUrl(threeDModuleVO.getPrintMinioUrl(), CommonConstant.MINIO_IMAGE_EXPIRE_TIME)); +// } + moduleChooseVO.setPatternMaking3D(threeDModuleVO); + } + } + }else if (module.equals(Module.mannequin.name())) { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(CollectionElement::getProjectId, projectDTO.getId()); + qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.MODEL.getRealName()); + List collectionElements = collectionElementMapper.selectList(qw); + List list = new ArrayList<>(); + if (CollectionUtil.isNotEmpty(collectionElements)) { + for (CollectionElement collectionElement : collectionElements) { + CollectionElementVO collectionElementVO = CopyUtil.copyObject(collectionElement, CollectionElementVO.class); + CollectionElementRelModel relModel = getCollectionElementRelModel(collectionElementVO.getId()); + collectionElementVO.setUrl(getMinioUrl(collectionElementVO.getUrl())); + collectionElementVO.setMannequinRelationId(relModel.getRelationId()); + collectionElementVO.setMannequinRelationType(relModel.getRelationType()); + collectionElementVO.setCollectionId(collectionElement.getId()); + list.add(collectionElementVO); + } + } + moduleChooseVO.setMannequin(list); + }else if (module.equals(Module.uploadElement.name())) { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(ToProductElement::getProjectId, projectDTO.getId()); + List toProductElements = toProductElementMapper.selectList(qw); + for (ToProductElement toProductElement : toProductElements) { + toProductElement.setUrl(getMinioUrl(toProductElement.getUrl())); + } + moduleChooseVO.setUploadElement(toProductElements); + } + } + return moduleChooseVO; + } + + private String getMinioUrl(String minioPath){ + if (!StringUtil.isNullOrEmpty(minioPath)){ + return minioUtil.getPreSignedUrl(minioPath, CommonConstant.MINIO_IMAGE_EXPIRE_TIME); + } + return null; + } + + private String getMinioUrl(String minioPath, boolean resetCatch){ + if (!StringUtil.isNullOrEmpty(minioPath)){ + return minioUtil.getPreSignedUrl(minioPath, CommonConstant.MINIO_IMAGE_EXPIRE_TIME, resetCatch); + } + return null; + } + + // 提取出的公共方法 + private void setSourceUrl(ToProductImageResultVO vo) { + if (vo.getElementType().equals("ProductElement")) { + ToProductElement element = toProductElementMapper.selectById(vo.getElementId()); + vo.setSourceUrl(getMinioUrl(element.getUrl())); + } else if (vo.getElementType().equals("DesignOutfit")) { + TDesignPythonOutfit outfit = designPythonOutfitMapper.selectById(vo.getElementId()); + vo.setSourceUrl(getMinioUrl(outfit.getDesignUrl())); + } else { + ToProductImageResult result = toProductImageResultMapper.selectById(vo.getElementId()); + vo.setSourceUrl(getMinioUrl(result.getUrl())); + } + } + + private CollectionElementRelModel getCollectionElementRelModel(Long id) { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(CollectionElementRelModel::getCollectionElementId, id); + CollectionElementRelModel collectionElementRelModel = collectionElementRelModelMapper.selectOne(qw); + return collectionElementRelModel; + } + + private boolean isGenerateTaskFailed(String taskStatus, LocalDateTime createTime) { + // 1. 先判断TaskStatus字段是否有值；若有值且为Fail，则失败 + if (!StringUtil.isNullOrEmpty(taskStatus)) { + return "Fail".equals(taskStatus); + } + // 2. 若taskStatus无值，判断记录创建时间距离现在是否超过1小时 + if (createTime == null) { + // 如果创建时间为null，视为失败（根据业务需求决定） + return true; + } + // 计算当前时间与创建时间的差值 + Duration duration = Duration.between(createTime, LocalDateTime.now()); + // 判断是否超过1小时（3600秒） + return duration.toHours() >= 1; + } + + // 获取未被like的to product\relight的parentId + public Long getUnlikedResultParentId(ToProductImageResult toProductImageResult, String url) { + // 卫语句处理null情况 + if (toProductImageResult == null && StringUtil.isNullOrEmpty(url)) { + return null; + } + + // 如果需要查询结果对象 + if (toProductImageResult == null) { + toProductImageResult = toProductImageResultMapper.selectOne( + new QueryWrapper().eq("url", url)); + } + + // 卫语句处理查询结果为空的情况 + if (toProductImageResult == null) { + return null; + } + + // 根据不同类型处理 + String elementType = toProductImageResult.getElementType(); + if ("DesignOutfit".equals(elementType)) { + return handleDesignOutfitCase(toProductImageResult); + } else if ("ToProductImage".equals(elementType)) { + // 两种情况 resultType : ToProductImage | Relight + return handleToProductImageCase(toProductImageResult); + } + + return null; + } + + private Long handleDesignOutfitCase(ToProductImageResult result) { + UserLike userLike = userLikeService.getByDesignOutfitId(result.getElementId()); + if (userLike == null) { + return null; + } + + CollectionSort collectionSort = collectionSortService.queryCollectionSortByRelation( + userLike.getId(), "Design", result.getProjectId()); + return collectionSort != null ? collectionSort.getId() : null; + } + + private Long handleToProductImageCase(ToProductImageResult result) { + CollectionSort collectionSort = collectionSortService.queryCollectionSortByRelation( + result.getElementId(), result.getResultType(), result.getProjectId()); + if (Objects.isNull(collectionSort) && result.getElementType().equals("ToProductImage")){ + ToProductImageResult toProductImageResult = toProductImageResultMapper.selectById(result.getElementId()); + return getUnlikedResultParentId(toProductImageResult, null); + } + return collectionSort != null ? collectionSort.getParentId() : null; + } + + @Override + @Transactional + public ModuleChooseVO saveModuleContent(ModuleSaveDTO moduleSaveDTO) { + ModuleChooseVO result = new ModuleChooseVO(); + Long accountId = UserContext.getUserHolder().getId(); + Long projectId = moduleSaveDTO.getProjectId(); + Project project = projectService.getById(projectId); + + boolean boundingBox = false; + if (Objects.nonNull(moduleSaveDTO.getMoodBoard())) { + MoodBoardModuleSaveDTO moodBoard = moduleSaveDTO.getMoodBoard(); + if (moodBoard.getMoodTemplateId() != null) { + // moodboard合成图存储 + CollectionElement compositeImage = collectionElementMapper.selectById(moodBoard.getMoodTemplateId()); + if (compositeImage.getProjectId() == null) { + compositeImage.setProjectId(moduleSaveDTO.getProjectId()); + compositeImage.setIsCompositeImage(1); + collectionElementMapper.updateById(compositeImage); + if (!StringUtils.isEmpty(moodBoard.getMoodboardPosition())) { + // 合成图位置信息通过collectElementId关联（旧逻辑通过collectionId关联） + designService.parseMoodboardPosition(moodBoard.getMoodboardPosition(), compositeImage.getId()); + } + } + } + List moodBoards = moodBoard.getMoodBoards(); + if (CollectionUtil.isNotEmpty(moodBoards)) { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(CollectionElement::getProjectId, projectId); + qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.MOOD_BOARD.getRealName()); + if (moodBoard.getMoodTemplateId() != null) { + qw.lambda().ne(CollectionElement::getId, moodBoard.getMoodTemplateId()); + } +// qw.lambda().ne(CollectionElement::getCollectionId, 0); + List collectionElements = collectionElementMapper.selectList(qw); + Set old = collectionElements.stream().map(CollectionElement::getId).collect(Collectors.toSet()); + + for (DesignCollectionElementDTO board : moodBoards) { + if (board.getDesignType().equals(DesignTypeEnum.LIBRARY.getRealName())) { + Library library = libraryMapper.selectById(board.getId()); + CollectionElement collectionElement = new CollectionElement(); + collectionElement.setAccountId(accountId); + collectionElement.setProjectId(projectId); + collectionElement.setLevel1Type(CollectionLevel1TypeEnum.MOOD_BOARD.getRealName()); + collectionElement.setName(library.getName()); + collectionElement.setUrl(library.getUrl()); + collectionElement.setHasPin((byte) 0); + collectionElement.setMd5(library.getMd5()); + collectionElement.setCreateDate(new Date()); + collectionElementMapper.insert(collectionElement); + }else if (board.getDesignType().equals(DesignTypeEnum.GENERATE.getRealName())) { + GenerateDetail generateDetail = generateDetailMapper.selectById(board.getId()); + CollectionElement collectionElement = new CollectionElement(); + collectionElement.setAccountId(accountId); + collectionElement.setProjectId(projectId); + collectionElement.setLevel1Type(CollectionLevel1TypeEnum.MOOD_BOARD.getRealName()); +// collectionElement.setName(generateDetail.get()); + collectionElement.setUrl(generateDetail.getUrl()); + collectionElement.setHasPin((byte) 0); + collectionElement.setMd5(generateDetail.getMd5()); + collectionElement.setCreateDate(new Date()); + collectionElementMapper.insert(collectionElement); + }else { + if (old.contains(board.getId())) { + old.remove(board.getId()); + }else { + CollectionElement collectionElement = collectionElementMapper.selectById(board.getId()); + collectionElement.setProjectId(projectId); + collectionElementMapper.updateById(collectionElement); + } + } + } + if (!CollectionUtils.isEmpty(old)) { + collectionElementMapper.deleteBatchIds(old); + } + } + } + if (!CollectionUtils.isEmpty(moduleSaveDTO.getPrintBoard())){ + List printBoards = moduleSaveDTO.getPrintBoard(); + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(CollectionElement::getProjectId, projectId); + qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.PRINT_BOARD.getRealName()); + List collectionElements = collectionElementMapper.selectList(qw); + Set old = collectionElements.stream().map(CollectionElement::getId).collect(Collectors.toSet()); + + for (DesignCollectionPrintElementDTO board : printBoards) { + if (board.getDesignType().equals(DesignTypeEnum.LIBRARY.getRealName())) { + Library library = libraryMapper.selectById(board.getId()); + CollectionElement collectionElement = new CollectionElement(); + collectionElement.setAccountId(accountId); + collectionElement.setProjectId(projectId); + collectionElement.setLevel1Type(CollectionLevel1TypeEnum.PRINT_BOARD.getRealName()); + collectionElement.setLevel2Type(board.getLevel2Type()); + collectionElement.setName(library.getName()); + collectionElement.setUrl(library.getUrl()); + collectionElement.setHasPin(board.getIsPin()); + collectionElement.setMd5(library.getMd5()); + collectionElement.setCreateDate(new Date()); + collectionElementMapper.insert(collectionElement); + }else if (board.getDesignType().equals(DesignTypeEnum.GENERATE.getRealName())) { + GenerateDetail generateDetail = generateDetailMapper.selectById(board.getId()); + CollectionElement collectionElement = new CollectionElement(); + collectionElement.setAccountId(accountId); + collectionElement.setProjectId(projectId); + collectionElement.setLevel1Type(CollectionLevel1TypeEnum.PRINT_BOARD.getRealName()); +// collectionElement.setLevel2Type(board.getLevel2Type()); +// collectionElement.setName(generateDetail.get()); + collectionElement.setUrl(generateDetail.getUrl()); + collectionElement.setHasPin(board.getIsPin()); + collectionElement.setMd5(generateDetail.getMd5()); + collectionElement.setCreateDate(new Date()); + collectionElementMapper.insert(collectionElement); + }else { + if (old.contains(board.getId())) { + CollectionElement collectionElement = collectionElementMapper.selectById(board.getId()); +// collectionElement.setLevel2Type(board.getLevel2Type()); + collectionElement.setHasPin(board.getIsPin()); + collectionElement.setUpdateDate(new Date()); + collectionElementMapper.updateById(collectionElement); + old.remove(board.getId()); + }else { + CollectionElement collectionElement = collectionElementMapper.selectById(board.getId()); + collectionElement.setProjectId(projectId); + collectionElement.setHasPin(board.getIsPin()); + collectionElement.setUpdateDate(new Date()); + collectionElementMapper.updateById(collectionElement); + } + } + } + if (!CollectionUtils.isEmpty(old)) { + collectionElementMapper.deleteBatchIds(old); + } + } + if (!CollectionUtils.isEmpty(moduleSaveDTO.getColorBoard())){ + List colorBoards = moduleSaveDTO.getColorBoard(); + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(CollectionElement::getProjectId, projectId); + qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.COLOR_BOARD.getRealName()); + List collectionElements = collectionElementMapper.selectList(qw); + Set old = collectionElements.stream().map(CollectionElement::getId).collect(Collectors.toSet()); + + for (CollectionColorDTO board : colorBoards) { + if (Objects.nonNull(board.getId()) && old.contains(Long.valueOf(board.getId()))) { + // ??? + old.remove(Long.valueOf(board.getId())); + }else { + CollectionElement collectionElement = new CollectionElement(); + collectionElement.setAccountId(accountId); + collectionElement.setProjectId(projectId); + collectionElement.setLevel1Type(CollectionLevel1TypeEnum.COLOR_BOARD.getRealName()); + collectionElement.setName(board.getName()); + collectionElement.setHasPin((byte) 0); + collectionElement.setColorRgb(board.getRgbValue()); + collectionElement.setMd5("0"); + // 渐变色 + if (!Objects.isNull(board.getGradient()) && !StringUtil.isNullOrEmpty(board.getGradient().getColorImg())){ + board.getGradient().setColorImg(null); + } + collectionElement.setGradientString(JSON.toJSONString(board.getGradient())); + collectionElement.setCreateDate(new Date()); + collectionElementMapper.insert(collectionElement); + } + } + if (!CollectionUtils.isEmpty(old)) { + collectionElementMapper.deleteBatchIds(old); + } + } + if (!CollectionUtils.isEmpty(moduleSaveDTO.getSketchBoard())){ + boundingBox = true; + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(CollectionElement::getProjectId, projectId); + qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.SKETCH_BOARD.getRealName()); + List collectionElements = collectionElementMapper.selectList(qw); + Set old = collectionElements.stream().map(CollectionElement::getId).collect(Collectors.toSet()); + List sketchBoards = moduleSaveDTO.getSketchBoard(); + for (CollectionSketchDTO board : sketchBoards) { + if (board.getDesignType().equals(DesignTypeEnum.LIBRARY.getRealName())) { + Library library = libraryMapper.selectById(board.getSketchBoardId()); + CollectionElement collectionElement = new CollectionElement(); + collectionElement.setAccountId(accountId); + collectionElement.setProjectId(projectId); + collectionElement.setLevel1Type(CollectionLevel1TypeEnum.SKETCH_BOARD.getRealName()); + collectionElement.setLevel2Type(board.getLevel2Type()); + collectionElement.setName(library.getName()); + collectionElement.setUrl(library.getUrl()); + collectionElement.setHasPin(board.getIsPin()); + collectionElement.setMd5(library.getMd5()); + collectionElement.setCreateDate(new Date()); + collectionElementMapper.insert(collectionElement); + }else if (board.getDesignType().equals(DesignTypeEnum.GENERATE.getRealName())) { + GenerateDetail generateDetail = generateDetailMapper.selectById(board.getSketchBoardId()); + CollectionElement collectionElement = new CollectionElement(); + collectionElement.setAccountId(accountId); + collectionElement.setProjectId(projectId); + collectionElement.setLevel1Type(CollectionLevel1TypeEnum.SKETCH_BOARD.getRealName()); + collectionElement.setLevel2Type(board.getLevel2Type()); +// collectionElement.setName(generateDetail.get()); + collectionElement.setUrl(generateDetail.getUrl()); + collectionElement.setHasPin(board.getIsPin()); + collectionElement.setMd5(generateDetail.getMd5()); + collectionElement.setCreateDate(new Date()); + collectionElementMapper.insert(collectionElement); + }else { + if (old.contains(board.getSketchBoardId())) { + CollectionElement collectionElement = collectionElementMapper.selectById(board.getSketchBoardId()); + collectionElement.setLevel2Type(board.getLevel2Type()); + collectionElement.setHasPin(board.getIsPin()); + collectionElement.setUpdateDate(new Date()); + collectionElementMapper.updateById(collectionElement); + old.remove(board.getSketchBoardId()); + }else { + CollectionElement collectionElement = collectionElementMapper.selectById(board.getSketchBoardId()); + collectionElement.setProjectId(projectId); + collectionElement.setLevel2Type(board.getLevel2Type()); + collectionElement.setHasPin(board.getIsPin()); + collectionElement.setUpdateDate(new Date()); + collectionElementMapper.updateById(collectionElement); + } + } + } + if (!CollectionUtils.isEmpty(old)) { + collectionElementMapper.deleteBatchIds(old); + } + } + if (Objects.nonNull(moduleSaveDTO.getPatternMaking3D())) { + PatternMaking3DDTO patternMaking3D = moduleSaveDTO.getPatternMaking3D(); + + QueryWrapper threeDModuleQueryWrapper = new QueryWrapper<>(); + threeDModuleQueryWrapper.lambda().eq(ThreeDModule::getProjectId, projectId); + List threeDModuleList = threeDModuleMapper.selectList(threeDModuleQueryWrapper); + + if (CollectionUtils.isEmpty(threeDModuleList)) { + ThreeDModule threeDModule = new ThreeDModule(); + threeDModule.setProjectId(projectId); + threeDModule.setThreeDSimpleId(patternMaking3D.getThreeDSimpleId()); + threeDModule.setX(patternMaking3D.getX()); + threeDModule.setY(patternMaking3D.getY()); +// threeDModule.setPrintMinioUrl(patternMaking3D.getPrintMinioUrl()); + threeDModule.setCollectionElementId(patternMaking3D.getCollectionElementId()); + threeDModuleMapper.insert(threeDModule); + }else { + ThreeDModule threeDModule = threeDModuleList.get(0); +// if (!Objects.equals(patternMaking3D.getThreeDSimpleId(), threeDModule.getThreeDSimpleId())) { +// +// } + threeDModule.setThreeDSimpleId(patternMaking3D.getThreeDSimpleId()); + threeDModule.setX(patternMaking3D.getX()); + threeDModule.setY(patternMaking3D.getY()); +// threeDModule.setPrintMinioUrl(patternMaking3D.getPrintMinioUrl()); + threeDModule.setCollectionElementId(patternMaking3D.getCollectionElementId()); + threeDModuleMapper.updateById(threeDModule); + } + } + if (CollectionUtil.isNotEmpty(moduleSaveDTO.getMannequin())) { + List collectionElementRelModels = collectionElementMapper.selectByProject(projectId); + List mannequin = moduleSaveDTO.getMannequin(); + List inputMannequinId = mannequin.stream().map(MannequinDTO::getId).collect(Collectors.toList()); + + // 创建relationId到Model的映射 + Map relationIdToModelMap = collectionElementRelModels.stream() + .collect(Collectors.toMap( + CollectionElementRelModel::getRelationId, + java.util.function.Function.identity(), + (existing, replacement) -> existing + )); + List newReletionIds = new ArrayList<>(); + List toBeDeletedRelationIds = new ArrayList<>(); + + if (collectionElementRelModels.isEmpty()) { + // 如果原本没有记录，所有输入的Mannequin都是新的 + newReletionIds = inputMannequinId; + } else { + // 获取所有已有的relationId集合 + Set relationIds = relationIdToModelMap.keySet(); + + // 1. 找出需要删除的：relationIds中有但inputMannequinId中没有的 + toBeDeletedRelationIds = relationIds.stream() + .filter(id -> !inputMannequinId.contains(id)) + .collect(Collectors.toList()); + + // 2. 找出新增的：inputMannequinId中有但relationIds中没有的 + newReletionIds = inputMannequinId.stream() + .filter(id -> !relationIds.contains(id)) + .collect(Collectors.toList()); + } + + if (!toBeDeletedRelationIds.isEmpty()){ + Set toBeDeletedElementIdList = toBeDeletedRelationIds.stream() + .map(relationIdToModelMap::get) // 获取对应的Model + .filter(Objects::nonNull) // 过滤掉可能为null的情况 + .map(CollectionElementRelModel::getCollectionElementId) // 提取collectionElementId + .collect(Collectors.toSet()); + collectionElementMapper.deleteBatchIds(toBeDeletedElementIdList); + // 删除关联关系 + deleteByCollectionElementIdList(toBeDeletedElementIdList); + } + + for (MannequinDTO dto : mannequin) { + if (newReletionIds.isEmpty()){ + break; + }else if (!newReletionIds.contains(dto.getId())){ + continue; + } + if (dto.getType().equals("System")) { + SysFile sysFile = sysFileMapper.selectById(dto.getId()); + + CollectionElement collectionElement = new CollectionElement(); + collectionElement.setAccountId(accountId); + collectionElement.setProjectId(projectId); + collectionElement.setLevel1Type(CollectionLevel1TypeEnum.MODEL.getRealName()); +// collectionElement.setLevel2Type(board.getLevel2Type()); + collectionElement.setName(sysFile.getName()); + collectionElement.setUrl(sysFile.getUrl()); +// collectionElement.setHasPin(board.getIsPin()); + collectionElement.setMd5(sysFile.getMd5()); + collectionElement.setCreateDate(new Date()); + collectionElement.setHasPin((byte) 0); + collectionElementMapper.insert(collectionElement); + CollectionElementRelModel collectionElementRelModel = new CollectionElementRelModel(); + collectionElementRelModel.setCollectionElementId(collectionElement.getId()); + collectionElementRelModel.setRelationId(dto.getId()); + collectionElementRelModel.setRelationType("System"); + collectionElementRelModelMapper.insert(collectionElementRelModel); + }else if (dto.getType().equals("Library")) { + Library library = libraryMapper.selectById(dto.getId()); + + CollectionElement collectionElement = new CollectionElement(); + collectionElement.setAccountId(accountId); + collectionElement.setProjectId(projectId); + collectionElement.setLevel1Type(CollectionLevel1TypeEnum.MODEL.getRealName()); +// collectionElement.setLevel2Type(board.getLevel2Type()); + collectionElement.setName(library.getName()); + collectionElement.setUrl(library.getUrl()); +// collectionElement.setHasPin(board.getIsPin()); + collectionElement.setMd5(library.getMd5()); + collectionElement.setCreateDate(new Date()); + collectionElement.setHasPin((byte) 0); + collectionElementMapper.insert(collectionElement); + CollectionElementRelModel collectionElementRelModel = new CollectionElementRelModel(); + collectionElementRelModel.setCollectionElementId(collectionElement.getId()); + collectionElementRelModel.setRelationId(dto.getId()); + collectionElementRelModel.setRelationType("System"); + collectionElementRelModelMapper.insert(collectionElementRelModel); + + } + } + } + if (boundingBox) { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(CollectionElement::getLevel1Type, CollectionLevel1TypeEnum.SKETCH_BOARD.getRealName()); + qw.lambda().eq(CollectionElement::getProjectId, projectId); + List collectionElements = collectionElementMapper.selectList(qw); + + List voList = new ArrayList<>(); + for (CollectionElement collectionElement : collectionElements) { + CollectionSketchVO vo = CopyUtil.copyObject(collectionElement, CollectionSketchVO.class); + String url = collectionElement.getUrl(); + libraryService.processSketchBoards(url, collectionElement.getLevel2Type()); + if (url.contains(".")) { + String[] split = url.split("\\."); + vo.setUrlWithWhiteSide(minioUtil.getPreSignedUrl(split[0] + "-show." + split[1], 24 * 60, true)); + }else { + vo.setUrlWithWhiteSide(minioUtil.getPreSignedUrl(url + "-show", 24 * 60, true)); + } + vo.setIsPin(collectionElement.getHasPin()); + vo.setLevel2Type(collectionElement.getLevel2Type()); + vo.setDesignType("Collection"); + vo.setSketchBoardId(collectionElement.getId()); + voList.add(vo); + } + result.setBoundingBox(voList); + } + project.setUpdateTime(LocalDateTime.now()); + projectService.updateById(project); + return result; + } + + private void deleteByCollectionElementIdList(Set old) { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().in(CollectionElementRelModel::getCollectionElementId, old); + collectionElementRelModelMapper.delete(qw); + } + + @Override + public QueryLibraryPageVO getMannequinDetail(MannequinDTO mannequinDTO) { + QueryLibraryPageVO vo = new QueryLibraryPageVO(); + if (mannequinDTO.getType().equals("System")) { + SysFile sysFile = sysFileMapper.selectById(mannequinDTO.getId()); + vo.setDesignType("System"); + vo.setLevel1Type(sysFile.getLevel1Type()); + vo.setLevel2Type(sysFile.getLevel2Type()); + vo.setName(sysFile.getName()); + vo.setMinIOPath(minioUtil.getPreSignedUrl(sysFile.getUrl(), 24 * 60)); + }else { + Library library = libraryMapper.selectById(mannequinDTO.getId()); + vo.setDesignType("Library"); + vo.setLevel1Type(library.getLevel1Type()); + vo.setLevel2Type(library.getLevel2Type()); + vo.setName(library.getName()); + vo.setMinIOPath(minioUtil.getPreSignedUrl(library.getUrl(), 24 * 60)); + } + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(LibraryModelPoint::getRelationId, mannequinDTO.getId()); + qw.lambda().eq(LibraryModelPoint::getModelType, mannequinDTO.getType()); + List libraryModelPoints = libraryModelPointMapper.selectList(qw); + if (CollectionUtil.isEmpty(libraryModelPoints)) { + throw new BusinessException("No model doting information available."); + } + LibraryModelPoint libraryModelPoint = libraryModelPoints.get(0); + LibraryModelPointVO libraryModelPointVO = CopyUtil.copyObject(libraryModelPoint, LibraryModelPointVO.class); + libraryModelPointVO.setTemplateId(libraryModelPoint.getId()); + libraryModelPointVO.setHandLeft(JSON.parseObject(libraryModelPoint.getHandLeft(), List.class)); + libraryModelPointVO.setHandRight(JSON.parseObject(libraryModelPoint.getHandRight(), List.class)); + libraryModelPointVO.setShoulderRight(JSON.parseObject(libraryModelPoint.getShoulderRight(), List.class)); + libraryModelPointVO.setShoulderLeft(JSON.parseObject(libraryModelPoint.getShoulderLeft(), List.class)); + libraryModelPointVO.setWaistbandRight(JSON.parseObject(libraryModelPoint.getWaistbandRight(), List.class)); + libraryModelPointVO.setWaistbandLeft(JSON.parseObject(libraryModelPoint.getWaistbandLeft(), List.class)); + + vo.setLibraryModelPoint(libraryModelPointVO); + return vo; + } + + @Override + public BrandLogoUploadVO brandLogoUpload(MultipartFile file) { + BrandLogoUploadVO vo = new BrandLogoUploadVO(); + AuthPrincipalVo userHolder = UserContext.getUserHolder(); + String path = userHolder.getId().toString() + "/brandLogo"; + vo.setBrandLogo(minioUtil.upload("aida-users",path,file)); + vo.setMinioUrl(minioUtil.getPreSignedUrl(vo.getBrandLogo(), 24 * 60)); + return vo; +// return null; + } + + @Override + public Boolean brandDNASaveOrUpdate(BrandDNADTO brandDNADTO) { + if (brandDNADTO.getId() != null) { + BrandDNA brandDNA = CopyUtil.copyObject(brandDNADTO, BrandDNA.class); + brandDNA.setUpdateTime(LocalDateTime.now()); + brandDNAMapper.updateById(brandDNA); + }else { + BrandDNA brandDNA = CopyUtil.copyObject(brandDNADTO, BrandDNA.class); + AuthPrincipalVo userHolder = UserContext.getUserHolder(); + brandDNA.setAccountId(userHolder.getId()); + brandDNA.setCreateTime(LocalDateTime.now()); + brandDNAMapper.insert(brandDNA); + } + return Boolean.TRUE; + } + + @Override + public LibraryUpdateVo brandDNAUpload(MultipartFile file, Long brandId) throws IOException { + AuthPrincipalVo userHolder = UserContext.getUserHolder(); + String path = userHolder.getId().toString() + "/brandLogo"; + String upload = minioUtil.upload("aida-users", path, file); + Library library = new Library(); + library.setAccountId(userHolder.getId()); + library.setLevel1Type(LibraryLevel1TypeEnum.BRAND_DNA.getRealName()); + library.setName(file.getOriginalFilename()); + library.setUrl(upload); + library.setMd5(MD5Utils.encryptFile(file.getInputStream())); + library.setCreateDate(new Date()); + libraryMapper.insert(library); + + BrandRelLibrary brandRelLibrary = new BrandRelLibrary(); + brandRelLibrary.setLibraryId(library.getId()); + brandRelLibrary.setBrandId(brandId); + brandRelLibraryMapper.insert(brandRelLibrary); + + LibraryUpdateVo vo = CopyUtil.copyObject(library, LibraryUpdateVo.class); + vo.setMinIOPath(minioUtil.getPreSignedUrl(vo.getUrl(), 24 * 60)); + + return vo; + } + + @Override + public PageBaseResponse brandDNAPage(BrandDNAQueryDTO brandDNAQueryDTO) { + AuthPrincipalVo userHolder = UserContext.getUserHolder(); + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(BrandDNA::getAccountId, userHolder.getId()); + IPage page = brandDNAMapper.selectPage(new Page<>(brandDNAQueryDTO.getPage(), brandDNAQueryDTO.getSize()), qw); + IPage convert = page.convert((Function) brand -> { + if (Objects.nonNull(brand)) { + BrandDNAVO brandDNAVO = CopyUtil.copyObject(brand, BrandDNAVO.class); + brandDNAVO.setMinioUrl(minioUtil.getPreSignedUrl(brandDNAVO.getBrandLogo(), 24 * 60)); + return brandDNAVO; + } + return null; + }); + return PageBaseResponse.success(convert); + } + + @Override + public BrandDNAGenerateVO brandDNAGenerate(String prompt) { + BrandDNAGenerateVO vo = new BrandDNAGenerateVO(); + JSONObject jsonObject = pythonService.brandDNAGenerate(prompt); + JSONObject data = jsonObject.getJSONObject("data"); + vo.setBrandSlogan(data.getString("brand_slogan")); + vo.setBrandName(data.getString("brand_name")); + vo.setBrandLogo(data.getString("brand_logo")); + vo.setMinioUrl(minioUtil.getPreSignedUrl(vo.getBrandLogo(), 24 * 60)); + return vo; + } + + @Override + public IPage getThreeDLayoutPage(ThreeDLayoutQueryDTO query) { +// AuthPrincipalVo authPrincipalVo = UserContext.getUserHolder(); +// Long workspaceServiceByProjectId = workspaceService.getByProjectId(query.getProjectId()); +// Workspace byId = workspaceService.getById(workspaceServiceByProjectId); + // 分页数据 + QueryWrapper queryWrapper = new QueryWrapper<>(); + queryWrapper.lambda().eq(ThreeDLayout::getType, "front"); + queryWrapper.lambda().eq(ThreeDLayout::getGender, query.getSex().toLowerCase()); + IPage page = threeDLayoutMapper.selectPage( + new Page<>(query.getPage(), query.getSize()), queryWrapper); + if (CollectionUtils.isEmpty(page.getRecords())) { + return new Page<>(); + } + + IPage convert = page.convert((Function) threeDLayout -> { + if (threeDLayout != null) { + ThreeDLayoutVO threeDLayoutVO = CopyUtil.copyObject(threeDLayout, ThreeDLayoutVO.class); + threeDLayoutVO.setUrl(minioUtil.getPreSignedUrl(threeDLayoutVO.getUrl(), 24 * 60)); + return threeDLayoutVO; + } + return null; + }); + return convert; + } + + @Override + public ThreeDVO getLayoutDetail(Long threeDSimpleId) { + ThreeDVO result = new ThreeDVO(); + + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(ThreeDLayout::getThreeDSimpleId, threeDSimpleId); + List threeDLayoutList = threeDLayoutMapper.selectList(qw); + for (ThreeDLayout threeDLayout : threeDLayoutList) { + threeDLayout.setUrl(minioUtil.getPreSignedUrl(threeDLayout.getUrl(), 24* 60)); + } + result.setThreeDLayoutList(threeDLayoutList); + + QueryWrapper threeDPatternLayoutQueryWrapper = new QueryWrapper<>(); + threeDPatternLayoutQueryWrapper.lambda().eq(ThreeDPatternLayout::getThreeDSimpleId, threeDSimpleId); + List threeDPatternLayoutList = threeDPatternLayoutMapper.selectList(threeDPatternLayoutQueryWrapper); + if (CollectionUtil.isNotEmpty(threeDPatternLayoutList)) { + ThreeDPatternLayout threeDPatternLayout = threeDPatternLayoutList.get(0); +// threeDPatternLayout.setUrl(minioUtil.getPreSignedUrl(threeDPatternLayout.getUrl(), 24 * 60)); + result.setThreeDPatternLayoutUrl(minioUtil.getPreSignedUrl(threeDPatternLayout.getUrl(), 24 * 60)); + } + ThreeDSimple threeDSimple = threeDSimpleMapper.selectById(threeDSimpleId); + result.setThreeDSimpleUrl(minioUtil.getPreSignedUrl(threeDSimple.getUrl(), 24 * 60)); + return result; + } + + @Override + public ThreeDSizeVO getThreeDSize(Long threeDSimpleId) { + ThreeDSizeVO result = new ThreeDSizeVO(); + QueryWrapper threeDDetailQueryWrapper = new QueryWrapper<>(); + threeDDetailQueryWrapper.lambda().eq(ThreeDDetail::getThreeDSimpleId, threeDSimpleId); + List threeDDetailList = threeDDetailMapper.selectList(threeDDetailQueryWrapper); + if (CollectionUtil.isNotEmpty(threeDDetailList)) { + // 以 sizeType 分组 + Map> sizeTypeMap = new HashMap<>(); + for (ThreeDDetail detail : threeDDetailList) { + sizeTypeMap + .computeIfAbsent(detail.getSizeType(), k -> new ArrayList<>()) + .add(detail.getSize()); + } + result.setSizeTypeMap(sizeTypeMap); + } + return result; + } + + @Override + public void getThreeDGlb(Long threeDSimpleId, HttpServletResponse response) throws MinioException, IOException { + ThreeDSimple threeDSimple = threeDSimpleMapper.selectById(threeDSimpleId); + if (ObjectUtils.isAllFieldNull(threeDSimple)) { + throw new BusinessException("3D file is not exist"); + } + InputStream inputStream = minioUtil.download(threeDSimple.getUrl()); + + // 设置响应头 + response.setContentType("model/gltf-binary"); // 确保 MIME 类型正确 + response.setHeader("Access-Control-Expose-Headers", "Content-Disposition"); + String fileName = threeDSimple.getName(); + if (!fileName.toLowerCase().endsWith(".glb")) { + fileName += ".glb"; + } + response.setHeader("Content-Disposition", "attachment; filename*=UTF-8''" + URLEncoder.encode(fileName, StandardCharsets.UTF_8.toString())); + // 将文件内容写入响应输出流 + try { + byte[] buffer = new byte[8192]; + int bytesRead; + while ((bytesRead = inputStream.read(buffer)) != -1) { + response.getOutputStream().write(buffer, 0, bytesRead); + } + inputStream.close(); + response.getOutputStream().flush(); + } catch (Exception e) { + throw new RuntimeException("Failed to get 3D .glb file", e); + } + } + + @Override + public String downloadZip(Long threeDSimpleId, String sizeType, String size, HttpServletResponse response) throws MinioException, IOException { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(ThreeDDetail::getThreeDSimpleId, threeDSimpleId); + qw.lambda().eq(ThreeDDetail::getSizeType, sizeType); + qw.lambda().eq(ThreeDDetail::getSize, size); + ThreeDDetail threeDDetail = threeDDetailMapper.selectOne(qw); + if (ObjectUtils.isAllFieldNull(threeDDetail)) { + throw new BusinessException("3D file is not exist"); + } + return minioUtil.getPreSignedUrl(threeDDetail.getUrl(), 24 * 60); + +// // 设置响应头 +// response.setContentType("application/zip"); // 确保 ZIP 格式 +// response.setHeader("Access-Control-Expose-Headers", "Content-Disposition"); +// String fileName = threeDDetail.getName(); +// if (!fileName.toLowerCase().endsWith(".zip")) { +// fileName += ".zip"; +// } +// response.setHeader("Content-Disposition", "attachment; filename*=UTF-8''" + URLEncoder.encode(fileName, StandardCharsets.UTF_8.toString())); +// +// // 将文件内容写入响应输出流 +// try { +// byte[] buffer = new byte[8192]; +// int bytesRead; +// while ((bytesRead = inputStream.read(buffer)) != -1) { +// response.getOutputStream().write(buffer, 0, bytesRead); +// } +// inputStream.close(); +// response.getOutputStream().flush(); +// } catch (Exception e) { +// throw new RuntimeException("Failed to download ZIP file", e); +// } + } + + @Override + public Boolean delete(Long projectId) { + projectService.removeById(projectId); + return Boolean.TRUE; + } + + @Override + public Boolean brandDNADelete(BrandDNADTO brandDNADTO) { + brandDNAMapper.deleteById(brandDNADTO.getId()); + return Boolean.TRUE; + } + + @Override + @Transactional(isolation = Isolation.SERIALIZABLE) + public void toProductBatch(String taskId, String url, String progress) throws InterruptedException { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(ToProductImageResult::getTaskId, taskId); + ToProductImageResult toProductImageResult = toProductImageResultMapper.selectOne(qw); + if (Objects.nonNull(toProductImageResult)) { + toProductImageResult.setUrl(url); + toProductImageResult.setStatus("Success"); + toProductImageResultMapper.updateById(toProductImageResult); + String taskIdBatch = toProductImageResult.getTaskIdBatch(); + log.info("toProductImage云生成 batchTaskId:" + taskIdBatch + " 完成数+1"); + + CloudTask task = cloudTaskMapper.selectForUpdate(taskIdBatch); + log.info("排查完成数量" + task.getCompletedNum()); + task.setCompletedNum(task.getCompletedNum() + 1); + cloudTaskMapper.updateById(task); + + String key = toProductImageResultKey + ":" + taskId; + String imageName = url.substring(url.lastIndexOf("/") + 1); + String status = imageName.equals("white_image.jpg") ? "Invalid" : "Success"; + GenerateResultVO generateResultVO = new GenerateResultVO(taskId, toProductImageResult.getId(), url, status, null); + redisUtil.addToString(key, new Gson().toJson(generateResultVO), CommonConstant.GENERATE_RESULT_EXPIRE_TIME); + + /*Long accountId = Long.parseLong(taskId.substring(taskId.lastIndexOf("-") + 1)); + if (!status.equals("Invalid")) { + // 4、扣除积分 + Boolean b = creditsService.taskCreditsDeduction(accountId, taskId); + // 3、记录积分变更 + if (b) creditsService.updateChangedCredits(String.valueOf(accountId), taskId); + }*/ + } + + } + + @Override + @Transactional + public void relightBatch(String taskId, String url, String progress) { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(ToProductImageResult::getTaskId, taskId); + ToProductImageResult toProductImageResult = toProductImageResultMapper.selectOne(qw); + if (Objects.nonNull(toProductImageResult)) { + if (toProductImageResult.getBrightenValue() != null && toProductImageResult.getBrightenValue() != 1.0) { + pythonService.bright(url, toProductImageResult.getBrightenValue()); + } + toProductImageResult.setUrl(url); +// toProductImageResult.setResultType("Relight"); + toProductImageResult.setStatus("Success"); + toProductImageResultMapper.updateById(toProductImageResult); + + String taskIdBatch = toProductImageResult.getTaskIdBatch(); +// QueryWrapper cloudTaskQueryWrapper = new QueryWrapper<>(); +// cloudTaskQueryWrapper.lambda().eq(CloudTask::getTaskId, taskIdBatch); +// CloudTask cloudTask = cloudTaskMapper.selectOne(cloudTaskQueryWrapper); +// if (Objects.nonNull(cloudTask)) { +// if (cloudTask.getCompletedNum() == null) { +// cloudTask.setCompletedNum(1); +// }else { +// cloudTask.setCompletedNum(cloudTask.getCompletedNum() + 1); +// } +// if (progress.equals("OK")) { +// cloudTask.setStatus(1); +// cloudTask.setCompletedNum(cloudTask.getNums()); +// } +// cloudTaskMapper.updateById(cloudTask); +// } + cloudTaskMapper.increaseCompletedNum(taskIdBatch); + + String key = relightResultKey + ":" + taskId; + String imageName = url.substring(url.lastIndexOf("/") + 1); + String status = imageName.equals("white_image.jpg") ? "Invalid" : "Success"; + GenerateResultVO generateResultVO = new GenerateResultVO(taskId, toProductImageResult.getId(), url, status, null); + redisUtil.addToString(key, new Gson().toJson(generateResultVO), CommonConstant.GENERATE_RESULT_EXPIRE_TIME); + + /*Long accountId = Long.parseLong(taskId.substring(taskId.lastIndexOf("-") + 1)); + if (!status.equals("Invalid")) { + // 4、扣除积分 + Boolean b = creditsService.taskCreditsDeduction(accountId, taskId); + // 3、记录积分变更 + if (b) creditsService.updateChangedCredits(String.valueOf(accountId), taskId); + }*/ + } + } + + @Override + @Transactional + public Boolean collectionLikeUpdate(CollectionLikeUpdateDTO collectionLikeUpdateDTO) { + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(CollectionSort::getRelationId, collectionLikeUpdateDTO.getOldRelationId()); + qw.lambda().eq(CollectionSort::getRelationType, collectionLikeUpdateDTO.getRelationType()); + CollectionSort collectionSort = collectionSortMapper.selectOne(qw); +// Long relationIdOld = collectionSort.getRelationId(); + collectionSort.setRelationId(collectionLikeUpdateDTO.getRelationId()); + collectionSortMapper.updateById(collectionSort); + if (collectionLikeUpdateDTO.getRelationType().equals(CollectionType.TO_PRODUCT_IMAGE.getValue()) || collectionLikeUpdateDTO.getRelationType().equals(CollectionType.RELIGHT.getValue())) { + ToProductImageResult toProductImageResult = toProductImageResultMapper.selectById(collectionLikeUpdateDTO.getRelationId()); + toProductImageResult.setIsLike(1); + toProductImageResultMapper.updateById(toProductImageResult); + ToProductImageResult toProductImageResultOld = toProductImageResultMapper.selectById(collectionLikeUpdateDTO.getOldRelationId()); + toProductImageResultOld.setIsLike(0); + toProductImageResultMapper.updateById(toProductImageResultOld); + }else if (collectionLikeUpdateDTO.getRelationType().equals(CollectionType.POSE_TRANSFORM.getValue())) { + PoseTransformation poseTransformation = poseTransformationMapper.selectById(collectionLikeUpdateDTO.getRelationId()); + poseTransformation.setIsLiked((byte) 1); + poseTransformationMapper.updateById(poseTransformation); + PoseTransformation poseTransformationOld = poseTransformationMapper.selectById(collectionLikeUpdateDTO.getOldRelationId()); + poseTransformationOld.setIsLiked((byte) 0); + poseTransformationMapper.updateById(poseTransformationOld); + } + return Boolean.TRUE; + } + + @Override + public Boolean toProductImageElementDelete(Long id) { + UpdateWrapper updateWrapper = new UpdateWrapper<>(); + updateWrapper + .set("project_id", null) // 显式设置 NULL + .eq("id", id); // 指定条件 + toProductElementMapper.update(null, updateWrapper); + return Boolean.TRUE; + } + + @Override + public ToProductElementVO convertRelightElement(Long id) { + ToProductImageResult toProductImageResult = toProductImageResultMapper.selectById(id); + + ToProductElement toProductElement = new ToProductElement(); + toProductElement.setUrl(toProductImageResult.getUrl()); +// toProductElement.setUserLikeGroupId(userLikeGroupId); + toProductElement.setProjectId(toProductImageResult.getProjectId()); + toProductElement.setCreateTime(LocalDateTime.now()); + toProductElementMapper.insert(toProductElement); + ToProductElementVO toProductElementVO = CopyUtil.copyObject(toProductElement, ToProductElementVO.class); + toProductElementVO.setMinioUrl(toProductElementVO.getUrl()); + toProductElementVO.setUrl(minioUtil.getPreSignedUrl(toProductElementVO.getUrl(), 24 * 60)); + return toProductElementVO; + } +}