线程池ThreadPoolExecutor的使用Demo

由于阿里开发手册提出禁用Executor创建线程池，主要原因是容易引起OOM，推荐使用ThreadPoolExecutor，自己手动创建线程池，根据实际情况设置核心线程数，防止OOM

public class ThreadTest {

    /**
     * ThreadPoolExecutor使用
     */
    public static void main(String[] args) throws InterruptedException, IOException {
        //核心线程池大小
        int corePoolSize = 2;
        //最大线程池大小
        int maximumPoolSize = 4;
        //线程最大空闲时间
        long keepAliveTime = 10;
        //时间单位
        TimeUnit unit = TimeUnit.SECONDS;
        //线程等待队列
        BlockingQueue<Runnable> workQueue = new ArrayBlockingQueue<>(2);
        //线程创建工厂
        ThreadFactory threadFactory = new NameTreadFactory();
        //拒绝策略
        RejectedExecutionHandler handler = new SelfIgnorePolicy();

        //创建线程(根据实际情况确定核心线程池大小)
        ThreadPoolExecutor executor = new ThreadPoolExecutor(corePoolSize, maximumPoolSize, keepAliveTime, unit,
                workQueue, threadFactory, handler);

        //预启动所有核心线程
        executor.prestartAllCoreThreads();

        for (int i = 1; i <= 10; i++) {
            MyTask task = new MyTask(String.valueOf(i));
            executor.execute(task);
        }

        //关闭线程池
        executor.shutdown();

        //等待线程执行结束
        executor.awaitTermination(Long.MAX_VALUE, TimeUnit.DAYS);

        System.in.read(); //阻塞主线程
    }

    static class NameTreadFactory implements ThreadFactory {

        private final AtomicInteger mThreadNum = new AtomicInteger(1);

        @Override
        public Thread newThread(Runnable r) {
            Thread t = new Thread(r, "my-thread-" + mThreadNum.getAndIncrement());
            System.out.println(t.getName() + " has been created");
            return t;
        }
    }

    /**
     * @Description 自定义策略(拒绝策略)
     *
     */
    public static class SelfIgnorePolicy implements RejectedExecutionHandler {

        public void rejectedExecution(Runnable r, ThreadPoolExecutor e) {
            doLog(r, e);
        }

        private void doLog(Runnable r, ThreadPoolExecutor e) {
            //可做日志的记录
            System.err.println( r.toString() + " rejected");
        }
    }

    /**
     * @Description 线程执行逻辑
     */
    static class MyTask implements Runnable {
        private String name;

        public MyTask(String name) {
            this.name = name;
        }

        @Override
        public void run() {
            try {
                //
                System.out.println(this.toString() + " is running!");
                Thread.sleep(3000); //程序睡眠3秒
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }

        public String getName() {
            return name;
        }

        @Override
        public String toString() {
            return "MyTask [name=" + name + "]";
        }
    }
}

newFixedThreadPool线程池大小确定方式：

 如果是CPU密集型应用，则线程池大小设置为N+1，N代表机器的核数
如果是IO密集型应用，则线程池大小设置为2N+1，N代表机器的核数

如果程序计算多那就是CPU密集型，如果程序操作文件多那就是IO密集型

查看cpu核心数：

cat /proc/cpuinfo

当然实际开发还是需要根据实际情况设置线程数。