快捷導(dǎo)航

Java線程池ForkJoinPool實例解析

更新時間：2020年02月04日 10:38:03 作者：塵世間迷茫的小書童

這篇文章主要介紹了Java線程池ForkJoinPool實例解析,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下

背景：ForkJoinPool的優(yōu)勢在于，可以充分利用多cpu，多核cpu的優(yōu)勢，把一個任務(wù)拆分成多個“小任務(wù)”，把多個“小任務(wù)”放到多個處理器核心上并行執(zhí)行；當多個“小任務(wù)”執(zhí)行完成之后，再將這些執(zhí)行結(jié)果合并起來即可。這種思想值得學(xué)習(xí)。

import java.io.IOException;
import java.util.Arrays;
import java.util.Random;
import java.util.concurrent.RecursiveAction;
import java.util.concurrent.RecursiveTask;

/**
 * 分割 合并 線程池
 * 類似于遞歸的歸并排序，這個是多線程的遞歸排序
 * jdk1.7之后加的
 */
public class ForkJoinPool {

  static int[] nums = new int[1000000];
  static final int MAX_NUM = 50000;
  static Random random = new Random();

  static {
    for(int i=0; i<nums.length; i++) {
      nums[i] = random.nextInt(100);
    }
    System.out.println("總和1:" + Arrays.stream(nums).sum());
  }

  /**
   * ForkJoinPool里面必須是ForkJoinTask，
   * 直接繼承ForkJoinTask類寫起來比較麻煩，
   * 可以繼承RecursiveAction(無返回值)和RecursiveTask(遞歸任務(wù)，有返回值)
   */
//  static class AddTask extends RecursiveAction {
//
//    int start, end;
//
//    public AddTask(int start, int end) {
//      this.start = start;
//      this.end = end;
//    }
//
//    @Override
//    protected void compute() {
//      if(end-start <= MAX_NUM) {
//        long sum = 0L;
//        for(int i=start; i<end; i++) sum += nums[i];
//        System.out.println("from:" + start + " to:" + end + " =" + sum);
//      }else{
//        int middle = start + (end - start) / 2;
//
//        AddTask subTask1 = new AddTask(start, middle);
//        AddTask subTask2 = new AddTask(middle, end);
//        subTask1.fork();
//        subTask2.fork();
//      }
//    }
//  }

  /**
   * 有返回值
   */
  static class AddTask extends RecursiveTask<Long> {

    int start, end;

    public AddTask(int start, int end) {
      this.start = start;
      this.end = end;
    }

    @Override
    protected Long compute() {
      if(end-start <= MAX_NUM) {
        long sum = 0L;
        for(int i=start; i<end; i++) sum += nums[i];
        System.out.println("from:" + start + " to:" + end + " =" + sum);
        return sum;
      }

      int middle = start + (end - start) / 2;
      AddTask subTask1 = new AddTask(start, middle);
      AddTask subTask2 = new AddTask(middle, end);
      subTask1.fork();
      subTask2.fork();
      return subTask1.join() + subTask2.join();
    }
  }

  public static void main(String[] args) {

    java.util.concurrent.ForkJoinPool pool = new java.util.concurrent.ForkJoinPool();
    AddTask task = new AddTask(0, nums.length);
    pool.execute(task); //精靈線程
    System.out.println("總和2:" + task.join());
    try {
      System.in.read(); //阻塞主線程
    } catch (IOException e) {
      e.printStackTrace();
    }
  }

}

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: