在Springboot中,针对多线程以及线程池,提供了@EnableAsync以及@Async两个注解,只需要简单配置即可完成多线程、线程池的开发。
1 线程池配置
Springboot提供了多线程的配置接口,只要实现该接口,并使用@Configuration和@EnableAsync进行注解,即可定义一个线程池。
@Configuration
@ComponentScan("com")
@EnableAsync
public class CustomMultiThreadingConfig implements AsyncConfigurer {
@Override
public Executor getAsyncExecutor() {
ThreadPoolTaskExecutor taskExecutor = new ThreadPoolTaskExecutor();
//如果池中的实际线程数小于corePoolSize,无论是否其中有空闲的线程,都会给新的任务产生新的线程
taskExecutor.setCorePoolSize(5);
//queueCapacity 线程池所使用的缓冲队列
taskExecutor.setQueueCapacity(256);
//连接池中保留的最大连接数。
taskExecutor.setMaxPoolSize(1024);
//线程名称前缀
taskExecutor.setThreadNamePrefix("Application Thread-");
taskExecutor.initialize();
return taskExecutor;
}
@Override
public AsyncUncaughtExceptionHandler getAsyncUncaughtExceptionHandler() {
return AsyncConfigurer.super.getAsyncUncaughtExceptionHandler();
}
}
ThreadPoolTaskExecutor的参数:
- int corePoolSize:线程池维护线程的最小数量.
- int maximumPoolSize:线程池维护线程的最大数量.
- long keepAliveTime:空闲线程的存活时间.
- TimeUnit unit: 时间单位,现有纳秒,微秒,毫秒,秒枚举值.
- BlockingQueue<Runnable> workQueue:持有等待执行的任务队列.
- RejectedExecutionHandler handler: 用来拒绝一个任务的执行,有两种情况会发生这种情况。
- 一是在execute方法中若addIfUnderMaximumPoolSize(command)为false,即线程池已经饱和;
- 二是在execute方法中, 发现runState!=RUNNING || poolSize == 0,即已经shutdown,就调用ensureQueuedTaskHandled(Runnable command),在该方法中有可能调用reject。
2 定义异步任务
使用的方式非常简单,一个标注即可解决所有的问题:
@Async
public void demo() {
...
}
也可以将注解加到带参数的方法上:
@Async
public void demo(int i) {
...
}
3 在线程中注入Bean的方法
在Springboot中定义多线程并不复杂,网上能够检索的内容很多,基本误导的东西很少。但实际开发中会遇到一些不易发现的问题。
个人使用场景时这样的:
- 自己实现了一个时序数据流计算的小框架
- 不同的算法对应的不同的计算任务,也就是需要单独启动一个线程
- 算法名称和类名称对应,通过反射机制启动
但因为每个计算都是单独启动的线程,由于业务逻辑较为复杂,在线程中需要@Autowired注入Bean,但在方法调用的时候,会抛出ava.lang.NullPointerException异常。
解决思路
针对上述问题的解决方法也很简单——实现 org.springframework.context.ApplicationContextAware接口
网上找到个链接,里面解释的很好
https://www.cnblogs.com/codecat/p/11149893.html
“ApplicationContextAware的作用是可以方便获取Spring容器ApplicationContext,从而可以获取容器内的Bean。ApplicationContextAware接口只有一个方法setApplicationContext,如果实现了这个方法,那么Spring创建这个实现类的时候就会自动执行这个方法,把ApplicationContext注入到这个类中,也就是说,spring 在启动的时候就需要实例化这个 class(如果是懒加载就是你需要用到的时候实例化),在实例化这个 class 的时候,发现它包含这个 ApplicationContextAware 接口的话,sping 就会调用这个对象的 setApplicationContext 方法,把 applicationContext Set 进去了。”
代码如下:
@Component
public class ApplicationContextProvider implements ApplicationContextAware {
/**
* 上下文对象
*/
private static ApplicationContext applicationContext;
/**
* 设置上下文对象
* @param applicationContext 应用上下文
* @throws BeansException Bean异常
*/
@Override
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
ApplicationContextProvider.applicationContext = applicationContext;
}
/**
* 获取上下文对象
*
* @return application context
*/
private static ApplicationContext getApplicationContext() {
return applicationContext;
}
/**
* 通过名称获取Bean.
*
* @param name Bean名称
* @return Object对象
*/
public static Object getBean(String name) {
return getApplicationContext().getBean(name);
}
/**
* 通过类名获取Bean.
*
* @param clazz 类名
* @param <T> T
* @return T
*/
public static <T> T getBean(Class<T> clazz) {
return getApplicationContext().getBean(clazz);
}
/**
* 通过名称以及Clazz返回指定的Bean
*
* @param name name
* @param clazz class
* @param <T> class
* @return class
*/
public static <T> T getBean(String name, Class<T> clazz) {
return getApplicationContext().getBean(name, clazz);
}
}
具体使用时:
private DemoClass demoClass = ApplicationContextProvider.getBean(DemoClass.class);
这样即完成了Bean的注入。