xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-04-13 20:11:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: 解决不登陆无法问答问题；修复余额不足后流不关闭问题；

Browse Source
This commit is contained in:
LM20230311
2025-09-24 20:27:55 +08:00
parent 2caf9a47ed
commit bd346f1e85
3 changed files with 16 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
ruoyi-common/ruoyi-common-core/src/main/java/org/ruoyi/common/core/config/ThreadPoolConfig.java
View File

@@ -1,5 +1,6 @@
package org.ruoyi.common.core.config; package org.ruoyi.common.core.config;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.concurrent.BasicThreadFactory; import org.apache.commons.lang3.concurrent.BasicThreadFactory;
import org.ruoyi.common.core.config.properties.ThreadPoolProperties; import org.ruoyi.common.core.config.properties.ThreadPoolProperties;
import org.ruoyi.common.core.utils.Threads; import org.ruoyi.common.core.utils.Threads;
@@ -18,6 +19,7 @@ import java.util.concurrent.ThreadPoolExecutor;
* *
* @author Lion Li * @author Lion Li
**/ **/
@Slf4j
@AutoConfiguration @AutoConfiguration
@EnableConfigurationProperties(ThreadPoolProperties.class) @EnableConfigurationProperties(ThreadPoolProperties.class)
public class ThreadPoolConfig { public class ThreadPoolConfig {
@@ -43,7 +45,8 @@ public class ThreadPoolConfig {
* 执行周期性或定时任务 * 执行周期性或定时任务
*/ */
@Bean(name = "scheduledExecutorService") @Bean(name = "scheduledExecutorService")
protected ScheduledExecutorService scheduledExecutorService() { public ScheduledExecutorService scheduledExecutorService() {
log.info("====创建定时任务线程池====");
return new ScheduledThreadPoolExecutor(core, return new ScheduledThreadPoolExecutor(core,
new BasicThreadFactory.Builder().namingPattern("schedule-pool-%d").daemon(true).build(), new BasicThreadFactory.Builder().namingPattern("schedule-pool-%d").daemon(true).build(),
new ThreadPoolExecutor.CallerRunsPolicy()) { new ThreadPoolExecutor.CallerRunsPolicy()) {

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

@@ -435,8 +435,8 @@ public class ChatCostServiceImpl implements IChatCostService {
@Override @Override
public boolean checkBalanceSufficient(ChatRequest chatRequest) { public boolean checkBalanceSufficient(ChatRequest chatRequest) {
if (chatRequest.getUserId() == null) { if (chatRequest.getUserId() == null) {
log.warn("checkBalanceSufficient->用户ID为空，视为余额不足"); log.warn("当前未登录");
return false; return true;
} }
try { try {

11
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/proxy/BillingChatServiceProxy.java
View File

@@ -34,7 +34,16 @@ public class BillingChatServiceProxy implements IChatService {
String errorMsg = "余额不足，无法使用AI服务，请充值后再试"; String errorMsg = "余额不足，无法使用AI服务，请充值后再试";
log.warn("余额不足阻止AI回复，用户ID: {}, 模型: {}", log.warn("余额不足阻止AI回复，用户ID: {}, 模型: {}",
chatRequest.getUserId(), chatRequest.getModel()); chatRequest.getUserId(), chatRequest.getModel());
throw new RuntimeException(errorMsg); try {
emitter.send(errorMsg);
emitter.complete();
} catch (IOException e) {
log.error("推送流异常，用户ID: {}, 模型: {}",
chatRequest.getUserId(), chatRequest.getModel());
emitter.complete();
throw new RuntimeException(errorMsg);
}
return emitter;
} }
log.debug("余额检查通过，开始AI回复，用户ID: {}, 模型: {}", log.debug("余额检查通过，开始AI回复，用户ID: {}, 模型: {}",