在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 ()默认会创建一个与现有处理个数相同的线程数去执行任务,合理有效利用资源。