由于一些业务的需要,我们需要增加异步处理的代码,提高代码的响应速度,我们这里简单的进行一个总结;
查看服务器的核心数
cat /proc/cpuinfo | grep 'core id' | sort | uniq | wc -l
查出的数量作为核心线程数,最大数量为 coreNum*2+1。
异步线程池代码
在service的方法上方需要增加@Async("myAsyncExecutor")的注解,同时在启动类或者配置类中需要增加@EnableAsync,我们可以集中放到一个类上,对于这里配置的参数,还需要根据项目的需要来进行修改,包括但不限于拒绝策略,等待时间等。
我在代码中增加了异常的处理日志打印,如果有业务增加通知可以放在此处实现。
java
mport lombok.extern.slf4j.Slf4j;
import org.springframework.aop.interceptor.AsyncUncaughtExceptionHandler;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.task.TaskExecutor;
import org.springframework.scheduling.annotation.AsyncConfigurer;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import java.lang.reflect.Method;
import java.util.concurrent.ThreadPoolExecutor;
@EnableAsync
@Slf4j
@Configuration
public class AsyncConfig implements AsyncConfigurer {
// 这里可以定义一些异步任务执行器的配置,如果默认的执行器满足需求,这个类可以保持空白
@Bean(name = "myAsyncExecutor")
public TaskExecutor asyncExecutor() {
ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
executor.setCorePoolSize(6);
executor.setMaxPoolSize(13);
executor.setQueueCapacity(100);
executor.setThreadNamePrefix("my-async-");
// 设置拒绝策略,直接在execute方法的调用线程中运行被拒绝的任务
executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
//线程池中 任务的等待时间
executor.setAwaitTerminationSeconds(60);
executor.setWaitForTasksToCompleteOnShutdown(true);
// 初始化线程池
executor.initialize();
return executor;
}
@Override
public AsyncUncaughtExceptionHandler getAsyncUncaughtExceptionHandler() {
return new AsyncUncaughtExceptionHandler() {
@Override
public void handleUncaughtException(Throwable ex, Method method, Object... params) {
//当目标方法执行过程中抛出异常的时候,此时会自动回调这个方法,可以在这个方法中处理异常
log.error("thread-name={}, method={}, exception={}",
Thread.currentThread().getName(), method.getName(), ex.getMessage());
}
};
}
}