fix: 扣费时无法获取用户id

ageerle

2025-04-22 c5c375dc6d091b8fd60856aee46dd92ecd53ff7c

fix: 扣费时无法获取用户id

 ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/ChatCostServiceImpl.java

@@ -61,7 +61,7 @@
            chatMessageBo.setUserId(getUserId());
        }
        // 计算总token数
        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 {
                // 按token扣费
@@ -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);