From c7334ab7d97c96fc555c1fe918cf9061badc1faf Mon Sep 17 00:00:00 2001 From: old burden Date: Mon, 20 Apr 2026 16:28:27 +0800 Subject: [PATCH] =?UTF-8?q?fix=EF=BC=9A=E7=B3=BB=E7=BB=9F=E7=94=A8?= =?UTF-8?q?=E6=88=B7=E8=A1=A8=E5=AE=9E=E7=8E=B0=E9=80=9A=E8=BF=87=E9=83=A8?= =?UTF-8?q?=E9=97=A8=E5=85=B3=E8=81=94ai=5Fuser?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../web/controller/ai/AiOrderController.java | 2 +- .../ai/service/impl/AiOrderServiceImpl.java | 18 ++++++++++++++++++ .../ai/service/impl/AiUserServiceImpl.java | 2 +- 3 files changed, 20 insertions(+), 2 deletions(-) diff --git a/web-api/ruoyi-admin/src/main/java/com/ruoyi/web/controller/ai/AiOrderController.java b/web-api/ruoyi-admin/src/main/java/com/ruoyi/web/controller/ai/AiOrderController.java index eeea47d..bcb817e 100644 --- a/web-api/ruoyi-admin/src/main/java/com/ruoyi/web/controller/ai/AiOrderController.java +++ b/web-api/ruoyi-admin/src/main/java/com/ruoyi/web/controller/ai/AiOrderController.java @@ -24,7 +24,7 @@ import java.util.List; * @date 2025-11-13 */ @RestController -@RequestMapping("/ai/order") +@RequestMapping("/ai/video/order") public class AiOrderController extends BaseController { @Autowired diff --git a/web-api/ruoyi-system/src/main/java/com/ruoyi/ai/service/impl/AiOrderServiceImpl.java b/web-api/ruoyi-system/src/main/java/com/ruoyi/ai/service/impl/AiOrderServiceImpl.java index d06659b..46a12d7 100644 --- a/web-api/ruoyi-system/src/main/java/com/ruoyi/ai/service/impl/AiOrderServiceImpl.java +++ b/web-api/ruoyi-system/src/main/java/com/ruoyi/ai/service/impl/AiOrderServiceImpl.java @@ -59,6 +59,9 @@ public class AiOrderServiceImpl implements IAiOrderService { @Autowired private IByteDeptApiKeyService byteDeptApiKeyService; + @Autowired + private IAiVideoReportDataService aiVideoReportDataService; + // 流水表：任务成功时回补 private static final String TASK_SUCCESS_BACK_FILL_REMARK = "order.number.generation.successbackfill"; @@ -244,6 +247,7 @@ public class AiOrderServiceImpl implements IAiOrderService { public void orderSuccess(AiOrder aiOrder) { aiOrder.setStatus(1); aiOrderMapper.updateById(aiOrder); + syncVideoReportData(aiOrder); AiStatistics aiStatistics = new AiStatistics(); aiStatistics.setSource(aiOrder.getSource()); aiStatistics.setGenerateCount(1L); @@ -349,4 +353,18 @@ public class AiOrderServiceImpl implements IAiOrderService { // BalanceChangerConstants.QUICK_VIDEO_GENERATION, TASK_SUCCESS_BALANCE_REMARK); return AjaxResult.success("callback success"); } + + /** + * 任务成功后按“提交时间”写入团队消耗统计来源表。 + * + * @param aiOrder 已成功订单 + */ + private void syncVideoReportData(AiOrder aiOrder) { + if (aiOrder == null || aiOrder.getCreateTime() == null || aiOrder.getDeptId() == null || aiOrder.getUserId() == null) { + return; + } + BigDecimal score = aiOrder.getAmount() != null ? aiOrder.getAmount() : BigDecimal.ZERO; + Long useTokens = aiOrder.getTotalUsage() != null ? aiOrder.getTotalUsage().longValue() : 0L; + aiVideoReportDataService.syncVideoConsumeIncrement(aiOrder.getCreateTime(), aiOrder.getDeptId(), aiOrder.getUserId(), score, useTokens); + } } diff --git a/web-api/ruoyi-system/src/main/java/com/ruoyi/ai/service/impl/AiUserServiceImpl.java b/web-api/ruoyi-system/src/main/java/com/ruoyi/ai/service/impl/AiUserServiceImpl.java index 432c927..1b12ca5 100644 --- a/web-api/ruoyi-system/src/main/java/com/ruoyi/ai/service/impl/AiUserServiceImpl.java +++ b/web-api/ruoyi-system/src/main/java/com/ruoyi/ai/service/impl/AiUserServiceImpl.java @@ -303,7 +303,7 @@ public class AiUserServiceImpl implements IAiUserService { private AiUser getUserByInvitationCode(String invitationCode) { LambdaQueryWrapper query = Wrappers.lambdaQuery(); query.eq(AiUser::getInvitationCode, invitationCode); - query.eq(AiUser::getDelFlag, 0); + query.eq(AiUser::getDelFlag, "0"); return aiUserMapper.selectOne(query); }