并发工具类之 Semaphore

Semaphore (信号量) 是用来控制同时访问特定资源的线程数量,它可以协调各个线程,以保证公共资源被合理使用。

Semaphore 可以用于流量控制,特别是公共资源有限的场景,比如数据库连接。假如同时有 10 个线程需要连接数据库,但数据库的连接数只有 2 个,我们必须控制只有 2 个线程能同时获取数据库连接,否则会报错无法获取数据库连接,这时就可以使用 Semaphore 来做流量控制,代码如下:

public class TestSemaphore {
    private static final int THREAD_COUNT = 10;

    private static ExecutorService threadPool = Executors.newFixedThreadPool(THREAD_COUNT);

    private static Semaphore s = new Semaphore(2);

    public static void main(String[] args) {
        for (int i = 0; i < THREAD_COUNT; i++) {
            threadPool.execute(() -> {
                try {
                    // 获取许可,未获取许可前线程就停在这里等着
                    s.acquire();
                    Thread.sleep(1000);
                    System.out.println(Thread.currentThread().getName() + " save data");
                    // 释放许可
                    s.release();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            });
        }
        threadPool.shutdown();
    }
}

控制台输出如下:

pool-1-thread-2 save data
pool-1-thread-1 save data
pool-1-thread-3 save data
pool-1-thread-4 save data
pool-1-thread-5 save data
pool-1-thread-6 save data
pool-1-thread-7 save data
pool-1-thread-8 save data
pool-1-thread-10 save data
pool-1-thread-9 save data
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值