open-source/MallChat
1
0
Fork 0
mirror of https://github.com/zongzibinbin/MallChat.git synced 2026-03-13 21:53:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

chatai相关线程池配置添加

Browse Source
This commit is contained in:
Babelyj
2023-07-02 21:30:38 +08:00
parent b12be7c6df
commit c87b968735
2 changed files with 23 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
mallchat-common/src/main/java/com/abin/mallchat/common/common/config/ThreadPoolConfig.java
View File

@@ -1,9 +1,6 @@
package com.abin.mallchat.common.common.config;
import cn.hutool.core.thread.NamedThreadFactory;
import cn.hutool.core.thread.ThreadFactoryBuilder;
import com.abin.mallchat.common.common.factory.MyThreadFactory;
import com.abin.mallchat.common.common.handler.GlobalUncaughtExceptionHandler;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
@@ -11,9 +8,7 @@ import org.springframework.scheduling.annotation.AsyncConfigurer;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import java.util.concurrent.Executor;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.*;
/**
* Description: 线程池配置
@@ -32,6 +27,9 @@ public class ThreadPoolConfig implements AsyncConfigurer {
*/
public static final String WS_EXECUTOR = "websocketExecutor";
public static final String AICHAT_EXECUTOR = "aichatExecutor";
@Override
public Executor getAsyncExecutor() {
return mallchatExecutor();
@@ -64,5 +62,15 @@ public class ThreadPoolConfig implements AsyncConfigurer {
return executor;
}
@Bean(AICHAT_EXECUTOR)
public ThreadPoolTaskExecutor chatAiExecutor() {
ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
executor.setCorePoolSize(10);
executor.setMaxPoolSize(10);
executor.setQueueCapacity(15);
executor.setThreadNamePrefix("aichat-executor-");
executor.setRejectedExecutionHandler(new ThreadPoolExecutor.DiscardPolicy());//满了直接丢弃，默认为不重要消息推送
executor.setThreadFactory(new MyThreadFactory(executor));
return executor;
}
}