Semaphore实战

简介

Semaphore信号量计数器。和CountDownLatch，CyclicBarrier类似，是多线程协作的工具类，相对于join，wait，notify方法使用起来简单高效。下面我们主要看看它的用法吧！

实战

• 限流。限制线程的并发数。

比如在一个系统中同时只能保证5个用户同时在线。

import java.util.concurrent.Semaphore;

/**
 * @author ：jiaolian
 * @date ：Created in 2021-03-04 11:13
 * @description：Semaphore限流
 * @modified By：
 * 公众号:叫练
 */
public class LimitCurrnet {
    public static void main(String[] args) throws InterruptedException {
        //定义20个线程，每次最多只能执行5个线程;
        Semaphore semaphore = new Semaphore(5);
        for (int i=0; i<20; i++) {
            new Thread(()->{
                try {
                    //获取凭证
                    semaphore.acquire();
                    System.out.println(Thread.currentThread().getName()+"登录成功");
                    Thread.sleep(2000);
                    //释放凭证
                    semaphore.release();
                    System.out.println(Thread.currentThread().getName()+"用户退出");
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }).start();
        }
    }
}

如上代码所示：我们定义了20个用户同时访问系统，Semaphore参数是5，表示同时只能有5个用户可以获取凭证，其他用户必须等待直到有在线用户退出。调用semaphore.acquire()表示获取凭证，此时凭证数会减一，调用semaphore.release()表示释放凭证，凭证数会加一，如果系统中有等待的用户，操作此方法会通知等待的一个用户获取凭证成功，执行登录操作。最后打印部分结果如下：证明系统最多能保持5个用户同时在线。

注意：上面举出的这个案例，出个思考题：线程池是否可以实现呢？

• 模拟CyclicBarrier，CountDownLatch重用！

Semaphore可以轻松实现CountDownLatch计数器，CyclicBarrier回环屏障，还记得CountDownLatch用法么？它是个计数器，可以帮我们统计线程执行时间，常用来测试多线程高并发执行接口效率，我们下面用Semaphore模拟多线程主线程等待子线程执行完毕再返回。

import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Semaphore;
/**
 * @author ：jiaolian
 * @date ：Created in 2021-03-01 21:04
 * @description：信号量测试
 * @modified By：
 * 公众号:叫练
 */
public class SemaphoreTest {

    //定义线程数量;
    private static final int THREAD_COUNT = 2;
    //初始化信号量为0，默认是非公平锁
    private static Semaphore semaphore = new Semaphore(0,false);
    private static ExecutorService executorService = Executors.newFixedThreadPool(THREAD_COUNT);

    public static void main(String[] args) throws InterruptedException {
        for (int i=0; i<THREAD_COUNT; i++) {
            executorService.submit(()->{
                try {
                    Thread.sleep(2000);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                System.out.println(Thread.currentThread().getName()+"执行");
                semaphore.release();
            });
        }
        //获取2个信号量
        semaphore.acquire(2);
        System.out.println("主线程执行完毕");
        executorService.shutdown();
    }
}

如上代码所示：我们定义了Semaphore初始化信号量为0，默认是非公平锁，在主线程中用线程池提交2个线程，主线程调用semaphore.acquire(2)表示需要获取两个信号量，但此时初始化信号量为0，此时AQS中的state会是0-2=-2，state值小于0，所以主线程执行这句话会阻塞将其加入AQS同步队列，线程池两个线程等待2秒后会调用semaphore.release()释放2个信号量，此时AQS中的state会自增到0，会通知主线程退出等待继续往下执行。执行结果如下图所示。

有没有发现Semaphore用法可以模拟CountDownLatch，另外Semaphore通过调用acquire，release方法，还可以实现CyclicBarrier功能！我们不举例了。

实现原理

• 相同点：本质上都是计数器，底层是依赖AQS操作state实现。
• 异同点：CountDownLatch是共享锁实现，CyclicBarrier是独占锁实现，CountDownLatch通过调用countDown递减计数器只能使用一次，而CyclicBarrier通过调用await递减计数器可以达到“回环”重复的效果。Semaphore也是共享锁实现，通过调用release计数器是递增的，通过设置信号量可以实现CyclicBarrier，CountDownLatch功能。

总结

今天我们介绍了Semaphore，整理出来希望能对你有帮助，写的比不全，同时还有许多需要修正的地方，希望亲们加以指正和点评，喜欢的请点赞加关注哦。点关注，不迷路，我是【叫练】公众号，微信号【jiaolian123abc】边叫边练。