快捷導(dǎo)航

詳解SpringBoot中異步請求的實現(xiàn)與并行執(zhí)行

更新時間：2024年02月06日 08:07:33 作者：一只愛擼貓的程序猿

這篇文章主要為大家詳細(xì)介紹了在SpringBoot中如何是實現(xiàn)異步請求、并行執(zhí)行,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下

開啟異步支持

首先，需要在Spring Boot應(yīng)用中開啟異步支持。這可以通過在一個配置類上添加@EnableAsync注解來實現(xiàn)。AsyncConfig 類開啟異步支持并自定義Executor

import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.AsyncConfigurer;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;

import java.util.concurrent.Executor;

@Configuration
@EnableAsync
public class AsyncConfig implements AsyncConfigurer {

    @Override
    public Executor getAsyncExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(2);
        executor.setMaxPoolSize(5);
        executor.setQueueCapacity(100);
        executor.setThreadNamePrefix("Async-");
        executor.initialize();
        return executor;
    }
}

定義異步服務(wù)

然后，定義一個服務(wù)類，并在其中的方法上使用@Async注解來標(biāo)記它們?yōu)楫惒椒椒?。這里可以返回Future、CompletableFuture或ListenableFuture等，以便處理異步結(jié)果。

import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.stream.Collectors;

@Service
public class AsyncService {

    @Async
    public CompletableFuture<String> processAsync() {
        // 模擬異步任務(wù)
        try {
            Thread.sleep(1000);
            return CompletableFuture.completedFuture("單個任務(wù)執(zhí)行完成");
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
            return CompletableFuture.failedFuture(e);
        }
    }

    @Async
    public CompletableFuture<String> processAsyncParallel(int taskNumber) {
        // 模擬異步任務(wù)
        try {
            Thread.sleep(1000); // 假設(shè)這是耗時操作
            return CompletableFuture.completedFuture("任務(wù) #" + taskNumber + " 執(zhí)行完成");
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
            return CompletableFuture.failedFuture(e);
        }
    }

    public List<CompletableFuture<String>> executeTasksInParallel() {
        List<CompletableFuture<String>> futures = new ArrayList<>();
        for (int i = 0; i < 5; i++) {
            futures.add(processAsyncParallel(i));
        }
        return futures;
    }
}

在Controller中調(diào)用并行執(zhí)行方法并等待所有任務(wù)完成

在AsyncController中，我們將添加一個映射來啟動并行任務(wù)并等待它們?nèi)客瓿伞?/p>

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.stream.Collectors;

@RestController
public class AsyncController {

    @Autowired
    private AsyncService asyncService;

    @GetMapping("/startParallelTasks")
    public String startParallelTasks() {
        List<CompletableFuture<String>> futures = asyncService.executeTasksInParallel();

        // 等待所有任務(wù)完成
        CompletableFuture<Void> allDoneFuture = CompletableFuture.allOf(futures.toArray(new CompletableFuture[0]))
                .thenApply(v -> futures.stream()
                        .map(CompletableFuture::join)
                        .collect(Collectors.toList()));

        allDoneFuture.thenAccept(results -> results.forEach(System.out::println));

        return "并行任務(wù)開始執(zhí)行";
    }

    @GetMapping("/waitForSpecificTask")
    public String waitForSpecificTask() throws Exception {
        CompletableFuture<String> future = asyncService.processAsync();

        // 等待特定任務(wù)完成
        String result = future.get(); // 阻塞直到任務(wù)完成

        return "特定任務(wù)完成，結(jié)果：" + result;
    }
}

說明

executeTasksInParallel方法在AsyncService中啟動了多個并行的異步任務(wù)。這些任務(wù)通過CompletableFuture并行執(zhí)行。
在startParallelTasks中，我們使用CompletableFuture.allOf來等待所有任務(wù)完成。allOf返回一個新的CompletableFuture，該CompletableFuture在所有給定的CompletableFutures完成時完成。
waitForSpecificTask方法演示了如何等待一個特定的異步任務(wù)完成。使用CompletableFuture.get()方法可以阻塞當(dāng)前線程直到異步任務(wù)完成，并獲取結(jié)果。