From c5c375dc6d091b8fd60856aee46dd92ecd53ff7c Mon Sep 17 00:00:00 2001
From: ageerle <ageerle@163.com>
Date: 星期二, 22 四月 2025 10:43:54 +0800
Subject: [PATCH] fix: 扣费时无法获取用户id
---
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/ChatCostServiceImpl.java | 10 +++++-----
1 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/ChatCostServiceImpl.java b/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/ChatCostServiceImpl.java
index 619a175..2a4e5f0 100644
--- a/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/ChatCostServiceImpl.java
+++ b/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/ChatCostServiceImpl.java
@@ -61,7 +61,7 @@
chatMessageBo.setUserId(getUserId());
}
// 璁＄畻鎬籺oken鏁�
- ChatToken chatToken = chatTokenService.queryByUserId(getUserId(), modelName);
+ ChatToken chatToken = chatTokenService.queryByUserId(chatMessageBo.getUserId(), modelName);
if (chatToken == null) {
chatToken = new ChatToken();
chatToken.setToken(0);
@@ -75,17 +75,17 @@
if (token2 > 0) {
// 淇濆瓨鍓╀綑tokens
chatToken.setModelName(modelName);
- chatToken.setUserId(getUserId());
+ chatToken.setUserId(chatMessageBo.getUserId());
chatToken.setToken(token2);
chatTokenService.editToken(chatToken);
} else {
- chatTokenService.resetToken(getUserId(), modelName);
+ chatTokenService.resetToken(chatMessageBo.getUserId(), modelName);
}
ChatModelVo chatModelVo = chatModelService.selectModelByName(modelName);
double cost = chatModelVo.getModelPrice();
if (BillingType.TIMES.getCode().equals(chatModelVo.getModelType())) {
// 鎸夋鏁版墸璐�
- deductUserBalance(getUserId(), cost);
+ deductUserBalance(chatMessageBo.getUserId(), cost);
chatMessageBo.setDeductCost(cost);
}else {
// 鎸塼oken鎵ｈ垂
@@ -95,7 +95,7 @@
}
chatMessageBo.setContent(chatRequest.getPrompt());
} else {
- deductUserBalance(getUserId(), 0.0);
+ deductUserBalance(chatMessageBo.getUserId(), 0.0);
chatMessageBo.setDeductCost(0d);
chatMessageBo.setRemark("涓嶆弧1kToken,璁″叆涓嬩竴娆�!");
chatToken.setToken(totalTokens);
--
Gitblit v1.9.3