Semaphore也是redis分布式锁支持的一种,同步组件
之前给大家的讲解的锁,基本上都是同时间只能一个客户端获取这个锁,然后做一些事情,处理完了以后释放锁
Semaphore,信号量,他作为一个锁机制,可以允许多个线程同时获取一把锁,任何一个线程释放锁之后,其他等待的一个线程就可以尝试进来获取一下这个锁
使用
RSemaphore semaphore = redisson.getSemaphore("semaphore");
//设置允许多少客户端获取这个锁。
//只有第一次set有效,如果设置成功的话会返回3,如果已经被设置的话,返回0
//修改的话需要用addPermits方法
semaphore.trySetPermits(3);
for(int i = 0; i < 10; i++) {
new Thread(new Runnable() {
@Override
public void run() {
try {
System.out.println(new Date() + ":线程[" + Thread.currentThread().getName() + "]尝试获取Semaphore锁");
semaphore.acquire();
System.out.println(new Date() + ":线程[" + Thread.currentThread().getName() + "]成功获取到了Semaphore锁,开始工作");
Thread.sleep(3000);
semaphore.release();
System.out.println(new Date() + ":线程[" + Thread.currentThread().getName() + "]释放Semaphore锁");
} catch (Exception e) {
e.printStackTrace();
}
}
}).start();
}
源码
设置锁的数量
semaphore.trySetPermits(3)
老规矩 先看lua脚本
还是先看这个lua脚本
KEYS[1] = semaphore (这是我们自定义的锁名称)
KEYS[2] = redisson_sc:{semaphore}
ARGV[1] = 3 (这是我们传的值)
先判断这个key是否存在
redis.call(‘get’, KEYS[1]);
不存在的话,将这个对应的值设置为3
redis.call(‘set’, KEYS[1], ARGV[1]);
这是广播发布数据,不用管
redis.call(‘publish’, KEYS[2], ARGV[1]);
那这个逻辑就是去设置这个锁的值,谁先设置,谁就成功,后来设置的不会生效。
获取锁
semaphore.acquire()
获取锁失败的话,会一直在这里重复获取直到成功
这很明显啊
获取当前值
redis.call(‘get’, KEYS[1]);
当前值不等于空,或者是当前值大于设置的值,才能进入下面的逻辑
if (value ~= false and tonumber(value) >= tonumber(ARGV[1])) then
decrby 命令是减去制定的值,也就是说每次进来都会把”semaphore“ 原本的值 -1,那每个线程都-1,超过3个线程的话,不就减成负数了么,都不能再进这个逻辑了。
redis.call(‘decrby’, KEYS[1], ARGV[1]);
释放锁
增量更新,每次释放就+1,并且公布广播