ThreadPoolExecutor execute 和 submit

ThreadPoolExecutor 除了execute 方法用来提交任务，还有submit 方法，他们的区别就是后者有返回值，其原理是在前者的基础上使用了Future 接口。因此，我之前的示例改用submit 方法来实现会更加简单。原因有二：

1、submit 方法提交的任务有返回值，方便判断每个任务的最终运行结果，无需引入状态标识变量

2、future 的get方法是阻塞的，无需引入CountDownLatch 并发包工具也可以等待所有线程运行完毕

直接上代码

package com.yzy.test;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.*;
import java.util.concurrent.atomic.AtomicInteger;

public class Main2 {

    public static void main(String[] args) throws ExecutionException, InterruptedException {
        // 线程安全的计数器
        AtomicInteger totalRows = new AtomicInteger(0);

        // 创建线程池，其中核心线程10，也是我期望的最大并发数，最大线程数和队列大小都为30，即我的总任务数
        ThreadPoolExecutor executor = new ThreadPoolExecutor(10, 30, 60L, TimeUnit.SECONDS, new ArrayBlockingQueue<>(30));

        // 创建 Future 集合，用来存储Future 实例
        List<Future<Integer>> futureList = new ArrayList<>();

        for (int i = 0; i < 30; i++) {
            final int index = i;
            // 这里创建 Callable 实例，返回整型
            Future<Integer> future = executor.submit(new Callable<Integer>() {
                @Override
                public Integer call() throws Exception {
                    // 模拟异常的线程，返回0
                    if (index % 5 == 0) {
                        return 0;
                    }

                    int times = 0;
                    boolean loop = true;
                    // 模拟数据拉取过程可能需要分页
                    while (loop) {
                        // 模拟每个任务需要分页5次
                        if (times >= 5) {
                            break;
                        }
                        times++;

                        // 模拟计数
                        totalRows.incrementAndGet();
                        // 模拟耗时
                        try {
                            // 打印运行情况
                            System.out.println(Thread.currentThread().getName() + ":" + index + ":" + times);
                            Thread.sleep(Long.valueOf(String.valueOf(new Double(Math.random() * 1000).intValue())));
                        } catch (InterruptedException e) {
                            e.printStackTrace();
                        }
                    }
                    // 成功返回1
                    return 1;
                }
            });

            futureList.add(future);
        }

        // 标记多线程关闭，但不会立马关闭
        executor.shutdown();

        boolean flag = true;
        for (Future<Integer> future : futureList) {
            // 获取每一个线程的返回结果，如果该线程未完成，会阻塞
            if (future.get() == 0) {
                flag = false;
            }
        }

        if (flag) {
            System.out.println("全部成功了,计数：" + totalRows.get());
        } else {
            System.out.println("失败了");
        }
    }
}

运行结果

//省略若干行
pool-1-thread-7:24:5
pool-1-thread-3:29:3
pool-1-thread-3:29:4
pool-1-thread-5:28:3
pool-1-thread-6:26:4
pool-1-thread-6:26:5
pool-1-thread-5:28:4
pool-1-thread-5:28:5
pool-1-thread-3:29:5
失败了

然后注释这部分代码

if (index % 5 == 0) {
    return 0;
}

再次运行：

//省略若干行
pool-1-thread-7:24:4
pool-1-thread-7:24:5
pool-1-thread-1:26:5
pool-1-thread-8:25:4
pool-1-thread-8:25:5
pool-1-thread-3:29:4
pool-1-thread-3:29:5
全部成功了,计数：150

链接：https://www.jianshu.com/p/4060dd274dbd