ForkJoin有參無返回值、有參有返回值實例

介紹：

　　a . Fork/Join為JKD1.7引入，適用於對大量數據進行拆分成多個小任務進行計算的框架，最后把所有小任務的結果匯總合並得到最終的結果

　　b . 相關類

public abstract class RecursiveTask<V> extends ForkJoinTask<V>;
public abstract class RecursiveAction extends ForkJoinTask<Void>;

　　c . 其中RecursiveTask在執行有返回值的任務時使用，RecursiveAction在執行沒有返回值的任務時使用

實例代碼：

　　ForkJoin有參無返回值

　　參數：map

public class UpdatePlayersTotalTimeTask extends RecursiveAction {

    private static final int THRESHOLD_NUM = 30;//定義任務的切分閥值
    private Map<String,String> players;
    private PlayerTotalTimeService playerTotalTimeService;
    private MongoDao mongoDao;

    public UpdatePlayersTotalTimeTask(Map<String,String> players, MongoDao mongoDao) {
        this.players = players;
        this.mongoDao = mongoDao;
    }

    @Override
    protected void compute() {
        boolean canCompute = players.size() <= THRESHOLD_NUM;
        if (canCompute) {
            playerTotalTimeService = new PlayerTotalTimeService();
            playerTotalTimeService.updatePlayersTotalTime(players, mongoDao);
        } else {
　　　　　　　// 將任務一份為二
            int middle = players.size() / 2;

            int i = 0;
            Map<String, String> leftMap = new HashMap<>();
            Map<String, String> rightMap = new HashMap<>();

            for (Map.Entry<String, String> entry : players.entrySet()) {
                if (i < middle) {
                    leftMap.put(entry.getKey(), entry.getValue());
                } else {
                    rightMap.put(entry.getKey(), entry.getValue());
                }
                i++;
            }

            UpdatePlayersTotalTimeTask leftTask = new UpdatePlayersTotalTimeTask(leftMap, mongoDao);
            UpdatePlayersTotalTimeTask rightTask = new UpdatePlayersTotalTimeTask(rightMap, mongoDao);

            // 執行子任務
            leftTask.fork();
            rightTask.fork();
        }
    }

　　//調用測試
    public static void main(String[] args) throws InterruptedException {
        // 創建包含Runtime.getRuntime().availableProcessors()返回值作為個數的並行線程的ForkJoinPool
        ForkJoinPool forkjoinPool = new ForkJoinPool();

        Map<String, String> map = new HashMap<>();
　　　　 map.put("key1","value1");
        MongoDao mongoDao = new MongoDaoImpl();
        //生成一個計算任務
        UpdatePlayersTotalTimeTask task = new UpdatePlayersTotalTimeTask(map, mongoDao);
        // 提交可分解的PrintTask任務
        forkjoinPool.excute(task);
        forkjoinPool.awaitTermination(2, TimeUnit.SECONDS);//阻塞當前線程直到 ForkJoinPool 中所有的任務都執行結束
        // 關閉線程池
        forkjoinPool.shutdown();
    }

}

　　ForkJoin有參有返回值        (繼承RecursiveTask<T>類)

　　參數：set<String>   返回值：map<String,String>

public class CalcRoomPlayersTotalTimeTask extends RecursiveTask<Map<String,String>> {
    private static final int THRESHOLD_NUM = 15;
    private Set<String> roomSet;
    private PlayerTotalTimeService playerTotalTimeService;
    private MongoDao mongoDao;

    public CalcRoomPlayersTotalTimeTask(Set<String> roomSet, MongoDao mongoDao) {
        this.roomSet = roomSet;
        this.mongoDao = mongoDao;
    }

    @Override
    protected Map<String,String> compute() {

        playerTotalTimeService = new PlayerTotalTimeService();
        //如果任務足夠小就計算任務
        boolean canCompute = roomSet.size() <= THRESHOLD_NUM;
        if (canCompute) {

            Map<String,String> playersResult = new HashMap<>();

            playersResult =  playerTotalTimeService.calcRoomPlayersTotalTime(roomSet, mongoDao);

            return playersResult;
        } else {
            // 如果任務大於閾值，就分裂成兩個子任務計算
            long middle = roomSet.size() / 2;
            Set<String> leftSet = new HashSet<>();
            Set<String> rightSet = new HashSet<>();

            long i = 0;
            for (String room : roomSet) {
                if (i < middle) {
                    leftSet.add(room);
                } else {
                    rightSet.add(room);
                }
                i++;
            }

            CalcRoomPlayersTotalTimeTask leftTask = new CalcRoomPlayersTotalTimeTask(leftSet,mongoDao);
            CalcRoomPlayersTotalTimeTask rightTask = new CalcRoomPlayersTotalTimeTask(rightSet,mongoDao);

            // 執行子任務
            invokeAll(leftTask,rightTask);

            HashMap<String,String> result = new HashMap<>();

            Map<String,String> leftResult = leftTask.join();
            Map<String,String> rightResult = rightTask.join();

            result.putAll(leftResult);

            for (Map.Entry<String, String> entry : rightResult.entrySet()) {
                boolean contains = result.containsKey(entry.getKey());
                if(contains){
                    String playerTotalTimeStr =  entry.getValue();
                    Long playerTotalTimeLong = playerTotalTimeService.timeStringToLong(result.get(entry.getKey()))  + playerTotalTimeService.timeStringToLong(playerTotalTimeStr);
                    playerTotalTimeStr = playerTotalTimeService.timeLongToString(playerTotalTimeLong);
                    result.put(entry.getKey(),playerTotalTimeStr);
                }else {
                    result.put(entry.getKey(),entry.getValue());
                }
            }

            return result;
        }
    }

    public static void main(String[] args) throws InterruptedException {
        // 創建包含Runtime.getRuntime().availableProcessors()返回值作為個數的並行線程的ForkJoinPool
        ForkJoinPool forkjoinPool = new ForkJoinPool();

        Set<String> roomSet = new HashSet<>();
        Map<String,String> map = new HashMap<String,String>();
        MongoDao mongoDao = new MongoDaoImpl();
        //生成一個計算任務
        CalcRoomPlayersTotalTimeTask task = new CalcRoomPlayersTotalTimeTask(roomSet, mongoDao);

        // 提交可分解的PrintTask任務         
　　　　　　 map = forkjoinPool.invoke(task);
        forkjoinPool.awaitTermination(2, TimeUnit.SECONDS);//阻塞當前線程直到 ForkJoinPool 中所有的任務都執行結束
        // 關閉線程池
        forkjoinPool.shutdown();
    }
}    

 

 說明：

　　a .在有大量計算任務時，此框架方法可進行並行計算效率高，以上示例，可以根據具體的業務需求更改屬性及相關方法用於匹配自己的業務邏輯

       b .JDK1.8后由於加入Stream流的操作,集合框架可以使用Collection<E> default Stream<E> parallelStream()的方法轉換成並行流進行計算，此時效果與Fork/Join任務同效

       c .ForkJoinPool中的多種方法

public <T> ForkJoinTask<T> submit(ForkJoinTask<T> task);//等待獲取結果
public void execute(ForkJoinTask<?> task);//異步執行
public <T> T invoke(ForkJoinTask<T> task);//執行,獲取Future

      d .ForkJoinTask在執行的時候可能會拋出異常，但是沒辦法在主線程里直接捕獲異常，所以ForkJoinTask提供了isCompletedAbnormally()方法來檢查任務是否已經拋出異常或已經被取消了，並且可以通過ForkJoinTask的getException方法獲取異常。getException方             法返回Throwable對象，如果任務被取消了則返回CancellationException。如果任務沒有完成或者沒有拋出異常則返回null。

if(task.isCompletedAbnormally()) {
    System.out.println(task.getException());
}

 

注：部分內容引自https://segmentfault.com/a/1190000010209196