From be019dcdd04f4a4e697106d56978c39969aab6c3 Mon Sep 17 00:00:00 2001 From: shahaibo <1023316923@qq.com> Date: Fri, 2 May 2025 14:26:21 +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/service/impl/DesignServiceImpl.java | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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 912865f0..b94820ea 100644 --- a/src/main/java/com/ai/da/service/impl/DesignServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/DesignServiceImpl.java @@ -1813,7 +1813,7 @@ public class DesignServiceImpl extends ServiceImpl impleme }else if (cloudTaskDTO.getBuildType().equals(BuildType.TO_PRODUCT_IMAGE.getValue())) { ToProductImageDTO toProductImageDTO = cloudTaskDTO.getToProductImage(); // 判断用户当前积分是否够本次生成消耗 - Boolean preDeduction = creditsService.creditsPreDeduction(cloudTaskDTO.getCostCredits()); + Boolean preDeduction = creditsService.creditsPreDeduction(Integer.parseInt(CreditsEventsEnum.TO_PRODUCT_IMAGE.getValue()) * cloudTaskDTO.getNums()); if (!preDeduction) { throw new BusinessException("Your remaining credits are insufficient for this generation. Please recharge.", ResultEnum.WARNING.getCode()); } @@ -2035,7 +2035,7 @@ public class DesignServiceImpl extends ServiceImpl impleme }else if (cloudTaskDTO.getBuildType().equals(BuildType.RELIGHT.getValue())) { ToProductImageDTO toProductImageDTO = cloudTaskDTO.getToProductImage(); // 判断用户当前积分是否够本次生成消耗 - Boolean preDeduction = creditsService.creditsPreDeduction(cloudTaskDTO.getCostCredits()); + Boolean preDeduction = creditsService.creditsPreDeduction(Integer.parseInt(CreditsEventsEnum.RELIGHT.getValue()) * cloudTaskDTO.getNums()); if (!preDeduction) { throw new BusinessException("Your remaining credits are insufficient for this generation. Please recharge.", ResultEnum.WARNING.getCode()); } @@ -2142,7 +2142,7 @@ public class DesignServiceImpl extends ServiceImpl impleme return batchTaskId; } else if (cloudTaskDTO.getBuildType().equals(BuildType.POSE_TRANSFORM.getValue())) { Long accountId = UserContext.getUserHolder().getId(); - Boolean preDeduction = creditsService.creditsPreDeduction(cloudTaskDTO.getCostCredits()); + Boolean preDeduction = creditsService.creditsPreDeduction(Integer.parseInt(CreditsEventsEnum.POSE_TRANSFORMATION.getValue()) * cloudTaskDTO.getNums()); if (!preDeduction) { throw new BusinessException("Your remaining credits are insufficient for this generation. Please recharge.", ResultEnum.WARNING.getCode()); }