aidlab/aida_back
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

TASK:模块化；

Browse Source
This commit is contained in:
shahaibo
2025-04-21 22:13:53 +08:00
parent 98e1f37c1d
commit dfe99ef080
2 changed files with 25 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/main/java/com/ai/da/model/vo/CloudTaskResultVO.java
View File

@@ -1,13 +1,14 @@
package com.ai.da.model.vo;
import com.ai.da.common.response.PageBaseResponse;
import com.ai.da.mapper.primary.entity.TDesignPythonOutfit;
import lombok.Data;
import java.util.List;
@Data
public class CloudTaskResultVO {
private PageBaseResponse<DesignCollectionItemVO> design;
private List<DesignCollectionItemVO> design;
private List<MagicToolResultVO> toProductImage;
private List<MagicToolResultVO> relight;
}

37
src/main/java/com/ai/da/service/impl/DesignServiceImpl.java
View File

@@ -2501,25 +2501,34 @@ public class DesignServiceImpl extends ServiceImpl<DesignMapper, Design> impleme
if (null != collectionId) {
QueryWrapper<TDesignPythonOutfit> qw = new QueryWrapper<>();
qw.lambda().eq(TDesignPythonOutfit::getCollectionId, collectionId);
Page<TDesignPythonOutfit> page = designPythonOutfitMapper.selectPage(new Page<>(query.getPage(), query.getSize()), qw);
IPage<DesignCollectionItemVO> convert = page.convert((Function<TDesignPythonOutfit, DesignCollectionItemVO>) outfit -> {
if (outfit != null) {
DesignCollectionItemVO vo = new DesignCollectionItemVO();
vo.setDesignOutfitId(outfit.getId());
vo.setDesignOutfitUrl(minioUtil.getPreSignedUrl(outfit.getDesignUrl(), 24 * 60));
vo.setDesignItemId(outfit.getDesignItemId());
return vo;
}
return null;
});
result.setDesign(PageBaseResponse.success(convert));
List<TDesignPythonOutfit> list = designPythonOutfitMapper.selectList(qw);
List<DesignCollectionItemVO> voList = new ArrayList<>();
for (TDesignPythonOutfit tDesignPythonOutfit : list) {
DesignCollectionItemVO vo = new DesignCollectionItemVO();
vo.setDesignOutfitId(tDesignPythonOutfit.getId());
vo.setDesignOutfitUrl(minioUtil.getPreSignedUrl(tDesignPythonOutfit.getDesignUrl(), 24 * 60));
vo.setDesignItemId(tDesignPythonOutfit.getDesignItemId());
voList.add(vo);
}
result.setDesign(voList);
// IPage<DesignCollectionItemVO> convert = page.convert((Function<TDesignPythonOutfit, DesignCollectionItemVO>) outfit -> {
// if (outfit != null) {
// DesignCollectionItemVO vo = new DesignCollectionItemVO();
// vo.setDesignOutfitId(outfit.getId());
// vo.setDesignOutfitUrl(minioUtil.getPreSignedUrl(outfit.getDesignUrl(), 24 * 60));
// vo.setDesignItemId(outfit.getDesignItemId());
// return vo;
// }
// return null;
// });
// result.setDesign(PageBaseResponse.success(convert));
return result;
}else {
result.setDesign(PageBaseResponse.success(new Page<>()));
result.setDesign(new ArrayList<>());
return result;
}
}
result.setDesign(PageBaseResponse.success(new Page<>()));
result.setDesign(new ArrayList<>());
return result;
} else if (query.getBuildType().equals(BuildType.TO_PRODUCT_IMAGE.getValue())) {
List<MagicToolResultVO> results = new ArrayList<>();