• （十五） 共享模型之工具【线程池】

    一、自定义线程池

    1. 简易线程池

    1. @Slf4j(topic = "c.TestPool")
    2. public class TestPool {
    3.  
    4.     public static void main(String[] args) {
    5.         ThreadPool threadPool = new ThreadPool(2,
    6.                 1000, TimeUnit.MILLISECONDS, 10);
    7.         for (int i = 0; i < 5; i++) {
    8.             int j =i;
    9.             threadPool.execute(() -> {
    10.                 log.debug("{}",j);
    11.             });
    12.         }
    13.     }
    14.  
    15. }
    16.  
    17. @Slf4j(topic = "c.ThreadPool")
    18. class ThreadPool {
    19.     // 任务队列
    20.     private BlockingQueue taskQueue;
    21.  
    22.     // 线程集合
    23.     private HashSet workers = new HashSet<>();
    24.  
    25.     // 核心线程数
    26.     private int coreSize;
    27.  
    28.     // 获取任务时的超时时间
    29.     private long timeout;
    30.  
    31.     private TimeUnit timeUnit;
    32.  
    33.     public ThreadPool(int coreSize, long timeout, TimeUnit timeUnit, int queueCapcity) {
    34.         this.coreSize = coreSize;
    35.         this.timeout = timeout;
    36.         this.timeUnit = timeUnit;
    37.         this.taskQueue = new BlockingQueue<>(queueCapcity);
    38.     }
    39.  
    40.     // 执行任务
    41.     public void execute(Runnable task) {
    42.         // 当任务数没有超过 coreSize 时，直接交给 worker 对象执行
    43.         // 如果任务数超过 coreSize 时，加入任务队列暂存
    44.         synchronized (workers) {
    45.             if (workers.size() < coreSize) {
    46.                 Worker worker = new Worker(task);
    47.                 log.debug("新增 worker{}, {}", worker, task);
    48.                 workers.add(worker);
    49.                 worker.start();
    50.             } else {
    51.                 taskQueue.put(task);
    52.             }
    53.         }
    54.  
    55.     }
    56.  
    57.  
    58.     class Worker extends Thread {
    59.         private Runnable task;
    60.  
    61.         public Worker(Runnable task) {
    62.             this.task = task;
    63.         }
    64.  
    65.         @Override
    66.         public void run() {
    67.             // 执行任务
    68.             // 1) 当 task 不为空，执行任务
    69.             // 2) 当 task 执行完毕，再接着从任务队列获取任务并执行
    70. //            while (task != null || (task = taskQueue.take()) != null) {
    71.             while(task != null || (task = taskQueue.poll(timeout, timeUnit)) != null) {
    72.                 try {
    73.  
    74.                     log.debug("正在执行...{}", task);
    75.                     task.run();
    76.                 } catch (Exception e) {
    77.                     e.printStackTrace();
    78.                 } finally {
    79.                     task = null;
    80.                 }
    81.             }
    82.             synchronized (workers) {
    83.                 log.debug("worker 被移除{}", this);
    84.                 workers.remove(this);
    85.             }
    86.  
    87.         }
    88.     }
    89. }
    90.  
    91. // 阻塞队列
    92. @Slf4j(topic = "c.BlockingQueue")
    93. class BlockingQueue {
    94.     // 1. 任务队列
    95.     private Deque queue = new ArrayDeque<>();
    96.  
    97.     // 2. 锁
    98.     private ReentrantLock lock = new ReentrantLock();
    99.  
    100.     // 3. 生产者条件变量
    101.     private Condition fullWaitSet = lock.newCondition();
    102.  
    103.     // 4. 消费者条件变量
    104.     private Condition emptyWaitSet = lock.newCondition();
    105.  
    106.     // 5. 容量
    107.     private int capcity;
    108.  
    109.     public BlockingQueue(int capcity) {
    110.         this.capcity = capcity;
    111.     }
    112.  
    113.     // 带超时阻塞获取
    114.     public T poll(long timeout, TimeUnit unit) {
    115.         lock.lock();
    116.         try {
    117.             // 将 timeout 统一转换为 纳秒
    118.             long nanos = unit.toNanos(timeout);
    119.             while (queue.isEmpty()) {
    120.                 try {
    121.                     if (nanos <= 0) {
    122.                         return null;
    123.                     }
    124.                     // 返回值是剩余时间
    125.                     nanos = emptyWaitSet.awaitNanos(nanos);
    126.                 } catch (InterruptedException e) {
    127.                     e.printStackTrace();
    128.                 }
    129.             }
    130.             T t = queue.removeFirst();
    131.             fullWaitSet.signal();
    132.             return t;
    133.         } finally {
    134.             lock.unlock();
    135.         }
    136.     }
    137.  
    138.     // 阻塞获取
    139.     public T take() {
    140.         lock.lock();
    141.         try {
    142.             while (queue.isEmpty()) {
    143.                 try {
    144.                     emptyWaitSet.await();
    145.                 } catch (InterruptedException e) {
    146.                     e.printStackTrace();
    147.                 }
    148.             }
    149.             T t = queue.removeFirst();
    150.             fullWaitSet.signal();
    151.             return t;
    152.         } finally {
    153.             lock.unlock();
    154.         }
    155.     }
    156.  
    157.     // 阻塞添加
    158.     public void put(T task) {
    159.         lock.lock();
    160.         try {
    161.             while (queue.size() == capcity) {
    162.                 try {
    163.                     log.debug("等待加入任务队列 {} ...", task);
    164.                     fullWaitSet.await();
    165.                 } catch (InterruptedException e) {
    166.                     e.printStackTrace();
    167.                 }
    168.             }
    169.             log.debug("加入任务队列 {}", task);
    170.             queue.addLast(task);
    171.             emptyWaitSet.signal();
    172.         } finally {
    173.             lock.unlock();
    174.         }
    175.     }
    176.  
    177.     public int size() {
    178.         lock.lock();
    179.         try {
    180.             return queue.size();
    181.         } finally {
    182.             lock.unlock();
    183.         }
    184.     }
    185. }

    2. 拒绝策略

    1. @Slf4j(topic = "c.TestPool")
    2. public class TestPool {
    3.     public static void main(String[] args) {
    4.         ThreadPool threadPool = new ThreadPool(1,
    5.                 1000, TimeUnit.MILLISECONDS, 1, (queue, task)->{
    6.             // 1. 死等
    7. //            queue.put(task);
    8.             // 2) 带超时等待
    9. //            queue.offer(task, 1500, TimeUnit.MILLISECONDS);
    10.             // 3) 让调用者放弃任务执行
    11. //            log.debug("放弃{}", task);
    12.             // 4) 让调用者抛出异常
    13. //            throw new RuntimeException("任务执行失败 " + task);
    14.             // 5) 让调用者自己执行任务
    15.             task.run();
    16.         });
    17.         for (int i = 0; i < 4; i++) {
    18.             int j = i;
    19.             threadPool.execute(() -> {
    20.                 try {
    21.                     Thread.sleep(1000L);
    22.                 } catch (InterruptedException e) {
    23.                     e.printStackTrace();
    24.                 }
    25.                 log.debug("{}", j);
    26.             });
    27.         }
    28.     }
    29. }
    30.  
    31. @FunctionalInterface // 拒绝策略
    32. interface RejectPolicy {
    33.     void reject(BlockingQueue queue, T task);
    34. }
    35.  
    36. @Slf4j(topic = "c.ThreadPool")
    37. class ThreadPool {
    38.     // 任务队列
    39.     private BlockingQueue taskQueue;
    40.  
    41.     // 线程集合
    42.     private HashSet workers = new HashSet<>();
    43.  
    44.     // 核心线程数
    45.     private int coreSize;
    46.  
    47.     // 获取任务时的超时时间
    48.     private long timeout;
    49.  
    50.     private TimeUnit timeUnit;
    51.  
    52.     private RejectPolicy rejectPolicy;
    53.  
    54.     // 执行任务
    55.     public void execute(Runnable task) {
    56.         // 当任务数没有超过 coreSize 时，直接交给 worker 对象执行
    57.         // 如果任务数超过 coreSize 时，加入任务队列暂存
    58.         synchronized (workers) {
    59.             if(workers.size() < coreSize) {
    60.                 Worker worker = new Worker(task);
    61.                 log.debug("新增 worker{}, {}", worker, task);
    62.                 workers.add(worker);
    63.                 worker.start();
    64.             } else {
    65. //                taskQueue.put(task);
    66.                 // 1) 死等
    67.                 // 2) 带超时等待
    68.                 // 3) 让调用者放弃任务执行
    69.                 // 4) 让调用者抛出异常
    70.                 // 5) 让调用者自己执行任务
    71.                 taskQueue.tryPut(rejectPolicy, task);
    72.             }
    73.         }
    74.     }
    75.  
    76.     public ThreadPool(int coreSize, long timeout, TimeUnit timeUnit, int queueCapcity, RejectPolicy rejectPolicy) {
    77.         this.coreSize = coreSize;
    78.         this.timeout = timeout;
    79.         this.timeUnit = timeUnit;
    80.         this.taskQueue = new BlockingQueue<>(queueCapcity);
    81.         this.rejectPolicy = rejectPolicy;
    82.     }
    83.  
    84.     class Worker extends Thread{
    85.         private Runnable task;
    86.  
    87.         public Worker(Runnable task) {
    88.             this.task = task;
    89.         }
    90.  
    91.         @Override
    92.         public void run() {
    93.             // 执行任务
    94.             // 1) 当 task 不为空，执行任务
    95.             // 2) 当 task 执行完毕，再接着从任务队列获取任务并执行
    96. //            while(task != null || (task = taskQueue.take()) != null) {
    97.             while(task != null || (task = taskQueue.poll(timeout, timeUnit)) != null) {
    98.                 try {
    99.                     log.debug("正在执行...{}", task);
    100.                     task.run();
    101.                 } catch (Exception e) {
    102.                     e.printStackTrace();
    103.                 } finally {
    104.                     task = null;
    105.                 }
    106.             }
    107.             synchronized (workers) {
    108.                 log.debug("worker 被移除{}", this);
    109.                 workers.remove(this);
    110.             }
    111.         }
    112.     }
    113. }
    114. @Slf4j(topic = "c.BlockingQueue")
    115. class BlockingQueue {
    116.     // 1. 任务队列
    117.     private Deque queue = new ArrayDeque<>();
    118.  
    119.     // 2. 锁
    120.     private ReentrantLock lock = new ReentrantLock();
    121.  
    122.     // 3. 生产者条件变量
    123.     private Condition fullWaitSet = lock.newCondition();
    124.  
    125.     // 4. 消费者条件变量
    126.     private Condition emptyWaitSet = lock.newCondition();
    127.  
    128.     // 5. 容量
    129.     private int capcity;
    130.  
    131.     public BlockingQueue(int capcity) {
    132.         this.capcity = capcity;
    133.     }
    134.  
    135.     // 带超时阻塞获取
    136.     public T poll(long timeout, TimeUnit unit) {
    137.         lock.lock();
    138.         try {
    139.             // 将 timeout 统一转换为 纳秒
    140.             long nanos = unit.toNanos(timeout);
    141.             while (queue.isEmpty()) {
    142.                 try {
    143.                     // 返回值是剩余时间
    144.                     if (nanos <= 0) {
    145.                         return null;
    146.                     }
    147.                     nanos = emptyWaitSet.awaitNanos(nanos);
    148.                 } catch (InterruptedException e) {
    149.                     e.printStackTrace();
    150.                 }
    151.             }
    152.             T t = queue.removeFirst();
    153.             fullWaitSet.signal();
    154.             return t;
    155.         } finally {
    156.             lock.unlock();
    157.         }
    158.     }
    159.  
    160.     // 阻塞获取
    161.     public T take() {
    162.         lock.lock();
    163.         try {
    164.             while (queue.isEmpty()) {
    165.                 try {
    166.                     emptyWaitSet.await();
    167.                 } catch (InterruptedException e) {
    168.                     e.printStackTrace();
    169.                 }
    170.             }
    171.             T t = queue.removeFirst();
    172.             fullWaitSet.signal();
    173.             return t;
    174.         } finally {
    175.             lock.unlock();
    176.         }
    177.     }
    178.  
    179.     // 阻塞添加
    180.     public void put(T task) {
    181.         lock.lock();
    182.         try {
    183.             while (queue.size() == capcity) {
    184.                 try {
    185.                     log.debug("等待加入任务队列 {} ...", task);
    186.                     fullWaitSet.await();
    187.                 } catch (InterruptedException e) {
    188.                     e.printStackTrace();
    189.                 }
    190.             }
    191.             log.debug("加入任务队列 {}", task);
    192.             queue.addLast(task);
    193.             emptyWaitSet.signal();
    194.         } finally {
    195.             lock.unlock();
    196.         }
    197.     }
    198.  
    199.     // 带超时时间阻塞添加
    200.     public boolean offer(T task, long timeout, TimeUnit timeUnit) {
    201.         lock.lock();
    202.         try {
    203.             long nanos = timeUnit.toNanos(timeout);
    204.             while (queue.size() == capcity) {
    205.                 try {
    206.                     if(nanos <= 0) {
    207.                         return false;
    208.                     }
    209.                     log.debug("等待加入任务队列 {} ...", task);
    210.                     nanos = fullWaitSet.awaitNanos(nanos);
    211.                 } catch (InterruptedException e) {
    212.                     e.printStackTrace();
    213.                 }
    214.             }
    215.             log.debug("加入任务队列 {}", task);
    216.             queue.addLast(task);
    217.             emptyWaitSet.signal();
    218.             return true;
    219.         } finally {
    220.             lock.unlock();
    221.         }
    222.     }
    223.  
    224.     public int size() {
    225.         lock.lock();
    226.         try {
    227.             return queue.size();
    228.         } finally {
    229.             lock.unlock();
    230.         }
    231.     }
    232.  
    233.     public void tryPut(RejectPolicy rejectPolicy, T task) {
    234.         lock.lock();
    235.         try {
    236.             // 判断队列是否满
    237.             if(queue.size() == capcity) {
    238.                 rejectPolicy.reject(this, task);
    239.             } else {  // 有空闲
    240.                 log.debug("加入任务队列 {}", task);
    241.                 queue.addLast(task);
    242.                 emptyWaitSet.signal();
    243.             }
    244.         } finally {
    245.             lock.unlock();
    246.         }
    247.     }
    248. }

    二、ThreadPoolExecutor

     1. 线程池状态

    ThreadPoolExecutor 使用 int 的 高 3 位来表示线程池状态，低 29 位表示线程数量

    从数字上比较，TERMINATED > TIDYING > STOP > SHUTDOWN > RUNNING 

    这些信息存储在一个原子变量 ctl 中，目的是将线程池状态与线程个数合二为一，这样就可以用一次 cas 原子操作进行赋值

    2. 构造方法（***）

    1. public ThreadPoolExecutor(int corePoolSize,
    2.                               int maximumPoolSize,
    3.                               long keepAliveTime,
    4.                               TimeUnit unit,
    5.                               BlockingQueue workQueue,
    6.                               ThreadFactory threadFactory,
    7.                               RejectedExecutionHandler handler)
    （1）corePoolSize 核心线程数目 ( 最多保留的线程数 )
    （2）maximumPoolSize 最大线程数目
    （3）keepAliveTime 生存时间 - 针对救急线程
    （4）unit 时间单位 - 针对救急线程
    （5）workQueue 阻塞队列
    （6）threadFactory 线程工厂 - 可以为线程创建时起个好名字
    （7）handler 拒绝策略

    工作方式：  

     （1）线程池中刚开始没有线程，当一个任务提交给线程池后，线程池会创建一个新线程来执行任务。

    （2）当线程数达到 corePoolSize 并没有线程空闲，这时再加入任务，新加的任务会被加入 workQueue 队列排队，直到有空闲的线程。
    （3）如果队列选择了有界队列，那么任务超过了队列大小时，会创建 maximumPoolSize - corePoolSize 数目的线程来救急。
    （4）如果线程到达 maximumPoolSize 仍然有新任务这时会执行拒绝策略。拒绝策略 jdk 提供了 4 种实现，其它著名框架也提供了实现
    1️⃣AbortPolicy 让调用者抛出 RejectedExecutionException 异常，这是 默认策略
    2️⃣CallerRunsPolicy 让调用者运行任务
    3️⃣DiscardPolicy 放弃本次任务
    4️⃣DiscardOldestPolicy 放弃队列中最早的任务，本任务取而代之
    5️⃣Dubbo 的实现，在抛出 RejectedExecutionException 异常之前会记录日志，并 dump 线程栈信息，方便定位问题
    6️⃣Netty 的实现，是创建一个新线程来执行任务
    7️⃣ActiveMQ 的实现，带超时等待（ 60s ）尝试放入队列，类似我们之前自定义的拒绝策略
    8️⃣PinPoint 的实现，它使用了一个拒绝策略链，会逐一尝试策略链中每种拒绝策略
    （5） 当高峰过去后，超过 corePoolSize 的救急线程如果一段时间没有任务做，需要结束节省资源，这个时间由  keepAliveTime 和 unit 来控制。

    拒绝策略 

    根据这个构造方法， JDK Executors 类中提供了众多工厂方法来创建各种用途的线程池

    3. newFixedThreadPool（固定大小线程池）

    1. public static ExecutorService newFixedThreadPool(int nThreads) {
    2.         return new ThreadPoolExecutor(nThreads, nThreads,
    3.                                       0L, TimeUnit.MILLISECONDS,
    4.                                       new LinkedBlockingQueue());
    5.     }
    特点：
    （1）核心线程数 == 最大线程数（没有救急线程被创建），因此也无需超时时间
    （2）阻塞队列是无界的，可以放任意数量的任务
    评价 适用于任务量已知，相对耗时的任务 
    1. public class TestThreadPoolExecutors {
    2.     public static void main(String[] args) {
    3.         ExecutorService pool = Executors.newFixedThreadPool(2);
    4.  
    5.         ExecutorService pool2 = Executors.newFixedThreadPool(2, new ThreadFactory() {
    6.             private AtomicInteger t = new AtomicInteger(1);
    7.  
    8.             @Override
    9.             public Thread newThread(Runnable r) {
    10.                 return new Thread(r, "mypool_t" + t.getAndIncrement());
    11.             }
    12.         });
    13.  
    14.         pool.execute(() -> {
    15.             log.debug("1");
    16.         });
    17.  
    18.         pool.execute(() -> {
    19.             log.debug("1");
    20.         });
    21.  
    22.         pool.execute(() -> {
    23.             log.debug("1");
    24.         });
    25.     }
    26. }

    4. newCachedThreadPool（带缓冲功能线程池）

    1. public static ExecutorService newCachedThreadPool() {
    2.         return new ThreadPoolExecutor(0, Integer.MAX_VALUE,
    3.                                       60L, TimeUnit.SECONDS,
    4.                                       new SynchronousQueue());
    5.     }
    特点：
    （1） 核心线程数是 0， 最大线程数是 Integer.MAX_VALUE ，救急线程的空闲生存时间是 60s ，意味着
    1️⃣全部都是救急线程（ 60s 后可以回收）
    2️⃣ 救急线程可以无限创建
    （2） 队列采用了 SynchronousQueue 实现特点是，它没有容量，没有线程来取是放不进去的（一手交钱、一手交 货） 
    1. @Slf4j(topic = "c.TestSynchronousQueue")
    2. public class TestSynchronousQueue {
    3.     public static void main(String[] args) {
    4.         SynchronousQueue integers = new SynchronousQueue<>();
    5.         new Thread(() -> {
    6.             try {
    7.                 log.debug("putting {} ", 1);
    8.                 integers.put(1);
    9.                 log.debug("{} putted...", 1);
    10.  
    11.                 log.debug("putting...{} ", 2);
    12.                 integers.put(2);
    13.                 log.debug("{} putted...", 2);
    14.             } catch (InterruptedException e) {
    15.                 e.printStackTrace();
    16.             }
    17.         },"t1").start();
    18.  
    19.         sleep(1);
    20.  
    21.         new Thread(() -> {
    22.             try {
    23.                 log.debug("taking {}", 1);
    24.                 integers.take();
    25.             } catch (InterruptedException e) {
    26.                 e.printStackTrace();
    27.             }
    28.         },"t2").start();
    29.  
    30.         sleep(1);
    31.  
    32.         new Thread(() -> {
    33.             try {
    34.                 log.debug("taking {}", 2);
    35.                 integers.take();
    36.             } catch (InterruptedException e) {
    37.                 e.printStackTrace();
    38.             }
    39.         },"t3").start();
    40.     }
    41. }

    评价

    （1）整个线程池表现为线程数会根据任务量不断增长，没有上限，当任务执行完毕，空闲 1分钟后释放线程。
    （2）适合任务数比较密集，但每个任务执行时间较短的情况

    5. newSingleThreadExecutor（单线程线程池）

    1. public static ExecutorService newSingleThreadExecutor() {
    2.         return new FinalizableDelegatedExecutorService
    3.             (new ThreadPoolExecutor(1, 1,
    4.                                     0L, TimeUnit.MILLISECONDS,
    5.                                     new LinkedBlockingQueue()));
    6.     }
    使用场景：
    希望多个任务排队执行。线程数固定为 1 ，任务数多于 1 时，会放入无界队列排队。任务执行完毕，这唯一的线程也不会被释放。
    区别：
    （1） 自己创建一个单线程串行执行任务，如果任务执行失败而终止那么没有任何补救措施，而线程池还会新建一个线程，保证池的正常工作
    （2） Executors.newSingleThreadExecutor() 线程个数始终为 1 ，不能修改
    1️⃣FinalizableDelegatedExecutorService 应用的是装饰器模式，只对外暴露了  ExecutorService 接口，因此不能调用 ThreadPoolExecutor 中特有的方法
    （3） Executors.newFixedThreadPool(1) 初始时为 1 ，以后还可以修改
    1️⃣ 对外暴露的是 ThreadPoolExecutor 对象，可以强转后调用 setCorePoolSize 等方法进行修改 
    1. @Slf4j(topic = "c.TestExecutors")
    2. public class TestExecutors {
    3.     public static void main(String[] args) throws InterruptedException {
    4.         test2();
    5.     }
    6.  
    7.     public static void test2() {
    8.         ExecutorService pool = Executors.newSingleThreadExecutor();
    9.         pool.execute(() -> {
    10.             log.debug("1");
    11.             int i = 1 / 0;
    12.             log.debug("111");
    13.         });
    14.  
    15.         pool.execute(() -> {
    16.             log.debug("2");
    17.         });
    18.  
    19.         pool.execute(() -> {
    20.             log.debug("3");
    21.         });
    22.     }
    23.  
    24. }

    6. 提交任务

    1.     // 执行任务
    2.     void execute(Runnable command);
    3.  
    4.     // 提交任务 task，用返回值 Future 获得任务执行结果
    5.      Future submit(Callable task);
    6.  
    7.     // 提交 tasks 中所有任务
    8.      List> invokeAll(Collection> tasks)
    9.             throws InterruptedException;
    10.  
    11.     // 提交 tasks 中所有任务，带超时时间
    12.      List> invokeAll(Collection> tasks,
    13.                                   long timeout, TimeUnit unit)
    14.             throws InterruptedException;
    15.  
    16.     // 提交 tasks 中所有任务，哪个任务先成功执行完毕，返回此任务执行结果，其它任务取消
    17.      T invokeAny(Collection> tasks)
    18.             throws InterruptedException, ExecutionException;
    19.  
    20.     // 提交 tasks 中所有任务，哪个任务先成功执行完毕，返回此任务执行结果，其它任务取消，带超时时间
    21.      T invokeAny(Collection> tasks,
    22.                     long timeout, TimeUnit unit)
    23.             throws InterruptedException, ExecutionException, TimeoutException;
    24.     }

    1. @Slf4j(topic = "c.TestSubmit")
    2. public class TestSubmit {
    3.  
    4.     public static void main(String[] args) throws ExecutionException, InterruptedException {
    5.         ExecutorService pool = Executors.newFixedThreadPool(1);
    6.  
    7.     }
    8.  
    9.     private static void method3(ExecutorService pool) throws InterruptedException, ExecutionException {
    10.         String result = pool.invokeAny(Arrays.asList(
    11.                 () -> {
    12.                     log.debug("begin 1");
    13.                     Thread.sleep(1000);
    14.                     log.debug("end 1");
    15.                     return "1";
    16.                 },
    17.                 () -> {
    18.                     log.debug("begin 2");
    19.                     Thread.sleep(500);
    20.                     log.debug("end 2");
    21.                     return "2";
    22.                 },
    23.                 () -> {
    24.                     log.debug("begin 3");
    25.                     Thread.sleep(2000);
    26.                     log.debug("end 3");
    27.                     return "3";
    28.                 }
    29.         ));
    30.         log.debug("{}", result);
    31.     }
    32.  
    33.     private static void method2(ExecutorService pool) throws InterruptedException {
    34.         List> futures = pool.invokeAll(Arrays.asList(
    35.                 () -> {
    36.                     log.debug("begin");
    37.                     Thread.sleep(1000);
    38.                     return "1";
    39.                 },
    40.                 () -> {
    41.                     log.debug("begin");
    42.                     Thread.sleep(500);
    43.                     return "2";
    44.                 },
    45.                 () -> {
    46.                     log.debug("begin");
    47.                     Thread.sleep(2000);
    48.                     return "3";
    49.                 }
    50.         ));
    51.  
    52.         futures.forEach( f ->  {
    53.             try {
    54.                 log.debug("{}", f.get());
    55.             } catch (InterruptedException | ExecutionException e) {
    56.                 e.printStackTrace();
    57.             }
    58.         });
    59.     }
    60.  
    61.     private static void method1(ExecutorService pool) throws InterruptedException, ExecutionException {
    62.         // new Callable<>()
    63.         Future future = pool.submit(() -> {
    64.             log.debug("running");
    65.             Thread.sleep(1000);
    66.             return "ok";
    67.         });
    68.  
    69.         log.debug("{}", future.get());
    70.     }
    71. }

    7. 关闭线程池

    7.1 shutdown

    1. /*
    2. 线程池状态变为 SHUTDOWN
    3.     - 不会接收新任务
    4.     - 但已提交任务会执行完
    5.     - 此方法不会阻塞调用线程的执行
    6. */
    7. void shutdown();

    1.     public void shutdown() {
    2.         final ReentrantLock mainLock = this.mainLock;
    3.         mainLock.lock();
    4.         try {
    5.             checkShutdownAccess();
    6.             // 修改线程池状态
    7.             advanceRunState(SHUTDOWN);
    8.             // 仅会打断空闲线程
    9.             interruptIdleWorkers();
    10.             onShutdown(); // 扩展点 ScheduledThreadPoolExecutor
    11.         } finally {
    12.             mainLock.unlock();
    13.         }
    14.         // 尝试终结(没有运行的线程可以立刻终结，如果还有运行的线程也不会等)
    15.         tryTerminate();
    16.     }

    7.2 shutdownNow 

    1. /*
    2. 线程池状态变为 STOP
    3.     - 不会接收新任务
    4.     - 会将队列中的任务返回
    5.     - 并用 interrupt 的方式中断正在执行的任务
    6. */
    7. List shutdownNow();

    1.     public List shutdownNow() {List tasks;
    2.         final ReentrantLock mainLock = this.mainLock;
    3.         mainLock.lock();
    4.         try {
    5.             checkShutdownAccess();
    6.             // 修改线程池状态
    7.             advanceRunState(STOP);
    8.             // 打断所有线程
    9.             interruptWorkers();
    10.             // 获取队列中剩余任务
    11.             tasks = drainQueue();
    12.         } finally {
    13.             mainLock.unlock();
    14.         }
    15.         // 尝试终结
    16.         tryTerminate();
    17.         return tasks;
    18.     }

    7.3 其他方法 

    1.     // 不在 RUNNING 状态的线程池，此方法就返回 true
    2.     boolean isShutdown();
    3.     // 线程池状态是否是 TERMINATED
    4.     boolean isTerminated();
    5.     // 调用 shutdown 后，由于调用线程并不会等待所有任务运行结束，因此如果它想在线程池 TERMINATED 后做些事情，可以利用此方法等待
    6.     boolean awaitTermination(long timeout, TimeUnit unit) throws InterruptedException;

    7.4 演示

    1. @Slf4j(topic = "c.TestShutDown")
    2. public class TestShutDown {
    3.  
    4.     public static void main(String[] args) throws ExecutionException, InterruptedException {
    5.         ExecutorService pool = Executors.newFixedThreadPool(2);
    6.  
    7.         Future result1 = pool.submit(() -> {
    8.             log.debug("task 1 running...");
    9.             Thread.sleep(1000);
    10.             log.debug("task 1 finish...");
    11.             return 1;
    12.         });
    13.  
    14.         Future result2 = pool.submit(() -> {
    15.             log.debug("task 2 running...");
    16.             Thread.sleep(1000);
    17.             log.debug("task 2 finish...");
    18.             return 2;
    19.         });
    20.  
    21.         Future result3 = pool.submit(() -> {
    22.             log.debug("task 3 running...");
    23.             Thread.sleep(1000);
    24.             log.debug("task 3 finish...");
    25.             return 3;
    26.         });
    27.  
    28.         log.debug("shutdown");
    29. //        pool.shutdown();
    30. //        pool.awaitTermination(3, TimeUnit.SECONDS);
    31.         List runnables = pool.shutdownNow();
    32.         log.debug("other.... {}" , runnables);
    33.     }
    34. }

    8. * 模式之 Worker Thread

    8.1. 定义

    让有限的工作线程（ Worker Thread）来轮流异步处理无限多的任务。也可以将其归类为分工模式，它的典型实现就是线程池，也体现了经典设计模式中的享元模式。
    例如，海底捞的服务员（线程），轮流处理每位客人的点餐（任务），如果为每位客人都配一名专属的服务员，那么成本就太高了（对比另一种多线程设计模式： Thread-Per-Message ）
    注意，不同任务类型应该使用不同的线程池，这样能够避免饥饿，并能提升效率
    例如，如果一个餐馆的工人既要招呼客人（任务类型 A ），又要到后厨做菜（任务类型 B）显然效率不咋地，分成服务员（线程池 A ）与厨师（线程池 B ）更为合理，当然你能想到更细致的分工

    8.2. 饥饿

    固定大小线程池会有饥饿现象
    两个工人是同一个线程池中的两个线程
    他们要做的事情是：为客人点餐和到后厨做菜，这是两个阶段的工作
    比如工人 A 处理了点餐任务，接下来它要等着 工人 B 把菜做好，然后上菜，他俩也配合的蛮好
    但现在同时来了两个客人，这个时候工人 A 和工人 B 都去处理点餐了，这时没人做饭了，饥饿 
    1. @Slf4j(topic = "c.TestDeadLock")
    2. public class TestStarvation {
    3.  
    4.     static final List MENU = Arrays.asList("地三鲜", "宫保鸡丁", "辣子鸡丁", "烤鸡翅");
    5.     static Random RANDOM = new Random();
    6.     static String cooking() {
    7.         return MENU.get(RANDOM.nextInt(MENU.size()));
    8.     }
    9.     public static void main(String[] args) {
    10.         ExecutorService waiterPool = Executors.newFixedThreadPool(1);
    11.         ExecutorService cookPool = Executors.newFixedThreadPool(1);
    12.  
    13.         waiterPool.execute(() -> {
    14.             log.debug("处理点餐...");
    15.             Future f = cookPool.submit(() -> {
    16.                 log.debug("做菜");
    17.                 return cooking();
    18.             });
    19.             try {
    20.                 log.debug("上菜: {}", f.get());
    21.             } catch (InterruptedException | ExecutionException e) {
    22.                 e.printStackTrace();
    23.             }
    24.         });
    25.         waiterPool.execute(() -> {
    26.             log.debug("处理点餐...");
    27.             Future f = cookPool.submit(() -> {
    28.                 log.debug("做菜");
    29.                 return cooking();
    30.             });
    31.             try {
    32.                 log.debug("上菜: {}", f.get());
    33.             } catch (InterruptedException | ExecutionException e) {
    34.                 e.printStackTrace();
    35.             }
    36.         });
    37.  
    38.     }
    39. }
    解决方法可以增加线程池的大小，不过不是根本解决方案，还是前面提到的， 不同的任务类型，采用不同的线程池

    8.3. 创建多少线程池合适

    过小会导致程序不能充分地利用系统资源、容易导致饥饿
    过大会导致更多的线程上下文切换，占用更多内存

    （1）CPU 密集型运算

    通常采用 cpu 核数 + 1 能够实现最优的 CPU 利用率， +1 是保证当线程由于页缺失故障（操作系统）或其它原因导致暂停时，额外的这个线程就能顶上去，保证 CPU 时钟周期不被浪费

    （2）I/O 密集型运算

    CPU 不总是处于繁忙状态，例如，当你执行业务计算时，这时候会使用 CPU 资源，但当你执行 I/O 操作时、远程RPC 调用时，包括进行数据库操作时，这时候 CPU 就闲下来了，你可以利用多线程提高它的利用率。

    9. 任务调度线程池

    在『 任务调度线程池 』功能加入之前，可以使用 java.util.Timer 来实现定时功能，Timer 的优点在于简单易用，但由于所有任务都是由同一个线程来调度，因此所有任务都是串行执行的，同一时间只能有一个任务在执行，前一个任务的延迟或异常都将会影响到之后的任务。 
    1. @Slf4j(topic = "c.TestTimer")
    2. public class TestTimer {
    3.     public static void main(String[] args) throws ExecutionException, InterruptedException {
    4.         /*ScheduledExecutorService pool = Executors.newScheduledThreadPool(1);
    5.         pool.schedule(() -> {
    6.             try {
    7.                 log.debug("task1");
    8.                 int i = 1 / 0;
    9.             } catch (Exception e) {
    10.                 log.error("error:", e);
    11.             }
    12.         }, 1, TimeUnit.SECONDS);*/
    13.  
    14.         ExecutorService pool = Executors.newFixedThreadPool(1);
    15.         pool.submit(() -> {
    16.             try {
    17.                 log.debug("task1");
    18.                 int i = 1 / 0;
    19.             } catch (Exception e) {
    20.                 log.error("error:", e);
    21.             }
    22.         });
    23.     }
    24.  
    25.     private static void method3() {
    26.         ScheduledExecutorService pool = Executors.newScheduledThreadPool(1);
    27.         log.debug("start...");
    28.         pool.scheduleAtFixedRate(() -> {
    29.             log.debug("running...");
    30.         }, 1, 1, TimeUnit.SECONDS);
    31.     }
    32.  
    33.     private static void method2(ScheduledExecutorService pool) {
    34.         pool.schedule(() -> {
    35.             log.debug("task1");
    36.             int i = 1 / 0;
    37.         }, 1, TimeUnit.SECONDS);
    38.  
    39.         pool.schedule(() -> {
    40.             log.debug("task2");
    41.         }, 1, TimeUnit.SECONDS);
    42.     }
    43.  
    44.     private static void method1() {
    45.         Timer timer = new Timer();
    46.         TimerTask task1 = new TimerTask() {
    47.             @Override
    48.             public void run() {
    49.                 log.debug("task 1");
    50.                 sleep(2);
    51.             }
    52.         };
    53.         TimerTask task2 = new TimerTask() {
    54.             @Override
    55.             public void run() {
    56.                 log.debug("task 2");
    57.             }
    58.         };
    59.  
    60.         log.debug("start...");
    61.         timer.schedule(task1, 1000);
    62.         timer.schedule(task2, 1000);
    63.     }
    64. }

    10. 正确处理执行任务异常

    10.1 方法1：主动捉异常

    10.2 方法2：使用 Future

    11. * 应用之定时任务

    1. public class TestSchedule {
    2.  
    3.     // 如何让每周四 18:00:00 定时执行任务？
    4.     public static void main(String[] args) {
    5.         //  获取当前时间
    6.         LocalDateTime now = LocalDateTime.now();
    7.         System.out.println(now);
    8.         // 获取周四时间
    9.         LocalDateTime time = now.withHour(18).withMinute(0).withSecond(0).withNano(0).with(DayOfWeek.THURSDAY);
    10.         // 如果 当前时间 > 本周周四，必须找到下周周四
    11.         if(now.compareTo(time) > 0) {
    12.             time = time.plusWeeks(1);
    13.         }
    14.         System.out.println(time);
    15.         // initailDelay 代表当前时间和周四的时间差
    16.         // period 一周的间隔时间
    17.         long initailDelay = Duration.between(now, time).toMillis();
    18.         long period = 1000 * 60 * 60 * 24 * 7;
    19.         ScheduledExecutorService pool = Executors.newScheduledThreadPool(1);
    20.         pool.scheduleAtFixedRate(() -> {
    21.             System.out.println("running...");
    22.         }, initailDelay, period, TimeUnit.MILLISECONDS);
    23.     }
    24. }

    12. Tomcat 线程池

    三、Fork/Join（P233）

    1. 概念

    Fork/Join 是 JDK 1.7 加入的新的线程池实现，它体现的是一种分治思想，适用于能够进行任务拆分的 cpu 密集型运算

    所谓的任务拆分，是将一个大任务拆分为算法上相同的小任务，直至不能拆分可以直接求解。跟递归相关的一些计算，如归并排序、斐波那契数列、都可以用分治思想进行求解

    Fork/Join 在分治的基础上加入了多线程，可以把每个任务的分解和合并交给不同的线程来完成，进一步提升了运算效率

    Fork/Join 默认会创建与 cpu 核心数大小相同的线程池

    2. 使用

     提交给 Fork/Join 线程池的任务需要继承 RecursiveTask（有返回值）或 RecursiveAction（没有返回值），例如下面定义了一个对 1~n 之间的整数求和的任务

    1. public class TestForkJoin {
    2.  
    3.     public static void main(String[] args) {
    4.         ForkJoinPool pool = new ForkJoinPool(4);
    5.         System.out.println(pool.invoke(new AddTask1(5)));
    6.     }
    7. }
    8.  
    9. @Slf4j(topic = "c.AddTask")
    10. class AddTask1 extends RecursiveTask {
    11.  
    12.     int n;
    13.  
    14.     public AddTask1(int n) {
    15.         this.n = n;
    16.     }
    17.  
    18.     @Override
    19.     public String toString() {
    20.         return "{" + n + '}';
    21.     }
    22.  
    23.     @Override
    24.     protected Integer compute() {
    25.         if (n == 1) {
    26.             log.debug("join() {}", n);
    27.             return n;
    28.         }
    29.         AddTask1 t1 = new AddTask1(n - 1);
    30.  
    31.         t1.fork();
    32.         log.debug("fork() {} + {}", n, t1);
    33.         int result = n + t1.join();
    34.         log.debug("join() {} + {} = {}", n, t1, result);
    35.         return result;
    36.     }
    37. }

    3. 任务拆分优化 

    1. public class TestForkJoin {
    2.  
    3.     public static void main(String[] args) {
    4.         ForkJoinPool pool = new ForkJoinPool(4);
    5.         System.out.println(pool.invoke(new AddTask3(1, 5)));
    6.     }
    7. }
    8.  
    9. @Slf4j(topic = "c.AddTask")
    10. class AddTask3 extends RecursiveTask {
    11.  
    12.     int begin;
    13.     int end;
    14.  
    15.     public AddTask3(int begin, int end) {
    16.         this.begin = begin;
    17.         this.end = end;
    18.     }
    19.  
    20.     @Override
    21.     public String toString() {
    22.         return "{" + begin + "," + end + '}';
    23.     }
    24.  
    25.     @Override
    26.     protected Integer compute() {
    27.         if (begin == end) {
    28.             log.debug("join() {}", begin);
    29.             return begin;
    30.         }
    31.         if (end - begin == 1) {
    32.             log.debug("join() {} + {} = {}", begin, end, end + begin);
    33.             return end + begin;
    34.         }
    35.         int mid = (end + begin) / 2;
    36.  
    37.         AddTask3 t1 = new AddTask3(begin, mid);
    38.         t1.fork();
    39.         AddTask3 t2 = new AddTask3(mid + 1, end);
    40.         t2.fork();
    41.         log.debug("fork() {} + {} = ?", t1, t2);
    42.  
    43.         int result = t1.join() + t2.join();
    44.         log.debug("join() {} + {} = {}", t1, t2, result);
    45.         return result;
    46.     }
    47. }

  • 相关阅读:
    Leetcode 刷题日记 剑指 Offer II 053. 二叉搜索树中的中序后继
    微型神经网络库MicroGrad-基于标量自动微分的类pytorch接口的深度学习框架
    卡特-2.内存和变量&3.基本数学运算
    跨境电商小知识之跨境电商物流定义以及方式讲解
    TypeScript数组的多种定义方式
    母婴店做微信小程序开发的重要性
    20221106日常记录-奇安信源代码扫描代码注入、海豚调度2.0.5-condition节点使用
    多模态大模型升级：LLaVA→LLaVA-1.5，MiniGPT4→MiniGPT5
    [Games101] Lecture 05 Rasterization 1 (Triangles)
    python dict字典如何增加若干相同重复的key键的item子项？
  • 原文地址：https://blog.csdn.net/yirenyuan/article/details/128189979