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

BUGFIX: cloud;

Browse Source
This commit is contained in:
shahaibo
2025-06-13 15:09:32 +08:00
parent b6c43e534b
commit 4cca30ec75
2 changed files with 68 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1893,6 +1893,13 @@ public class DesignServiceImpl extends ServiceImpl<DesignMapper, Design> impleme
project.setUpdateTime(now);
projectService.save(project);
projectId = project.getId();
for (ToProductImageVO toProductImageVO : toProductImageDTO.getToProductImageVOList()) {
if (toProductImageVO.getElementType().equals("ProductElement")) {
ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageVO.getElementId());
toProductElement.setProjectId(projectId);
toProductElementMapper.updateById(toProductElement);
}
}
}
UserLikeGroup userLikeGroup = userLikeGroupService.getByProjectId(projectId);
Long userLikeGroupId = null;
@@ -1993,8 +2000,6 @@ public class DesignServiceImpl extends ServiceImpl<DesignMapper, Design> impleme
}else {
sb.append(",high quality clothing details,").append(prompt).append(",8K realistic,HDR");
}
taskId = UUID.randomUUID() + "-" + i + "-" + userHolder.getId();
ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageVO.getElementId());
// 走模型
List<BatchParamDTO> paramList = new ArrayList<>();
@@ -2065,6 +2070,13 @@ public class DesignServiceImpl extends ServiceImpl<DesignMapper, Design> impleme
project.setUpdateTime(now);
projectService.save(project);
projectId = project.getId();
for (ToProductImageVO toProductImageVO : toProductImageDTO.getToProductImageVOList()) {
if (toProductImageVO.getElementType().equals("ProductElement")) {
ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageVO.getElementId());
toProductElement.setProjectId(projectId);
toProductElementMapper.updateById(toProductElement);
}
}
}
UserLikeGroup userLikeGroup = userLikeGroupService.getByProjectId(projectId);
Long userLikeGroupId = null;
@@ -2198,6 +2210,27 @@ public class DesignServiceImpl extends ServiceImpl<DesignMapper, Design> impleme
cloudTaskMapper.insert(cloudTask);
return batchTaskId;
} else if (cloudTaskDTO.getBuildType().equals(BuildType.POSE_TRANSFORM.getValue())) {
Long projectId = cloudTaskDTO.getProjectId();
AuthPrincipalVo userHolder = UserContext.getUserHolder();
if (null == projectId) {
Project project = new Project();
project.setName(DesignProcess.POSE_TRANSFER.getValue());
project.setProcess(DesignProcess.POSE_TRANSFER.name());
project.setOriginal(1);
project.setAccountId(userHolder.getId());
LocalDateTime now = LocalDateTime.now();
project.setCreateTime(now);
project.setUpdateTime(now);
projectService.save(project);
projectId = project.getId();
for (ToProductImageVO toProductImageVO : cloudTaskDTO.getToProductImage().getToProductImageVOList()) {
if (toProductImageVO.getElementType().equals("ProductElement")) {
ToProductElement toProductElement = toProductElementMapper.selectById(toProductImageVO.getElementId());
toProductElement.setProjectId(projectId);
toProductElementMapper.updateById(toProductElement);
}
}
}
Long accountId = UserContext.getUserHolder().getId();
Boolean preDeduction = creditsService.creditsPreDeduction(Integer.parseInt(CreditsEventsEnum.POSE_TRANSFORMATION.getValue()) * cloudTaskDTO.getNums());
if (!preDeduction) {
@@ -2215,7 +2248,7 @@ public class DesignServiceImpl extends ServiceImpl<DesignMapper, Design> impleme
String taskId = uuid + "-" + accountId;
PoseTransformation poseTransformation = new PoseTransformation();
poseTransformation.setProjectId(cloudTaskDTO.getProjectId());
poseTransformation.setProjectId(projectId);
poseTransformation.setAccountId(accountId);
poseTransformation.setUniqueId(taskId);
poseTransformation.setProductImage(poseTransformBatchDTO.getProductImage());