63 lines
2.0 KiB
Java
63 lines
2.0 KiB
Java
package com.emotionmuseum.config;
|
|
|
|
import org.springframework.aop.interceptor.AsyncUncaughtExceptionHandler;
|
|
import org.springframework.context.annotation.Bean;
|
|
import org.springframework.context.annotation.Configuration;
|
|
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.ThreadPoolExecutor;
|
|
|
|
/**
|
|
* 异步配置类
|
|
*
|
|
* @author emotion-museum
|
|
* @version 1.0.0
|
|
* @since 2024-01-01
|
|
*/
|
|
@Configuration
|
|
@EnableAsync
|
|
public class AsyncConfig implements AsyncConfigurer {
|
|
|
|
/**
|
|
* 默认异步执行器
|
|
*/
|
|
@Override
|
|
public Executor getAsyncExecutor() {
|
|
ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
|
|
executor.setCorePoolSize(20);
|
|
executor.setMaxPoolSize(100);
|
|
executor.setQueueCapacity(500);
|
|
executor.setKeepAliveSeconds(60);
|
|
executor.setThreadNamePrefix("emotion-async-");
|
|
executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
|
|
executor.initialize();
|
|
return executor;
|
|
}
|
|
|
|
/**
|
|
* AI任务执行器
|
|
*/
|
|
@Bean("aiTaskExecutor")
|
|
public Executor aiTaskExecutor() {
|
|
ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
|
|
executor.setCorePoolSize(10);
|
|
executor.setMaxPoolSize(50);
|
|
executor.setQueueCapacity(200);
|
|
executor.setKeepAliveSeconds(60);
|
|
executor.setThreadNamePrefix("ai-task-");
|
|
executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
|
|
executor.initialize();
|
|
return executor;
|
|
}
|
|
|
|
/**
|
|
* 异步异常处理器
|
|
*/
|
|
@Override
|
|
public AsyncUncaughtExceptionHandler getAsyncUncaughtExceptionHandler() {
|
|
return new org.springframework.aop.interceptor.SimpleAsyncUncaughtExceptionHandler();
|
|
}
|
|
} |