Java 7新特性之ForkJoin框架

Bertha 。 2022-03-19 01:28 270阅读 0赞

## Fork/Join框架 ##

Fork/Join框架是[Java7][]提供了的一个用于并行执行任务的框架， 是一个把大任务分割成若干个小任务，最终汇总每个小任务结果后得到大任务结果的框架。

我们再通过Fork和Join这两个单词来理解下Fork/Join框架，Fork就是把一个大任务切分为若干子任务并行的执行，Join就是合并这些子任务的执行结果，最后得到这个大任务的结果。比如计算1+2+3+4，可以分割成2个子任务，每个子任务分别对2个数进行求和，最终汇总这2个子任务的结果。

直接看代码：

import java.util.concurrent.ForkJoinPool;
    import java.util.concurrent.Future;
    import java.util.concurrent.RecursiveTask;
    
    public class CountTask extends RecursiveTask<Integer>
    {
        private static final long serialVersionUID = -3611254198265061729L;
    
        public static final int threshold = 2;
        private int start;
        private int end;
    
        public CountTask(int start, int end)
        {
            this.start = start;
            this.end = end;
        }
    
        @Override
        protected Integer compute()
        {
            int sum = 0;
    
            //如果任务足够小就计算任务
            boolean canCompute = (end - start) <= threshold;
            if(canCompute)
            {
                for (int i=start; i<=end; i++)
                {
                    sum += i;
                }
            }
            else
            {
                // 如果任务大于阈值，就分裂成两个子任务计算
                int middle = (start + end)/2;
                CountTask leftTask = new CountTask(start, middle);
                CountTask rightTask = new CountTask(middle+1, end);
    
                // 执行子任务
                leftTask.fork();
                rightTask.fork();
    
                //等待任务执行结束合并其结果
                int leftResult = leftTask.join();
                int rightResult = rightTask.join();
    
                //合并子任务
                sum = leftResult + rightResult;
    
            }
    
            return sum;
        }
    
        public static void main(String[] args)
        {
            ForkJoinPool forkjoinPool = new ForkJoinPool();
    
            //生成一个计算任务，计算1+2+3+4
            CountTask task = new CountTask(1, 4);
    
            //执行一个任务
            Future<Integer> result = forkjoinPool.submit(task);
    
            try
            {
                System.out.println(result.get());
            }
            catch(Exception e)
            {
                System.out.println(e);
            }
        }
    
    }

[Java7]: https://www.baidu.com/s?wd=Java7&tn=24004469_oem_dg&rsv_dl=gh_pl_sl_csd