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

TASK:pose transfer添加返回参数

Browse Source
This commit is contained in:
xupei
2025-07-21 17:11:10 +08:00
parent 2041809ea6
commit f0fccaaf51
2 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/main/java/com/ai/da/model/vo/ToProductImageResultVO.java
View File

@@ -14,4 +14,6 @@ public class ToProductImageResultVO extends ToProductImageResult implements AllC
private Long parentId; private Long parentId;
@ApiModelProperty("用户输入的提示词") @ApiModelProperty("用户输入的提示词")
private String prompt; private String prompt;
@ApiModelProperty("姿势序号")
private Integer poseId;
} }

7
src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
View File

@@ -1246,6 +1246,12 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
ToProductImageResultVO toProductImageResultVO = new ToProductImageResultVO(); ToProductImageResultVO toProductImageResultVO = new ToProductImageResultVO();
toProductImageResultVO.setParentId(poseTransformDTO.getParentId()); toProductImageResultVO.setParentId(poseTransformDTO.getParentId());
toProductImageResultVO.setResultType(Module.poseTransfer.getValue()); toProductImageResultVO.setResultType(Module.poseTransfer.getValue());
toProductImageResultVO.setTaskId(taskId);
toProductImageResultVO.setTaskStatus(taskStatus);
toProductImageResultVO.setSourceUrl(minioUtil.getPreSignedUrl(productImage, CommonConstant.MINIO_IMAGE_EXPIRE_TIME));
toProductImageResultVO.setPoseId(poseId);
toProductImageResultVO.setModelName(poseTransformDTO.getModelName());
if (Objects.nonNull(poseTransformDTO.getIsDefaultLike()) && poseTransformDTO.getIsDefaultLike()) { if (Objects.nonNull(poseTransformDTO.getIsDefaultLike()) && poseTransformDTO.getIsDefaultLike()) {
// 满足条件下添加到like // 满足条件下添加到like
poseTransformation.setIsLiked((byte) 1); poseTransformation.setIsLiked((byte) 1);
@@ -1258,7 +1264,6 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
} else if (Objects.nonNull(poseTransformDTO.getIsDefaultLike()) && Objects.nonNull(poseTransformDTO.getParentId())) { } else if (Objects.nonNull(poseTransformDTO.getIsDefaultLike()) && Objects.nonNull(poseTransformDTO.getParentId())) {
toProductImageResultVO.setParentId(poseTransformDTO.getParentId()); toProductImageResultVO.setParentId(poseTransformDTO.getParentId());
} }
toProductImageResultVO.setTaskId(taskId);
if (flag) { if (flag) {