詳解Spring Boot 異步執(zhí)行方法

發(fā)布時(shí)間：2020-09-18 02:47:08 來(lái)源：腳本之家閱讀：166 作者：Simeone_xu 欄目：編程語(yǔ)言

最近遇到一個(gè)需求，就是當(dāng)服務(wù)器接到請(qǐng)求并不需要任務(wù)執(zhí)行完成才返回結(jié)果，可以立即返回結(jié)果，讓任務(wù)異步的去執(zhí)行。開(kāi)始考慮是直接啟一個(gè)新的線程去執(zhí)行任務(wù)或者把任務(wù)提交到一個(gè)線程池去執(zhí)行，這兩種方法都是可以的。但是 Spring 這么強(qiáng)大，肯定有什么更簡(jiǎn)單的方法，就 google 了一下，還真有呢。就是使用 @EnableAsync 和 @Async 這兩個(gè)注解就 ok 了。

給方法加上 @Async 注解

package me.deweixu.aysncdemo.service;

public interface AsyncService {

 void asyncMethod(String arg);
}

package me.deweixu.aysncdemo.service.ipml;

import me.deweixu.aysncdemo.service.AsyncService;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;

@Service
public class AsyncServiceImpl implements AsyncService {

 @Async
 @Override
 public void asyncMethod(String arg) {
  System.out.println("arg:" + arg);
  System.out.println("=====" + Thread.currentThread().getName() + "=========");
 }
}

@EnableAsync

在啟動(dòng)類(lèi)或者配置類(lèi)加上 @EnableAsync 注解

package me.deweixu.aysncdemo;

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.scheduling.annotation.EnableAsync;

@EnableAsync
@SpringBootApplication
public class AysncDemoApplication {

 public static void main(String[] args) {
  SpringApplication.run(AysncDemoApplication.class, args);
 }
}

測(cè)試

package me.deweixu.aysncdemo;

import me.deweixu.aysncdemo.service.AsyncService;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;

@RunWith(SpringRunner.class)
@SpringBootTest
public class AysncDemoApplicationTests {

 @Autowired
 AsyncService asyncService;

 @Test
 public void testAsync() {
  System.out.println("=====" + Thread.currentThread().getName() + "=========");
  asyncService.asyncMethod("Async");
 }

}

=====main=========
2018-03-25 21:30:31.391 INFO 28742 --- [ main] .s.a.AnnotationAsyncExecutionInterceptor : No task executor bean found for async processing: no bean of type TaskExecutor and no bean named 'taskExecutor' either
arg:Async
=====SimpleAsyncTaskExecutor-1=========

從上面的結(jié)果看 asyncService.asyncMethod("Async") 確實(shí)異步執(zhí)行了，它使用了一個(gè)新的線程。

指定 Executor

從上面執(zhí)行的日志可以猜測(cè)到 Spring 默認(rèn)使用 SimpleAsyncTaskExecutor 來(lái)異步執(zhí)行任務(wù)的，可以搜索到這個(gè)類(lèi)。@Async 也可以指定自定義的 Executor。

在啟動(dòng)類(lèi)中增加自定義的 Executor

package me.deweixu.aysncdemo;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.annotation.Bean;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import java.util.concurrent.Executor;
@EnableAsync
@SpringBootApplication
public class AysncDemoApplication {

 public static void main(String[] args) {
  SpringApplication.run(AysncDemoApplication.class, args);
 }

 @Bean(name = "threadPoolTaskExecutor")
 public Executor threadPoolTaskExecutor() {
  return new ThreadPoolTaskExecutor();
 }
}

指定 Executor

package me.deweixu.aysncdemo.service.ipml;
import me.deweixu.aysncdemo.service.AsyncService;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;

@Service
public class AsyncServiceImpl implements AsyncService {

 @Async("threadPoolTaskExecutor")
 @Override
 public void asyncMethod(String arg) {
  System.out.println("arg:" + arg);
  System.out.println("=====" + Thread.currentThread().getName() + "=========");
 }
}

這樣在異步執(zhí)行任務(wù)的時(shí)候就使用 threadPoolTaskExecutor

設(shè)置默認(rèn)的 Executor

上面提到如果 @Async 不指定 Executor 就默認(rèn)使用 SimpleAsyncTaskExecutor，其實(shí)默認(rèn)的 Executor 是可以使用 AsyncConfigurer 接口來(lái)配置的

@Configuration
public class SpringAsyncConfig implements AsyncConfigurer {  
 @Override
 public Executor getAsyncExecutor() {
  return new ThreadPoolTaskExecutor();
 }  
}

異常捕獲

在異步執(zhí)行的方法中是可能出現(xiàn)異常的，我們可以在任務(wù)內(nèi)部使用 try catch 來(lái)處理異常，當(dāng)任務(wù)拋出異常時(shí)，Spring 也提供了捕獲它的方法。

實(shí)現(xiàn) AsyncUncaughtExceptionHandler 接口

public class CustomAsyncExceptionHandler
 implements AsyncUncaughtExceptionHandler {
 
 @Override
 public void handleUncaughtException(
  Throwable throwable, Method method, Object... obj) { 
  System.out.println("Exception message - " + throwable.getMessage());
  System.out.println("Method name - " + method.getName());
  for (Object param : obj) {
   System.out.println("Parameter value - " + param);
  }
 }  
}

實(shí)現(xiàn) AsyncConfigurer 接口重寫(xiě) getAsyncUncaughtExceptionHandler 方法

@Configuration
public class SpringAsyncConfig implements AsyncConfigurer {
  
 @Override
 public Executor getAsyncExecutor() {
  return new ThreadPoolTaskExecutor();
 }
 
 @Override
 public AsyncUncaughtExceptionHandler getAsyncUncaughtExceptionHandler() {
  return new CustomAsyncExceptionHandler();
 }

  
}

改寫(xiě) asyncMethod 方法使它拋出異常

 @Async
 @Override
 public void asyncMethod(String arg) {
  System.out.println("arg:" + arg);
  System.out.println("=====" + Thread.currentThread().getName() + "=========");
  throw new NullPointerException();
 }

運(yùn)行結(jié)果：

=====main=========
arg:Async
=====threadPoolTaskExecutor-1=========
Exception message - Async NullPointerException
Method name - asyncMethod
Parameter value - Async

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持億速云。

向AI問(wèn)一下細(xì)節(jié)

詳解Spring Boot 異步執(zhí)行方法

猜你喜歡

最新資訊

相關(guān)推薦

相關(guān)標(biāo)簽