Semaphore源码解析

Semaphore(信号量)是控制访问线程的基于AQS的并发工具。也就是说使用Semaphore能够控制访问资源的并发量。比如在限流这块。

按照限流的逻辑,所以在使用的时候,需要进行设置最大的并发线程,按照我们之前对countDloadLatch的理解。通过设定总量,然后在条件满足之后进行下一工作。

semaphore则是通过设定最大的并发线程,在当前线程消耗的指定的掉总量之后。总量的剩余量与当初设定的总量进行对比,如果小于总量就排队,如果大于总量就进行阻塞。如果超过超时时间就剔除。

semaphore的大致逻辑就是这样。其中使用了AQS做基础的保证。

public class MySemaphore {




    public static Semaphore semaphore=new Semaphore(2);




    static class task implements Runnable{








        @Override
        public void run() {
            try {
                semaphore.acquire();
                System.out.println(Thread.currentThread().getName());
                Thread.sleep(500);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }finally {
                semaphore.release();
            }
        }
    }








    public static void main(String[] args) {
        ExecutorService executorService= Executors.newCachedThreadPool();
        for (int i=0;i<10;i++){
            executorService.submit(new task());
        }
        executorService.shutdown();




    }
}

我们可以修改Semaphore的值,然后观察结果,能够很明显的观察到这种并发控制的效果。

下面我们从源码角度看一下他是如何实现的。

从图中可以看出,Seamaphore的结构和其他的并发工具一样,都有公平锁和非公平锁,一版这种情况都是和AQS关系很大,然后看到很多之前我们看到的方法。

下面我们从初始化方法和上述demo的层面上进行深入。

    public Semaphore(int permits) {

        sync = new NonfairSync(permits);

    }

在初始化的时候,默认采用的是非公平锁。

    public Semaphore(int permits, boolean fair) {

        sync = fair ? new FairSync(permits) : new NonfairSync(permits);

    }

或者自行决定是公平还是非公平锁

    public void acquire() throws InterruptedException {

        sync.acquireSharedInterruptibly(1);

    }

当我们调用acquire()方法的时候,发现调用的是acquireSharedInterruptibly,我们跟踪一下。

doAcquireShared这个方法我们之前分析过,就是要从AQS链表中从后向前的寻找。剔除超时的,把能激活的线程向前移动,然后等待之前的额线程执行完毕之后抢夺head节点。

那么这块的意思就是说要先获取state的数量,然后看state减去申请的数量(这里是1),是否还有剩余,如果有剩余说明可以加入到队列。如果state没有剩余,那么就进行自旋,然后一直尝试往链表中添加。

按照这种逻辑,那么释放的时候,就应该是是对state的值进行相加。那么我们看看源码是如何实现的。

        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;
            }
        }

果然,Semaphore就是对state的值进行加法操作。这样保证信号量的值的统一。

好了关于Semaphore的源码解析就到这里了。

总结:

Semaphore使用AQS进行信号量的实现。通过对state值得维护来实现信号量模式,在实际应用中,我们可以采用Semaphore来做系统的限流业务。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值