import java.util.concurrent.ExecutionException;
import java.util.concurrent.ForkJoinPool;
import java.util.concurrent.ForkJoinTask;
import java.util.concurrent.RecursiveTask;
import java.util.stream.IntStream;
public class ForkJoinRecursiveTaskTest {
private static int threshold = 100;
public static void main(String[] args) {
final ForkJoinPool forkJoinPool = new ForkJoinPool();
ForkJoinTask<Integer> future = forkJoinPool.submit(new MyRecursiveTask(0, 1000));
try {
Integer result = future.get();
System.out.println(result);
} catch (InterruptedException | ExecutionException e) {
e.printStackTrace();
}
}
private static class MyRecursiveTask extends RecursiveTask<Integer> {
private final int start;
private final int end;
MyRecursiveTask(int start, int end) {
this.start = start;
this.end = end;
}
@Override
protected Integer compute() {
if (end - start <= threshold) {
return IntStream.rangeClosed(start, end).sum();
} else {
int middle = (start + end) / 2;
MyRecursiveTask firstTask = new MyRecursiveTask(start, middle);
MyRecursiveTask secondTask = new MyRecursiveTask(middle + 1, end);
invokeAll(firstTask, secondTask);
return firstTask.join() + secondTask.join();
}
}
}
}