ForkJoin 任务分解

 1 package com.thread.test.thread;
 2 
 3 import java.io.File;
 4 import java.util.ArrayList;
 5 import java.util.List;
 6 import java.util.concurrent.ForkJoinPool;
 7 import java.util.concurrent.RecursiveTask;
 8 
 9 /**
10  * Created by windwant on 2016/6/3.
11  */
12 public class MyForkJoin {
13 
14     public static void main(String[] args) {
15         MyTask task = new MyTask(new File("D:\\MPS"));
16         Integer sum = new ForkJoinPool().invoke(task);
17         System.out.println(sum);
18     }
19 }
20 
21 class MyTask extends RecursiveTask<Integer>{
22 
23     public Integer num = 0;
24 
25     private File file;
26     MyTask(File file){
27         this.file = file;
28     }
29 
30     @Override
31     protected Integer compute() {
32         List<MyTask> taskList = new ArrayList<MyTask>();
33         if(file.isDirectory()){
34             File[] list = file.listFiles();
35             for(File subf: list){
36                 if(subf.isDirectory()){
37                     MyTask mt = new MyTask(subf);
38                     taskList.add(mt);
39                 }else{
40                     num++;
41                 }
42             }
43         }else{
44             num = 1;
45         }
46 
47         if(!taskList.isEmpty()){
48             //同下
49 //            for(MyTask mtask: taskList){
50 //                mtask.fork();
51 //            }
52 //            for(MyTask mtask: taskList){
53 //                num += mtask.join();
54 //            }
55 
56             for(MyTask mtask: invokeAll(taskList)){
57                 num += mtask.join();
58             }
59         }
60         return num;
61     }
62 }