SpringBoot[第二课] spring环境下利用线程池写一个多线程任务执行器

前提

在公司的项目里看到一个多线程包，封装了一些原生多线程的代码，方便萌新使用多线程提高代码执行效率。使用方法就是把一个需要处理大量数据的任务等量拆分为多个子任务，然后像使用集合一样放到一个批次对象里（例如Batch），然后用Batch遍历启动批次中的子任务，并发执行。全部执行完毕之后返回执行结果。对新手来说确实方便，由于项目组中使用spring作为开发框架相当普遍，而spring从3.0以后就引入了对多线程的支持，再配合配置类代替xml文件的特性，可以考虑用spring的线程池，实现与上述一样的功能，实用性有待验证，暂且作为使用spring线程池的小练习。

步骤

要使用spring线程池实现并发，至少需要三块东西：开启并发支持、注册执行器、定义并发任务。

线程池配置类

package com.terry.config;
import java.util.concurrent.Executor;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
@Configuration
@EnableAsync
public class AsyncContextConfig {
	 @Bean
     public Executor getExecutor() {
          ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
          executor.setCorePoolSize(5);
          executor.setMaxPoolSize(10);
          executor.setQueueCapacity(25);
          executor.initialize();
          return executor;
	 }
		@Bean
		@Scope("prototype")
		public Task task() {
			return new Task();
		}
}

• @EnableAsync：开启spring对多线程的支持，可以放在启动类或者配置类上；
• Task：Task就是定义并发任务的类，必须交给spring容器去初始化才能调起线程池。

1. ThreadPoolTaskExecutor：getExecutor()方法将ThreadPoolTaskExecutor注册到spring容器中，其中就定义了一个可用的线程池，这是一种注册方法，还有另一种注册执行器的方法，看上去更符合spring的风格：