Springboot?@Async多線程獲取返回值方式

更新時(shí)間：2023年09月27日 09:21:17 作者：小白Alan

這篇文章主要介紹了Springboot?@Async多線程獲取返回值方式,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

@Async 多線程獲取返回值

最近需要用到多線程自己維護(hù)線程池很麻煩正好看到 springboot 集成線程池的例子這里自己做了個(gè)嘗試和總結(jié) 記錄一下也分享給需要的朋友；

不考慮事務(wù)的情況下這個(gè)多線程實(shí)現(xiàn)比較簡(jiǎn)單

主要有以下幾點(diǎn)

在啟動(dòng)類加上 @EnableAsync 注解，開啟異步執(zhí)行支持；
編寫線程池配置類，別忘了 @Configuration ，和 @Bean 注解；
編寫需要異步執(zhí)行的業(yè)務(wù)，放到單獨(dú)的類中 (可以定義為 service，因?yàn)樾枰?spring 管理起來才能用 )；
在業(yè)務(wù)service中調(diào)用異步執(zhí)行的service，注意這是重點(diǎn)，不能直接在業(yè)務(wù) service 中寫異步執(zhí)行的代碼，否則無法異步執(zhí)行( 這就是單獨(dú)放異步代碼的原因)；

好了, 上代碼：

// 啟動(dòng)類
@EnableAsync
@EnableWebSecurity
@ServletComponentScan
@SpringBootApplication(scanBasePackages={"com.example.demo"})
public class DemoApplication {
	public static void main(String[] args) {
		SpringApplication.run(DemoApplication.class, args);
	}
}

// 線程池配置類
@Slf4j
@Configuration
public class BootThreadpoolConfig {
    // 配置核心線程數(shù)
    private int corePoolSize = 5;
    // 配置最大線程數(shù)
    private int maxPoolSize = 20;
    // 配置任務(wù)隊(duì)列的長(zhǎng)度
    private int queueCapacity = 500;
    // 配置任務(wù)的空閑時(shí)間
    private int aliveSeconds = 600;
    // 配置線程前綴
    private String namePrefix = "localThreadPool";
    // 自定義線程池, 起個(gè)好記的名
    @Bean(name = "localBootAsyncExecutor")
    public Executor asyncServiceExecutor() {
        log.info("初始化 springboot 線程池");
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        //配置核心線程數(shù)
        executor.setCorePoolSize(corePoolSize);
        //配置最大線程數(shù)
        executor.setMaxPoolSize(maxPoolSize);
        //配置隊(duì)列大小
        executor.setQueueCapacity(queueCapacity);
        //配置線程池中的線程的名稱前綴
        executor.setThreadNamePrefix(namePrefix);
        //配置線程的空閑時(shí)間
        executor.setKeepAliveSeconds(aliveSeconds);
        // RejectedExecutionHandler：當(dāng)pool已經(jīng)達(dá)到max size的時(shí)候，如何處理新任務(wù)
        // CallerRunsPolicy：不在新線程中執(zhí)行任務(wù)，而是有調(diào)用者所在的線程來執(zhí)行
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
        //執(zhí)行初始化
        executor.initialize();
        log.info("springboot 線程池初始化完畢");
        return executor;
    }
}

// 異步執(zhí)行代碼
@Service("asyncExecutorTest")
public class AsyncExecutorTest {
    // 異步執(zhí)行的方法, 注解內(nèi)為自定義線程池類名
    @Async("localBootAsyncExecutor")
    public Future<Integer> test1(Integer i) throws InterruptedException {
        Thread.sleep(100);
        System.out.println("@Async 執(zhí)行: " + i);
        return new AsyncResult(i);
    }
    // 這里使用其它方式調(diào)用，詳見后面的 service3 方法
    public Integer test2(Integer i) throws InterruptedException {
        Thread.sleep(100);
        System.out.println(" excute.run 執(zhí)行: " + i);
        return i;
    }
}

// 業(yè)務(wù) service
@Service("asyncExcutorService")
public class AsyncExcutorService {
    @Autowired
    AsyncExecutorTest asyncExecutorTest;
    @Autowired
    Executor localBootAsyncExecutor;
    // 測(cè)試 無返回值異步執(zhí)行
    public void service1(){
        System.out.println("service1 執(zhí)行----->");
        for (int i = 0; i < 50; i++) {
            try {
                asyncExecutorTest.test1(i);
            } catch (InterruptedException e) {
                System.out.println("service1執(zhí)行出錯(cuò)");
            }
        }
        System.out.println("service1 結(jié)束----->");
    }
    // 測(cè)試 有返回值異步執(zhí)行
    public void service2(){
        long l = System.currentTimeMillis();
        System.out.println("service2 執(zhí)行----->");
        List<Future> result = new ArrayList<>();
        try {
            for (int i = 0; i < 300; i++) {
                Future<Integer> integerFuture = asyncExecutorTest.test1(i);
                result.add(integerFuture);
            }
            for (Future future : result) {
                System.out.println(future.get());
            }
        } catch (InterruptedException | ExecutionException e) {
            System.out.println("service2執(zhí)行出錯(cuò)");
        }
        System.out.println("service2 結(jié)束----->" + (System.currentTimeMillis() - l));
    }
    // 測(cè)試 有返回值異步執(zhí)行
    public void service3(){
        long l = System.currentTimeMillis();
        List<Integer> result = new ArrayList<>();
        try {
            System.out.println("service3 執(zhí)行----->");
            int total = 300;
            CountDownLatch latch = new CountDownLatch(total);
            for (int i = 0; i < total; i++) {
                final int y = i;
                localBootAsyncExecutor.execute(() -> {
                    try {
                        result.add(asyncExecutorTest.test2(y));
                    } catch (InterruptedException e) {
                        System.out.println("service3執(zhí)行出錯(cuò)");
                    } finally {
                        latch.countDown();
                    }
                });
            }
            latch.await();
        } catch (InterruptedException e) {
            System.out.println("service3執(zhí)行出錯(cuò)");
        }
        System.out.println("service3 結(jié)束----->" + (System.currentTimeMillis() - l));
    }
}