今天和大家分享的是:在开发服务端API时候,如何合理的运用线程池+CountDownLatch来保证API的高并发访问。
首先,作为Java开发的同学来说,java.util.concurrent并发包一定不会陌生,多多少少也会接触或使用过。今天的主角就是
java.util.concurrent.ThreadPoolExecutor和
java.util.concurrent.CountDownLatch。本文不详细赘述线程池的原理和数据结构,只是先普及下入门知识,然后再看如何正确的把这两种技术结合起来使用,来达到理想的效果。idea中直接command+N(mac下)输入这两个类即可打开JDK的源码:
CounDownLatch部分源码
CountDownLatch看起来很简单,就几个方法:
- public CountDownLatch(int count):用给定的计数初始化一个CountDownLatch;
- public void await() throws InterruptedException:等待所有线程执行完毕,即计数器值减到0或者当前线程被中断;
- public boolean await(long timeout, TimeUnit unit) throws InterruptedException:等待所有线程执行完毕,即计数器值减到0或者当前线程被中断或者超时了;
- public void countDown():计数器值-1,当latch值为0时候恢复所有等待的线程。
ThreadPoolExecutor类图
上图是JDK自带的ThreadPoolExecutor类结构图,其实也很简单,大家可以自行打开看一下源码。这里我们直接说JDK自带的几种线程池:
- newFixedThreadPool:固定大小的线程池,队列无限大(使用不当会把内存吃完)
- newSingleThr