Java任务拆分

在Java后端开发过程中，经常会遇到大任务量的计算或者执行任务，此时我们就可以使用java jdk里面自带的拆分思想和方法，拆分任务主要依靠两个抽象类，一个是RecursiveAction，执行不需要有返回值的任务时，可以继承这个类。一个是RecursiveTask，这个是RecursiveAction的父类。他可以堵塞等待返回值，获取相应的计算结果，这边我就举个有返回值的例子，计算1到100，每三个数分为一个计算，完成分批计算整合。

public class ComplexCaculateTask extends RecursiveTask<Integer> {
    private static final long serialVersionUID = -4068008344384908582L;
    private Integer max_limit = 3;
    private Integer start;
    private Integer end;
 
    public ComplexCaculateTask(Integer start, Integer end) {
        this.start = start;
        this.end = end;
    }
 
    @Override
    protected Integer compute() {
        Integer sum = 0;
        if (start - end <= max_limit) {
            for (int i = start; i <= end; i++) {
                sum += i;
            }
        } else {
            //任务拆分2分
            int middle = (end + start) / 2;
            ComplexCaculateTask leftTask = new ComplexCaculateTask(start, middle);
            ComplexCaculateTask rightTask = new ComplexCaculateTask(middle, end);
            //执行任务
            leftTask.fork();
            rightTask.fork();
            //汇总分任务
            Integer leftResult = leftTask.join();
            Integer rightResult = rightTask.join();
            sum = leftResult + rightResult;
        }
        return sum;
    }
 
    public Integer computeSum() {
        Integer sum = 0;
        for (int i = start; i <= end; i++) {
            sum += i;
        }
        return sum;
    }
 
    public static void main(String[] args) {
        long start = System.currentTimeMillis();
        ForkJoinPool forkJoinPool = new ForkJoinPool();
        ComplexCaculateTask task = new ComplexCaculateTask(1, (Integer) Integer.MAX_VALUE / 10);
        Future<Integer> result = forkJoinPool.submit(task);
        try {
            System.out.println("任务拆分: " + result.get());
            System.out.println("总共消耗: " + (System.currentTimeMillis() - start) + " ms.");
            System.out.println("普通计算: " + task.computeSum());
            System.out.println("总共消耗: " + (System.currentTimeMillis() - start) + " ms.");
        } catch (InterruptedException e) {
            e.printStackTrace();
        } catch (ExecutionException e) {
            e.printStackTrace();
        }
        forkJoinPool.shutdown();
    }
}

ForkJoinPool这个类是实现了ExecutorService接口，用future等待计算结果返回，关闭线程池。整个计算流程就是一个拆分任务组合结果的思想，在大任务下的场景下可以提高计算或者执行效率，有效利用cpu等。 ForkJoinPool poll = new ForkJoinPool ()默认会创建一个与现有处理个数相同的线程数去执行任务，合理有效利用资源。

猜你喜欢