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

TASK: 数据迁移拆分；

Browse Source
This commit is contained in:
shahaibo
2025-06-27 09:59:08 +08:00
parent a7347c9dc0
commit b01f5f59d1
3 changed files with 174 additions and 145 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/main/java/com/ai/da/service/impl/DesignServiceImpl.java
View File

@@ -2250,6 +2250,12 @@ public class DesignServiceImpl extends ServiceImpl<DesignMapper, Design> impleme
}
return batchTaskId;
} else if (cloudTaskDTO.getBuildType().equals(BuildType.POSE_TRANSFORM.getValue())) {
Boolean preDeduction = creditsService.creditsPreDeduction(CreditsEventsEnum.POSE_TRANSFORMATION, cloudTaskDTO.getNums());
if (!preDeduction) {
throw new BusinessException("Your remaining credits are insufficient for this generation. Please recharge.", ResultEnum.WARNING.getCode());
}
Long projectId = cloudTaskDTO.getProjectId();
AuthPrincipalVo userHolder = UserContext.getUserHolder();
if (null == projectId) {
@@ -2283,10 +2289,6 @@ public class DesignServiceImpl extends ServiceImpl<DesignMapper, Design> impleme
cloudTask.setStatus(0);
cloudTaskMapper.insert(cloudTask);
Boolean preDeduction = creditsService.creditsPreDeduction(CreditsEventsEnum.POSE_TRANSFORMATION, cloudTaskDTO.getNums());
if (!preDeduction) {
throw new BusinessException("Your remaining credits are insufficient for this generation. Please recharge.", ResultEnum.WARNING.getCode());
}
List<PoseTransformBatchDTO> poseTransformList = cloudTaskDTO.getPoseTransform();
if (CollectionUtil.isNotEmpty(poseTransformList)) {
for (PoseTransformBatchDTO poseTransformBatchDTO : poseTransformList) {