TASK:pose transfer添加返回参数

This commit is contained in:
2025-07-20 19:19:07 +08:00
parent e2e0bd6b47
commit 2041809ea6

View File

@@ -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<GenerateMapper, Generate> 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<GenerateMapper, Generate> 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<GenerateMapper, Generate> 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;