Semaphore

Semaphore

Semaphore 是 Java 并发包中提供的一个工具类,翻译过来为“信号量”,作用是控制并发线程的数量。

类的结构

先来看一下 Semaphore 的结构:

Semaphore-Outline

Semaphore 中有三个个内部类:

  • Sync 继承了 AbstractQueuedSynchronizer,重写了 tryReleaseShared 方法,还有一些在 Semaphore 中用到的辅助方法,都是对线程进行控制的
  • NonfairSync 继承了 Sync,通过重写 tryAcquireShared 方法实现了非公平的线程竞争机制,这个方法内部是调用了 SyncnonfairTryAcquireShared 方法
  • FairSync 继承了 Sync,也是通过重写 tryAcquireShared 方法实现了公平的线程竞争机制

由此可以看出,Semaphore 实际上是通过 AQS 的共享锁来控制线程的。

构造方法

Semaphore 的构造函数对变量 sync 进行了初始化,默认是非公平竞争的,也可以通过指定参数设置为公平竞争,其他所有的方法内部都是调用了 sync 变量的方法。

public Semaphore(int permits) {
    sync = new NonfairSync(permits);
}

public Semaphore(int permits, boolean fair) {
    sync = fair ? new FairSync(permits) : new NonfairSync(permits);
}

acquire

acquire 用来获取信号量,Semaphore 提供了重载方法,也可以获取多个信号量。

public void acquire() throws InterruptedException { // 获取 1 个信号量
    sync.acquireSharedInterruptibly(1);
}
    
public void acquire(int permits) throws InterruptedException {
    if (permits < 0) throw new IllegalArgumentException(); // 参数判断
    sync.acquireSharedInterruptibly(permits); // 获取指定个数的信号量
}

acquire 方法内部调用了 syncacquireSharedInterruptibly 方法,这里并没有对这个方法进行重写,所以调用的还是 AbstractQueuedSynchronizer 中的方法,这里就不贴代码了,但是 acquireSharedInterruptibly 方法内部又调用了 tryAcquireShared 方法,由于 Semaphore 类提供了公平和非公平两种竞争机制,所以 tryAcquireShared 也有两种不同的实现。

来看一下两种获取锁的方法。

tryAcquireShared

先来看一下非公平的 tryAcquireShared,这个方法是内部类 NonfairSync 中的:

protected int tryAcquireShared(int acquires) {
    return nonfairTryAcquireShared(acquires);
}

继续看 nonfairTryAcquireShared,这个方法是 Sync 提供的:

final int nonfairTryAcquireShared(int acquires) {
    for (;;) { // 循环直到获取成功
        int available = getState(); // 获取当前 state,在这里就是信号量
        int remaining = available - acquires; // 减去获取的信号量后剩余的信号量
        // 如果信号量小于 0(获取失败) 或者更新信号量成功,返回剩余信号量
        if (remaining < 0 ||
            compareAndSetState(available, remaining))
            return remaining;
    }
}

再看公平的 tryAcquireShared,这个方法是内部类 FairSync 中的:

protected int tryAcquireShared(int acquires) {
    for (;;) {
        // 先判断在当前线程之前是否有线程正在 acquire,如果有返回 -1 表示获取失败
        if (hasQueuedPredecessors())
            return -1;
        int available = getState();
        int remaining = available - acquires;
        if (remaining < 0 ||
            compareAndSetState(available, remaining))
            return remaining;
    }
}

相对于非公平的 nonfairTryAcquireShared,公平的 tryAcquireShared 先判断在当前线程之前是否有线程正在 acquire,如果有就直接返回 -1 表示 tryAcquire 失败了,这就是公平的体现。

acquire 还提供了忽略中断的 acquireUninterruptibly,这里就不展开来说了。

release

release 方法用来释放信号量,同样的,Semaphore 提供了重载方法,可以释放多个信号量。

public void release() {
    sync.releaseShared(1);
}

public void release(int permits) {
    if (permits < 0) throw new IllegalArgumentException();
    sync.releaseShared(permits);
}

acquire 一样,release 方法调用了 AbstractQueuedSynchronizer 中的 releaseShared 方法,releaseShared 方法内部又调用了 tryReleaseShared 方法,这个方法由子类 Sync 重写:

protected final boolean tryReleaseShared(int releases) {
    for (;;) {
        int current = getState();
        int next = current + releases;
        if (next < current) // overflow 溢出
            throw new Error("Maximum permit count exceeded");
        if (compareAndSetState(current, next))
            return true;
    }
}

tryReleaseShared 的逻辑比较简单,将信号量归还,CAS 更新 state 即可。

Semaphore 还提供了 tryAcquire 方法以及一些辅助方法,这里不再赘述。

总结

Semaphore 提供了线程的控制方案,对线程的竞争提供了公平和非公平的方式。

应用

import java.util.concurrent.Semaphore;

public class SemaphoreTest {

    private static final int numOfThreads = 5; // 线程数

    private static final int sleepTime = 3000; // 睡眠时间

    public static void main(String[] args) {
        Semaphore semaphore = new Semaphore(numOfThreads);
        System.out.println("停车场一共有 " + numOfThreads + " 个停车位");
        for (int i = 0; i < 2 * numOfThreads; i++) {
            new Thread(new Runnable() {
                @Override
                public void run() {
                    try {
                        semaphore.acquire();
                        System.out.println(Thread.currentThread().getName() + " 停车");
                        Thread.sleep(sleepTime);
                        System.out.println(Thread.currentThread().getName() + " 开走了");
                        semaphore.release();
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    }
                }
            }).start();
        }
    }

}

输出结果为:

停车场一共有 5 个停车位
Thread-2 停车
Thread-0 停车
Thread-1 停车
Thread-4 停车
Thread-3 停车
Thread-0 开走了
Thread-2 开走了
Thread-8 停车
Thread-7 停车
Thread-1 开走了
Thread-4 开走了
Thread-3 开走了
Thread-9 停车
Thread-6 停车
Thread-5 停车
Thread-7 开走了
Thread-8 开走了
Thread-9 开走了
Thread-6 开走了
Thread-5 开走了

CountDownLatch、CyclicBarrier 和 Semaphore

  • CountDownLatch 由一类线程控制另一类线程,CyclicBarrier 是一类线程都执行到了 await 方法后再继续执行,Semaphore 则是控制同时执行的线程的数量
  • CountDownLatch 主要通过 await 方法和 countDown 方法控制,CyclicBarrier 只通过 await 方法,Semaphore 通过 acquire 方法和 release 方法
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 202,802评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,109评论 2 379
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 149,683评论 0 335
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,458评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,452评论 5 364
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,505评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,901评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,550评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,763评论 1 296
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,556评论 2 319
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,629评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,330评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,898评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,897评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,140评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,807评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,339评论 2 342

推荐阅读更多精彩内容