TASK:pose transfer添加返回参数
This commit is contained in:
@@ -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;
|
||||
|
||||
Reference in New Issue
Block a user