Semaphore 简单使用

文字摘抄于：https://yq.aliyun.com/articles/632589
Semaphore（信号量）是用来控制同时访问特定资源的线程数量，它通过协调各个线程，以保证合理的使用公共资源。换句话说，锁（Lock锁或synchronized锁）在任何时刻只允许一个任务访问被加锁的资源，而计数信号量允许n个任务同时访问这个资源，还可以将信号量看作是向外分发使用资源的“许可证”，尽管内部没有这种许可证对象。

“许可证”的数量是有限的，所以当执有“许可证”的线程数量与“许可证”数量相同时，就会阻止其他线程对共享资源的使用，如果某一个或多个线程使用完共享资源后，就会归还“许可证”，此时Semaphore（信号量）就会将这些归还的“许可证”再次分发给阻塞中的线程。通过这种方式就实现了控制线程并发数。

应用：
Semaphore可以用于做流量控制，特别是公用资源有限的应用场景，比如数据库连接。假如有一个需求，要读取几万个文件的数据，因为都是IO密集型任务，我们可以启动几十个线程 并发地读取，但是如果读到内存后，还需要存储到数据库中，而数据库的连接数只有10个，这 时我们必须控制只有10个线程同时获取数据库连接保存数据，否则会报错无法获取数据库连 接。这个时候，就可以使用Semaphore来做流量控制。

代码

public static void main(String[] args) throws InterruptedException {

        // 同一时间，允许最大线程数量获取
        final MySemaphore semaphore = new MySemaphore(1);

        for(int i = 0; i<2; i++ )
            new Thread(() -> {
                try {
                    // 获取信号量
                    semaphore.acquire();
                    //boolean b = semaphore.tryAcquire(); 尝试去拿，这个方法有入参
                    // semaphore.drainPermits(); // 获取所有的许可证

                    TimeUnit.SECONDS.sleep(2);

                } catch (InterruptedException e) {
                    e.printStackTrace();
                }finally {
                    // 释放
                    semaphore.release();
                }
            }).start();

        while(true){
            System.out.println("AP: " + semaphore.availablePermits()); // 当前可用的信号量
            System.out.println("QL: " + semaphore.getQueueLength()); // 阻塞的信号量

            semaphore.hasQueuedThreads(); // 是否有等待线程
            Collection<Thread> queuedThreads = semaphore.getQueuedThreads(); // 在等待的线程队列

            TimeUnit.SECONDS.sleep(1);
        }

    }

    static class MySemaphore extends Semaphore{

        public MySemaphore(int permits) {
            super(permits);
        }

        public MySemaphore(int permits, boolean fair) {
            super(permits, fair);
        }

        @Override
        protected Collection<Thread> getQueuedThreads() {
            return super.getQueuedThreads();
        }
    }