詳解Java中的ForkJoin

更新時間：2023年05月16日 10:04:47 作者：越走越遠的風

Fork/Join框架是Java 7提供的一種用于并行執(zhí)行任務的框架，它將大任務分解為若干個小任務,并行執(zhí)行這些小任務，最終通過合并每個小任務的結果得到大任務的結果，文中有詳細的代碼示例,需要的朋友可以參考下

ForkJoin簡介

Fork/Join框架是Java 7提供的一種用于并行執(zhí)行任務的框架，它將大任務分解為若干個小任務,并行執(zhí)行這些小任務，最終通過合并每個小任務的結果得到大任務的結果。

Fork/Join采用的是分而治之的基本思想，分而治之就是將一個復雜的任務，按照規(guī)定的閾值劃分成多個簡單的小任務，然后將這些小任務的結果再進行匯總返回，得到最終的任務。

并行和并發(fā)的區(qū)別

并行和并發(fā)是計算機科學中的兩個概念，它們之間有一些相似之處，但也有明顯的區(qū)別。

并行是指多個處理器或者是多核的處理器同時處理多個不同的任務。并行可以在多處理器系統(tǒng)中實現(xiàn)，利用每個處理機來處理一個可并發(fā)執(zhí)行的程序，從而實現(xiàn)多個程序的同時執(zhí)行。在并行執(zhí)行時，每個處理器可以同時執(zhí)行多個程序，從而提高計算效率。

并發(fā)是指邏輯上的同時發(fā)生（即 true 的同時性），而并行是物理上的同時發(fā)生。在多道程序環(huán)境下，并發(fā)性是指在一段時間內宏觀上有多個程序在同時運行，但在單處理機系統(tǒng)中，每一時刻卻僅能有一道程序執(zhí)行，故微觀上這些程序只能是分時地交替執(zhí)行。

簡而言之，并行是指多個處理器或多核處理器同時處理多個任務，而并發(fā)是指在同一時間內多個任務同時發(fā)生。

工作竊取算法

工作竊取算法是指某個線程從其他隊列里竊取任務來執(zhí)行。當工作隊列中有空閑任務時，就將任務從原線程的隊列中竊取過來，執(zhí)行完成后再將結果返回給原線程。這樣就保證了原線程不會一直等待空閑任務，從而提高了程序的效率。

Fork/Join框架使用ForkJoinPool這個特殊的線程池來處理任務之間有依賴的情況，其實現(xiàn)了“work-stealing”算法（工作量竊取算法）并執(zhí)行ForkJoinTask對象。ForkJoinPool保持多個線程，其線程數(shù)量默認為機器cpu核心數(shù)。每個線程都有一個特殊類型的deques隊列（雙端隊列），放置該線程的所有任務，而不是所有線程共享一個公共隊列。

每個線程都會保證將自己隊列中的任務執(zhí)行完，當自己的任務執(zhí)行完成之后，在去看其他線程的任務隊列中是否有未處理完的任務，如果有則會幫助其他線程執(zhí)行。

這時雙端隊列的優(yōu)勢就體現(xiàn)出來了,被竊取的任務只會從隊列的頭部獲取任務，而正常處理的線程每次都是從隊列的尾部獲取任務。

求1到1億的和

package com.fandf.test.forkjoin;  
  
import lombok.extern.slf4j.Slf4j;  
import org.springframework.util.StopWatch;  
  
import java.util.concurrent.ForkJoinPool;  
import java.util.concurrent.Future;  
import java.util.concurrent.RecursiveTask;  
  
/**  
* @author fandongfeng  
*/  
@Slf4j  
public class ForkJoinDemo extends RecursiveTask<Long> {  
  
    /**  
    * 小任務的大小閾值  
    */  
    public static final int TASK_SIZE = 100000;  
    /**  
    * 開始數(shù)字  
    */  
    private final Long start;  
    /**  
    * 結束數(shù)字  
    */  
    private final Long end;  

    public ForkJoinDemo(Long start, Long end) {  
        this.start = start;  
        this.end = end;  
    }  

    @Override  
    protected Long compute() {  
        long sum = 0L;  
        //如果任務足夠小就計算任務  
        boolean canCompute = (end - start) <= TASK_SIZE;  
        if (canCompute) {  
            for (Long i = start; i <= end; i++) {  
                sum += i;  
            }  
        } else {  
            // 如果任務大于閾值，就分裂成兩個子任務計算  
            long middle = (start + end) / 2;  
            ForkJoinDemo leftTask = new ForkJoinDemo(start, middle);  
            ForkJoinDemo rightTask = new ForkJoinDemo(middle + 1, end);  

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

            // 等待任務執(zhí)行結束合并其結果  
            Long leftResult = leftTask.join();  
            Long rightResult = rightTask.join();  

            // 合并子任務  
            sum = leftResult + rightResult;  
        }  
        return sum;  
    }  

    public static void main(String[] args) {  
        ForkJoinPool forkjoinPool = new ForkJoinPool();  
        //生成一個計算任務，計算1+2+3+4+...+100000000  
        ForkJoinDemo task = new ForkJoinDemo(1L, 100000000L);  
        StopWatch stopWatch = new StopWatch();  
        stopWatch.start();  
        //執(zhí)行一個任務  
        Future<Long> result = forkjoinPool.submit(task);  

        try {  
            System.out.println("result:" + result.get());  
        } catch (Exception e) {  
            log.error("exception", e);  
        }  
        stopWatch.stop();  
        System.out.println("總耗時：" + stopWatch.getTotalTimeMillis() + "毫秒");  
        System.out.println("getParallelism:" + forkjoinPool.getParallelism());  
        System.out.println("getPoolSize:" + forkjoinPool.getPoolSize());  
    }  
}

輸出結果

result:5000000050000000
總耗時：330毫秒
getParallelism:6
getPoolSize:7

ForkJoin框架實現(xiàn)

ForkJoinPool

ForkJoinPool是用于運行ForkJoinTasks的線程池，實現(xiàn)了Executor接口

public ForkJoinPool() {  
    this(Math.min(MAX_CAP, Runtime.getRuntime().availableProcessors()),  
        defaultForkJoinWorkerThreadFactory, null, false);  
}

public ForkJoinPool(int parallelism,  
                    ForkJoinWorkerThreadFactory factory,  
                    UncaughtExceptionHandler handler,  
                    boolean asyncMode) {  
    this(checkParallelism(parallelism),  
        checkFactory(factory),  
        handler,  
        asyncMode ? FIFO_QUEUE : LIFO_QUEUE,  
        "ForkJoinPool-" + nextPoolId() + "-worker-");  
    checkPermission();  
}

ForkJoinPool構造方法有四個參數(shù)：