package xiaozhi.common.config;
|
|
import java.util.concurrent.Executor;
|
import java.util.concurrent.RejectedExecutionHandler;
|
import java.util.concurrent.ThreadPoolExecutor;
|
|
import org.springframework.context.annotation.Bean;
|
import org.springframework.context.annotation.Configuration;
|
import org.springframework.context.annotation.EnableAspectJAutoProxy;
|
import org.springframework.scheduling.annotation.EnableAsync;
|
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
|
|
@Configuration
|
@EnableAsync
|
@EnableAspectJAutoProxy(exposeProxy = true)
|
public class AsyncConfig {
|
|
@Bean(name = "taskExecutor")
|
public Executor taskExecutor() {
|
ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
|
executor.setCorePoolSize(2);
|
executor.setMaxPoolSize(4);
|
executor.setQueueCapacity(1000);
|
executor.setThreadNamePrefix("AsyncThread-");
|
// 设置拒绝策略：由调用线程执行
|
executor.setRejectedExecutionHandler(new RejectedExecutionHandler() {
|
@Override
|
public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
|
try {
|
// 如果线程池已满，则由调用线程执行
|
r.run();
|
} catch (Exception e) {
|
throw new RuntimeException("执行异步任务失败", e);
|
}
|
}
|
});
|
executor.initialize();
|
return executor;
|
}
|
}
|
