1. 程式人生 > >將複雜的大任務拆分成多個簡單的小任務進行計算

將複雜的大任務拆分成多個簡單的小任務進行計算

package thread;


import java.util.concurrent.ExecutionException;
import java.util.concurrent.ForkJoinPool;
import java.util.concurrent.Future;
import java.util.concurrent.RecursiveTask;
/**
 * Fork/Join框架實現多執行緒計算<br/>
 * 計算1+2+3+4+...+99+100的總和<br/>
 * 因為本程式模擬的是最簡單的計算,耗時極少,所以多執行緒計算還不如單執行緒計算快。<br/>
 * 但如果是比較耗時的複雜計算,多執行緒計算就快得多啦!
 * @author zhaoyujie
 *
 */
public class CountTask extends RecursiveTask<Integer> {


private static final long serialVersionUID = -4488036422261690638L;

private static final int THRESHOLD = 10;
private int start;
private int end;

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

@Override
protected Integer compute() {
int sum = 0;
//如果任務足夠小就計算任務
boolean canCompute = (end-start)<=THRESHOLD;
if(canCompute){
for(int i=start; i<=end; i++){
sum += i;
}
}else{
//如果任務大於閥值,就分裂成兩個子任務計算
int middle = (start+end)/2;
CountTask leftTask = new CountTask(start, middle);
CountTask rightTask = new CountTask(middle+1, end);
leftTask.fork();
rightTask.fork();
//等待子任務執行完,並得到其結果
int leftResult = leftTask.join();
int rightResult = rightTask.join();
sum = leftResult + rightResult;
}
return sum;
}

public static void main(String[] args) {
long currTime = System.currentTimeMillis();
ForkJoinPool forkJoinPool = new ForkJoinPool();
CountTask task = new CountTask(1, 100);
Future<Integer> future = forkJoinPool.submit(task);
try {
System.out.println(future.get());
System.out.println("用時毫秒數:"+(System.currentTimeMillis()-currTime));
} catch (InterruptedException e) {
e.printStackTrace();
} catch (ExecutionException e) {
e.printStackTrace();
}
}


}