我正在尝试了解 ForkJoinPool 框架并遇到以下示例:
public class ArrayCounter extends RecursiveTask<Integer> {
int[] array;
int threshold = 100_000;
int start;
int end;
public ArrayCounter(int[] array, int start, int end) {
this.array = array;
this.start = start;
this.end = end;
}
protected Integer compute() {
if (end - start < threshold) {
return computeDirectly();
} else {
int middle = (end + start) / 2;
ArrayCounter subTask1 = new ArrayCounter(array, start, middle);
ArrayCounter subTask2 = new ArrayCounter(array, middle, end);
invokeAll(subTask1, subTask2);
return subTask1.join() + subTask2.join();
}
}
protected Integer computeDirectly() {
Integer count = 0;
for (int i = start; i < end; i++) {
if (array[i] % 2 == 0) {
count++;
}
}
return count;
}
}
Main :
public class ForkJoinRecursiveTaskTest
{
static final int SIZE = 10_000_000;
static int[] array = randomArray();
public static void main(String[] args) {
ArrayCounter mainTask = new ArrayCounter(array, 0, SIZE);
ForkJoinPool pool = new ForkJoinPool();
Integer evenNumberCount = pool.invoke(mainTask);
System.out.println("Number of even numbers: " + evenNumberCount);
}
static int[] randomArray() {
int[] array = new int[SIZE];
Random random = new Random();
for (int i = 0; i < SIZE; i++) {
array[i] = random.nextInt(100);
}
return array;
}
}
根据Java文档,invokeAll()将任务提交到池中并返回结果。因此不需要单独的join()。有人可以解释一下为什么在这种情况下需要单独加入吗?
在你的例子中,你正在使用RecursiveTask<Integer>
所以你期望返回一个值compute()
method.
让我们看看invokAll(t1,t12)
签名。
static void invokeAll(ForkJoinTask<?> t1, ForkJoinTask<?> t2)
so invokeAll()
没有返回值。
根据文档:
分叉给定的任务,当每个任务的 isDone 成立或遇到(未经检查的)异常时返回,在这种情况下将重新抛出异常。
So:
return subTask1.join() + subTask2.join();
是你的例子的关键。
两个任务在每次完成任务后合并,并将结果递归地传递给下一个调用compute()
method.
task.join()
完成后返回计算结果。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)