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-20 19:19:07 +08:00
parent e2e0bd6b47
commit 2041809ea6
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

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