Fork/Join框架

ForkJoin框架是Java7的新特性，用于并行处理大型任务的。其核心思想是将一个大任务拆分成若干小任务，小任务继续拆分，直到不能再拆。开发人员可自行设置阈值用以控制任务拆到什么程度后不能再拆。拆分完毕后，将创建若干线程，每个线程都有一个任务队列，之前拆分的任务会被放到这些队列中。此处的队列的是双端队列。为提高资源利用率，先处理完任务的线程不会闲着，它随机找一个线程的任务队列偷任务来处理，从队尾偷，而任务被偷的那个线程执行任务时是从对头拿任务。这称为工作密取（work strealing）机制。Fork/Join框架可显著提高并行任务执行效率。

使用ForkJoin框架需要继承RecursiveTask类或RecursiveAction类，然后重写compute方法，前者有返回值，而后者没有。

下面的测试例子用于计算从一累加到一亿的和。

package com.umbrella.threadpool;

import java.io.IOException;
import java.util.concurrent.ForkJoinPool;
import java.util.concurrent.RecursiveTask;

public class ForkJoinPoolTest {
    static int[] nums = new int[100000000];
    static final int MAX_NUM = 50;

    static {
        int num = 1;
        for (int i = 0; i < nums.length; i++) {
            nums[i] = num;
            num += 1;
        }
    }

    static class AddTask extends RecursiveTask<Long> {
        int start, end;

        public AddTask(int start, int end) {
            this.start = start;
            this.end = end;
        }

        @Override
        protected Long compute() {
            if (end - start <= MAX_NUM) {
                long sum = 0L;
                for (int i = start; i < end; i++) {
                    sum += nums[i];
                }
                return sum;
            } else {
                int middle = start + (end - start) / 2;
                AddTask t1 = new AddTask(start, middle);
                AddTask t2 = new AddTask(middle, end);
                //fork方法会启动新线程
                t1.fork();
                t2.fork();
                return t1.join() + t2.join();
            }
        }
    }
    public static void main(String[] args) throws IOException {
        ForkJoinPool pool = new ForkJoinPool();
        AddTask task = new AddTask(0, nums.length);
        long start = System.nanoTime();
        pool.execute(task);
        long res = task.join();
        System.out.println(res + " 耗时：" + (System.nanoTime() - start) / 100000000 + "seconds");
    }
}