Spring Boot使用@Async實(shí)現(xiàn)異步調(diào)用：使用Future以及定義超時(shí)

管理

編輯

刪除

CRMEB 2025-02-21 10:30:22

暫不處理

定義異步任務(wù)

首先，我們先使用@Async注解來定義一個(gè)異步任務(wù)，這個(gè)方法返回Future類型，具體如下：

@Slf4j
@Component
public class Task {

    public static Random random = new Random();

    @Async("taskExecutor")
    public Future<String> run() throws Exception {
        long sleep = random.nextInt(10000);
        log.info("開始任務(wù)，需耗時(shí)：" + sleep + "毫秒");
        Thread.sleep(sleep);
        log.info("完成任務(wù)");
        return new AsyncResult<>("test");
    }

}

Tips：什么是Future類型？

Future是對(duì)于具體的Runnable或者Callable任務(wù)的執(zhí)行結(jié)果進(jìn)行取消、查詢是否完成、獲取結(jié)果的接口。必要時(shí)可以通過get方法獲取執(zhí)行結(jié)果，該方法會(huì)阻塞直到任務(wù)返回結(jié)果。

它的接口定義如下：

public interface Future<V> {
    boolean cancel(boolean mayInterruptIfRunning);
    boolean isCancelled();
    boolean isDone();
    V get() throws InterruptedException, ExecutionException;
    V get(long timeout, TimeUnit unit)
        throws InterruptedException, ExecutionException, TimeoutException;
}

它聲明這樣的五個(gè)方法：

cancel方法用來取消任務(wù)，如果取消任務(wù)成功則返回true，如果取消任務(wù)失敗則返回false。參數(shù)mayInterruptIfRunning表示是否允許取消正在執(zhí)行卻沒有執(zhí)行完畢的任務(wù)，如果設(shè)置true，則表示可以取消正在執(zhí)行過程中的任務(wù)。如果任務(wù)已經(jīng)完成，則無論mayInterruptIfRunning為true還是false，此方法肯定返回false，即如果取消已經(jīng)完成的任務(wù)會(huì)返回false；如果任務(wù)正在執(zhí)行，若mayInterruptIfRunning設(shè)置為true，則返回true，若mayInterruptIfRunning設(shè)置為false，則返回false；如果任務(wù)還沒有執(zhí)行，則無論mayInterruptIfRunning為true還是false，肯定返回true。
isCancelled方法表示任務(wù)是否被取消成功，如果在任務(wù)正常完成前被取消成功，則返回 true。
isDone方法表示任務(wù)是否已經(jīng)完成，若任務(wù)完成，則返回true；
get()方法用來獲取執(zhí)行結(jié)果，這個(gè)方法會(huì)產(chǎn)生阻塞，會(huì)一直等到任務(wù)執(zhí)行完畢才返回；
get(long timeout, TimeUnit unit)用來獲取執(zhí)行結(jié)果，如果在指定時(shí)間內(nèi)，還沒獲取到結(jié)果，就直接返回null。

也就是說Future提供了三種功能：

判斷任務(wù)是否完成；
能夠中斷任務(wù)；
能夠獲取任務(wù)執(zhí)行結(jié)果。

#測(cè)試執(zhí)行與定義超時(shí)

在完成了返回Future的異步任務(wù)定義之后，我們來嘗試實(shí)現(xiàn)一個(gè)單元測(cè)試來使用這個(gè)Future完成任務(wù)的執(zhí)行，比如：

@Slf4j
@RunWith(SpringJUnit4ClassRunner.class)
@SpringBootTest
public class ApplicationTests {

    @Autowired
    private Task task;

    @Test
    public void test() throws Exception {
        Future<String> futureResult = task.run();
        String result = futureResult.get(5, TimeUnit.SECONDS);
        log.info(result);
    }

}

上面的代碼中，我們?cè)趃et方法中還定義了該線程執(zhí)行的超時(shí)時(shí)間，通過執(zhí)行這個(gè)測(cè)試我們可以觀察到執(zhí)行時(shí)間超過5秒的時(shí)候，這里會(huì)拋出超時(shí)異常，該執(zhí)行線程就能夠因執(zhí)行超時(shí)而釋放回線程池，不至于一直阻塞而占用資源。