diff --git a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java index bd307380..113288ff 100644 --- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java @@ -16,6 +16,7 @@ import com.ai.da.mapper.primary.*; import com.ai.da.mapper.primary.entity.*; 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.enums.SketchStyle; import com.ai.da.model.vo.*; import com.ai.da.python.PythonService; @@ -1243,6 +1244,8 @@ public class GenerateServiceImpl extends ServiceImpl i poseTransformationMapper.insert(poseTransformation); // 当需要默认like ToProductImageResultVO toProductImageResultVO = new ToProductImageResultVO(); + toProductImageResultVO.setParentId(poseTransformDTO.getParentId()); + toProductImageResultVO.setResultType(Module.poseTransfer.getValue()); if (Objects.nonNull(poseTransformDTO.getIsDefaultLike()) && poseTransformDTO.getIsDefaultLike()) { // 满足条件下添加到like poseTransformation.setIsLiked((byte) 1); @@ -1255,6 +1258,7 @@ public class GenerateServiceImpl extends ServiceImpl i } else if (Objects.nonNull(poseTransformDTO.getIsDefaultLike()) && Objects.nonNull(poseTransformDTO.getParentId())) { toProductImageResultVO.setParentId(poseTransformDTO.getParentId()); } + toProductImageResultVO.setTaskId(taskId); if (flag) { @@ -1262,7 +1266,6 @@ public class GenerateServiceImpl extends ServiceImpl i creditsService.addRecordToCreditsDeduction(accountId, taskId, creditsEventsEnum); // 6.1 添加积分扣除记录到db creditsService.preInsert(accountId, creditsEventsEnum.getName(), taskId, Boolean.TRUE, null); - toProductImageResultVO.setTaskId(taskId); // 更新项目更新时间 projectService.modifyProjectUpdateTime(projectId); return toProductImageResultVO;