【七日打卡】Spring中如何方便的使用Async异步方法

1,066 阅读4分钟

在Spring提供了@Async来实现方法的异步调用。 即当调用标有@Async标识的方法时,调用线程不会等待被调用方法执行完后再继续执行后续操作,而对被调用的方法启动一个独立线程来执行。

这种异步执行的方式通常用于处理接口中不需要返回给用户的数据处理。比如:

当注册的时候,只需要将用户信息返回用户,而关于信息的保存操作可以使用异步执行。

在异步方法中都是开发自己去定义具体对业务逻辑代码块,然后交给Spring容器管理并启动新线程执行相应的方法。在使用异步方法的时候需要特别注意的是线程池的配置以及任务中异常的处理。下面对这个功能进行简单介绍。

简介

功能开启注解:@EnableAsync通过在Spring的配置类中添加这个注解来开启Spring的异步方法功能。

异步方法标识注解Async,其定义为:

// 用于type和method
@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface Async {
    String value() default "";
}

当此注解用于类的时候,表示此类中的所有方法都为异步方法。此注解中的value属性可用于指定执行此异步方法的线程池。线程池的具体确定方法下面具体分析。

Spring线程池的选择和自定义配置线程池

在项目中我们通常不会自己手动创建线程,而是通过统一的线程池来执行task或者异步方法,使用这种方式来避免多人团队中由于自定义线程导致的资源耗尽的问题。在自定义线程池之前首先要了解Spring在执行异步任务或者方法的时候是怎么选择线程池的。

@Async对于线程池的选择顺序

Spring在执行async标识的异步方法的时候首先会在Spring的上下文中搜索类型为TaskExecutor或者名称为“taskExecutor”的bean,当可以找到的时候,就将任务提交到此线程池中执行。当不存在以上线程池的时候,Spring会手动创建一个SimpleAsyncTaskExecutor执行异步任务。

另外当@Async注解的时候指定了value值,Spring会使用指定的线程池执行。比如以下:

@Async(value = "asyncTaskThreadPool") 这个时候Spring会去上下文中找名字为asyncTaskThreadPool的bean,并执行异步任务,如果找不到会抛出异常。

自定义线程池和异常处理

在了解了@Async对于线程池的选择后,我们需要自定义线程池。自定义Async线程池有三种方式。

方法一:重写@Async获取线程池的方法。

配置@Async方法的线程池需要继承AsyncConfigurerSupport类,或者实现AsyncConfigurer接口,并重写getAsyncExecutor方法,代码如下:

@Configuration
@EnableAsync
public class ThreadPoolBeanFactory extends AsyncConfigurerSupport{
    @Override
    public Executor getAsyncExecutor() {
        ThreadPoolTaskExecutor asyncTaskThreadPool = new ThreadPoolTaskExecutor();
        asyncTaskThreadPool.setCorePoolSize(100);
        asyncTaskThreadPool.setMaxPoolSize(200);
        asyncTaskThreadPool.setQueueCapacity(11);
        asyncTaskThreadPool.setThreadFactory(new ThreadFactory() {
            private final AtomicLong index = new AtomicLong(1);
            @Override
            public Thread newThread(Runnable r) {
                return new Thread(r, "Async-override-task-pool-thread-" + index.getAndIncrement());
            }
        });
        asyncTaskThreadPool.initialize();
        return asyncTaskThreadPool;
    }

    @Override
    public AsyncUncaughtExceptionHandler getAsyncUncaughtExceptionHandler() {
        //返回值为void的异步方法不会传递异常到此,当方法中出现异常的时候只会打印日志,重写此方法来自定义异常处理方法
        return null;
    }
}

这种定义的方法缺点是没有定义明确的线程池bean对象。

方法二:自定义相应类型的线程池bean。

第二种方法是基于Spring @Async对线程池选择的原理来实现的,定义一个类型为TaskExecutor的bean,定义方式如下:

@Bean
public TaskExecutor asyncTaskThreadPool() {
    ThreadPoolTaskExecutor asyncTaskThreadPool = new ThreadPoolTaskExecutor();
    asyncTaskThreadPool.setCorePoolSize(100);
    asyncTaskThreadPool.setMaxPoolSize(200);
    
    asyncTaskThreadPool.setThreadFactory(new ThreadFactory() {
        private final AtomicLong index = new AtomicLong(1);
        @Override
        public Thread newThread(Runnable r) {
            return new Thread(r, "Async-task-pool-thread-" + index.getAndIncrement());
        }
    });
  // asyncTaskThreadPool.initialize();//当使用@Bean的时候不需要调用此方法,装载容器的时候会自动调用
    return asyncTaskThreadPool;
}

测试

分别使用以上两种方式定义的线程池。代码如下:

@Async
public void test(){
    System.out.println(Thread.currentThread().getName());

}

此时Spring会自动使用以上定义的线程池执行此方法。使用以上两种配置输出结果分别是:

Async-task-override-pool-thread-1
Async-task-pool-thread-1

方法三 在Async注解中执行线程池名称

异步任务定义如下:

@Async(value = "asyncTaskThreadPool")
public void asyncTask2() {
    LOGGER.info("AsyncTask2 start.");
    LOGGER.info(Thread.currentThread().getName());
    LOGGER.info("AsyncTask2 finished.");
}

此时Spring会在上下文中找名称为asyncTaskThreadPool的线程池来执行此任务。

@Async返回操作结果

异步任务可以通过定义返回类型为Future来实现返回值的接收,定义如下:

@Async
public Future<String> asyncTaskWithResult() {
    LOGGER.info("AsyncTaskWithResult start.");
    try {
        Thread.sleep(1000 * 10);
    } catch (Exception e) {
        return new AsyncResult<>("error" + e.getMessage());
    }
    LOGGER.info("AsyncTaskWithResult finished.");

    return new AsyncResult<>("success");
}

单元测试代码如下:

@RunWith(SpringRunner.class)
@SpringBootTest
public class AsyncApplicationTests {
    @Autowired
    private AsyncTaskService asyncTaskService;

    @Test
    public void asyncTest() throws Exception{
        Future<String> future = asyncTaskService.asyncTaskWithResult();

        while (!future.isDone()) {
            System.out.println("Wait asyncTaskWithResult.");
            Thread.sleep(1000);
        }
        System.out.println("asyncTaskWithResult result is:" + future.get());
        System.out.println("asyncTask finished.");

    }
}

输出内容如下:

AsyncTaskWithResult start.
Wait asyncTaskWithResult.
Wait asyncTaskWithResult.
Wait asyncTaskWithResult.
Wait asyncTaskWithResult.
Wait asyncTaskWithResult.
Wait asyncTaskWithResult.
Wait asyncTaskWithResult.
AsyncTaskWithResult finished.
asyncTaskWithResult result is:success
asyncTask finished.

点关注,不迷路

文章每周持续更新,可以微信搜索「 十分钟学编程 」第一时间阅读和催更,如果这个文章写得还不错,觉得有点东西的话
~求点赞👍 求关注❤️ 求分享❤️
各位的支持和认可,就是我创作的最大动力,我们下篇文章见!