【Java 并发编程】ForkJoin 框架使用代码实例

在文章

JUC并行计算框架 Fork/Join 原理图文详解&代码示例

中，我们详细介绍了 ForkJoin 框架的原理，这里我们再给出一个代码实例：

代码语言：javascript复制

package i.juc;

import java.util.Arrays;
import java.util.List;

/**
 * @author: Jack
 * 2020-06-03 10:20
 */
public class ForkJoinTest {
    public static void main(String[] args) {
        List<String> list = Arrays.asList("1", "2", "3", "4", "5", "6", "7", "8", "9");

        ForkJoinTaskDemo demo = new ForkJoinTaskDemo(list);
        List<ReportLine> result = demo.compute();

        System.out.println(result);

    }
}

代码语言：javascript复制

package i.juc;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.RecursiveTask;
import java.util.stream.Collectors;

/**
 * @author: Jack
 * 2020-06-03 10:16
 */
public class ForkJoinTaskDemo extends RecursiveTask<List<ReportLine>> {

    List<String> workNoList;

    public ForkJoinTaskDemo(List<String> workNoList) {
        this.workNoList = workNoList;
    }

    @Override
    protected List<ReportLine> compute() {
        try {
            Thread.sleep(1000L);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        if (workNoList.size() > 0 && workNoList.size() <= 3) {
            return workNoList.stream().map(ReportLine::new).collect(Collectors.toList());
        }

        List<String> list1 = workNoList.subList(0, 3);
        List<String> list2 = workNoList.subList(3, 6);
        List<String> list3 = workNoList.subList(6, 9);

        ForkJoinTaskDemo t1 = new ForkJoinTaskDemo(list1);
        ForkJoinTaskDemo t2 = new ForkJoinTaskDemo(list2);
        ForkJoinTaskDemo t3 = new ForkJoinTaskDemo(list3);

        invokeAll(t1, t2, t3);

        List<ReportLine> result = new ArrayList<>();

        List<ReportLine> result1 = t1.join();
        List<ReportLine> result2 = t2.join();
        List<ReportLine> result3 = t3.join();

        result.addAll(result1);
        result.addAll(result2);
        result.addAll(result3);

        return result;

    }

}

class ReportLine {
    String id;

    public ReportLine(String id) {
        this.id = id;
    }

    @Override
    public String toString() {
        return "ReportLine{"  
            "id='"   id   '''  
            '}';
    }
}