TestForkjoin.java 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. import java.util.concurrent.*;
  2. public class TestForkjoin {
  3. public static void main(String[] args) throws Exception {
  4. MyRecursiveTask fj = new MyRecursiveTask(1, 100000);
  5. ForkJoinPool pool = new ForkJoinPool();
  6. Future future = pool.submit(fj);
  7. System.out.println(future.get());
  8. }
  9. }
  10. class MyRecursiveTask extends RecursiveTask<Integer> {
  11. private final int threshold = 5;
  12. private int beg;
  13. private int end;
  14. MyRecursiveTask(int beg, int end) {
  15. this.beg = beg;
  16. this.end = end;
  17. }
  18. @Override
  19. protected Integer compute() {
  20. if (end - beg <= threshold) {
  21. int count = 0;
  22. for (int i = beg; i <= end; i++)
  23. count++;
  24. return count;
  25. }
  26. int mid = beg + (end - beg) / 2;
  27. MyRecursiveTask left = new MyRecursiveTask(beg, mid);
  28. MyRecursiveTask right = new MyRecursiveTask(mid + 1, end);
  29. left.fork();
  30. right.fork();
  31. return left.join() + right.join();
  32. }
  33. }