ForkJoinPool 使用

ForkJoinPool 使用

public class CalculateTaskTest {

    // 具体调用
    public List<Test1> parallelGetTest1S(List<Test2> params) throws ExecutionException, InterruptedException {
        ForkJoinPool pool = new ForkJoinPool(10);
        ForkJoinTask<List<Test1>> submit = pool.submit(new CalculateTask(params));
        List<Test1> Test1 = submit.get();
        return Test1;
    }

    // 线程池分片操作
    private class CalculateTask extends RecursiveTask<List<Test1>> {
        // 调用参数
        private List<Test2> params;

        // 构造函数
        public CalculateTask(List<Test2> params) {
            this.params = params;
        }

        /**
         * The main computation performed by this task.     * @return the result of the computation
         */
        @Override
        protected List<Test1> compute() {
            List<Test1> Test1S = new ArrayList<>();
            // 判断是否进行分片
            // 阀值200 自行修改
            if (params.size() <= 200) {
                // 具体操作
//                Test1S = schemaMapper.getDataByCondition(params);
                return Test1S;
            }
            // 对参数进行分片 200 分片大小
            List<List<Test2>> partition = Lists.partition(params, 200);
            ArrayList<CalculateTask> tasks = Lists.newArrayList();
            for (List<Test2> item : partition) {
                // 对分片任务进行添加
                tasks.add(new CalculateTask(item));
            }
            invokeAll(tasks);
            // 获取结果值
            for (CalculateTask task : tasks) {
                Test1S.addAll(task.join());
            }
            return Test1S;
        }
    }
}