springboot 多線程執行

一.springboot開線程執行異步任務

1.Spring通過任務執行器TaskExecutor，來實現多線程和並發編程，使用ThreadPoolTaskExecutor可實現一個基於線程池的TaskExecutor;

2.異步需要在配置類上面加 @EnableAsync 來開啟對異步任務的支持在需要異步執行的方法上面加@Async 來聲明這個方法是一個需要異步執行的方法;

3. 讓配置類實現AsyncConfigurer接口，並重寫getAsyncExecutor方法，並返回一個ThreasPoolTaskExecutor，就可以獲取一個基於線程池的TaskExecutor;

4. @Async 用在方法上，表示這個方法是一個異步的方法，如果用在類上面，表明這個類中的所有方法都是異步的方法。

(1).配置類

package com.yunzhangfang.springboot1.config;

import org.springframework.aop.interceptor.AsyncUncaughtExceptionHandler;
import org.springframework.context.annotation.ComponentScan;
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
@ComponentScan("com.yunzhangfang.springboot1.service")
@EnableAsync
public class ThreadConfig implements AsyncConfigurer {
    
    // ThredPoolTaskExcutor的處理流程
    // 當池子大小小於corePoolSize，就新建線程，並處理請求
    // 當池子大小等於corePoolSize，把請求放入workQueue中，池子里的空閑線程就去workQueue中取任務並處理
    // 當workQueue放不下任務時，就新建線程入池，並處理請求，如果池子大小撐到了maximumPoolSize，就用RejectedExecutionHandler來做拒絕處理
    // 當池子的線程數大於corePoolSize時，多余的線程會等待keepAliveTime長時間，如果無請求可處理就自行銷毀
    @Override
    public Executor getAsyncExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(5);
        executor.setMaxPoolSize(15);
        executor.setQueueCapacity(25);
        executor.initialize();
        return executor;
    }

    @Override
    public AsyncUncaughtExceptionHandler getAsyncUncaughtExceptionHandler() {
        return null;
    }

}

(2).異步任務

package com.yunzhangfang.springboot1.service;

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

import java.util.concurrent.Future;

@Service
public class TaskService {

    @Async
    /**
     * 表明是異步調用
     * 沒有返回值
     */
    public void excutVoidTask(int i) {
        System.out.println("異步執行任務第[" + i + "] 個");
    }

    /**
     * 有返回值
     * 異常調用
     *
     * @param i
     * @return
     * @throws InterruptedException
     */
    @Async
    public Future<String> excuteValueTask(int i) throws InterruptedException {
        Thread.sleep(1000);
        Future<String> future = new AsyncResult<String>("success is " + i);
        System.out.println("異步執行任務第[" + i + "] 個");
        return future;
    }

}

(3).測試異步任務

package com.yunzhangfang.springboot1;

import com.yunzhangfang.springboot1.service.TaskService;
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.core.task.TaskRejectedException;
import org.springframework.test.context.junit4.SpringRunner;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Future;

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

    @Autowired
    private TaskService service;

    @Test
    public void contextLoads() {
    }

    /**
     * 沒有返回值測試
     */
    @Test
    public void testVoid() {
        for (int i = 0; i < 20; i++) {
            service.excutVoidTask(i);
        }
        System.out.println("========主線程執行完畢=========");
    }

    @Test
    public void testReturn() throws InterruptedException, ExecutionException {
        List<Future<String>> lstFuture = new ArrayList<>();// 存放所有的線程，用於獲取結果
        for (int i = 0; i < 100; i++) {
            while (true) {
                try {
                    // 線程池超過最大線程數時，會拋出TaskRejectedException，則等待1s，直到不拋出異常為止
                    Future<String> stringFuture = service.excuteValueTask(i);
                    lstFuture.add(stringFuture);
                    break;
                } catch (TaskRejectedException e) {
                    System.out.println("線程池滿，等待1S。");
                    Thread.sleep(1000);
                }
            }
        }

        // 獲取值.get是阻塞式，等待當前線程完成才返回值
        for (Future<String> future : lstFuture) {
            System.out.println(future.get());
        }

        System.out.println("========主線程執行完畢=========");
    }

}

---

有錯誤,希望指出,共同進步,天天向上