From aca21b9d98eb4fba281b3597cf8d211ee6b972a6 Mon Sep 17 00:00:00 2001 From: shahaibo <1023316923@qq.com> Date: Tue, 22 Apr 2025 15:26:55 +0800 Subject: [PATCH] =?UTF-8?q?TASK:=E6=A8=A1=E5=9D=97=E5=8C=96=EF=BC=9B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/main/java/com/ai/da/model/enums/BuildType.java | 2 +- src/main/java/com/ai/da/python/PythonService.java | 2 +- .../com/ai/da/service/impl/DesignServiceImpl.java | 12 ++++++++++-- 3 files changed, 12 insertions(+), 4 deletions(-) diff --git a/src/main/java/com/ai/da/model/enums/BuildType.java b/src/main/java/com/ai/da/model/enums/BuildType.java index 4265949b..ed097b55 100644 --- a/src/main/java/com/ai/da/model/enums/BuildType.java +++ b/src/main/java/com/ai/da/model/enums/BuildType.java @@ -6,7 +6,7 @@ public enum BuildType implements IEnumDisplay { DESIGN("design"), TO_PRODUCT_IMAGE("toProductImage"), RELIGHT("relight"), - POSE_TRANSFORM("poseTransform") + POSE_TRANSFORM("poseTransfer") ; private String value; diff --git a/src/main/java/com/ai/da/python/PythonService.java b/src/main/java/com/ai/da/python/PythonService.java index 4aa68528..8b3a503a 100644 --- a/src/main/java/com/ai/da/python/PythonService.java +++ b/src/main/java/com/ai/da/python/PythonService.java @@ -4360,7 +4360,7 @@ public class PythonService { log.info("poseTransformation 请求地址： {}", accessPythonIp + ":" + accessPythonPort + "/api/batch_generate_pose_transform_image"); Request request = new Request.Builder() - .url(accessPythonIp + ":" + accessPythonPort + "/api/pose_transform") + .url(accessPythonIp + ":" + accessPythonPort + "/api/batch_generate_pose_transform_image") .method("POST", body) .addHeader("Content-Type", "application/json") .build(); diff --git a/src/main/java/com/ai/da/service/impl/DesignServiceImpl.java b/src/main/java/com/ai/da/service/impl/DesignServiceImpl.java index 765a6a40..75c8238c 100644 --- a/src/main/java/com/ai/da/service/impl/DesignServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/DesignServiceImpl.java @@ -2098,10 +2098,18 @@ public class DesignServiceImpl extends ServiceImpl impleme creditsService.addRecordToCreditsDeduction(accountId, uuid, creditsEventsEnum); // 6.1 添加积分扣除记录到db creditsService.preInsert(accountId, creditsEventsEnum.getName(), uuid, Boolean.TRUE, null); - return taskId; + }else { + throw new BusinessException("pose transformation error", ResultEnum.ERROR.getCode()); } - throw new BusinessException("pose transformation error", ResultEnum.ERROR.getCode()); } + CloudTask cloudTask = CopyUtil.copyObject(cloudTaskDTO, CloudTask.class); + cloudTask.setProjectId(cloudTask.getProjectId()); + cloudTask.setTaskId(taskBatchId); + LocalDateTime now = LocalDateTime.now(); + cloudTask.setCreateTime(now); + cloudTask.setUpdateTime(now); + cloudTask.setStatus(0); + cloudTaskMapper.insert(cloudTask); return taskBatchId; } }