Redisson 加鎖解鎖的實(shí)現(xiàn)
分布式鎖使用
對(duì)于 redisson 分布式鎖的使用很簡(jiǎn)單:
1、調(diào)用 getLock 函數(shù)獲取鎖操作對(duì)象;
2、調(diào)用 tryLock 函數(shù)進(jìn)行加鎖;
3、調(diào)用 unlock 函數(shù)進(jìn)行解鎖;
注意 unlock 操作需要放到 finally 代碼段中,保證鎖可以被釋放。
private void sumLock() {
lock = redissonClient.getLock("sum-lock");
boolean b = lock.tryLock();
if (!b) {
log.info("獲取不到鎖");
return;
}
try {
for (int j = 0; j < 20000; j++) {
++sum;
}
} finally {
lock.unlock();
}
}getLock
getLock 實(shí)例化 RedissonLock,相當(dāng)于 Lock lock = new ReentrantLock() 操作;
public RLock getLock(String name) {
? ? // 實(shí)例化 RedissonLock,參數(shù)為指令執(zhí)行器和鎖名稱
? ? return new RedissonLock(this.connectionManager.getCommandExecutor(), name);
}
public RedissonLock(CommandAsyncExecutor commandExecutor, String name) {
? ? super(commandExecutor, name);
? ? // 命令執(zhí)行器,用于執(zhí)行l(wèi)ua腳本
? ? this.commandExecutor = commandExecutor;
? ? // 連接管理器的ID
? ? this.id = commandExecutor.getConnectionManager().getId();
? ? // 鎖續(xù)期時(shí)間(看門狗),鎖默認(rèn)續(xù)期時(shí)間是 30s。
? ? this.internalLockLeaseTime = commandExecutor.getConnectionManager().getCfg().getLockWatchdogTimeout();
? ? this.entryName = this.id + ":" + name;
? ? this.pubSub = commandExecutor.getConnectionManager().getSubscribeService().getLockPubSub();
}tryLock
@Override
public boolean tryLock() {
? ? return get(tryLockAsync());
}
@Override
public RFuture<Boolean> tryLockAsync() {
? ? return tryLockAsync(Thread.currentThread().getId());
}
@Override
public RFuture<Boolean> tryLockAsync(long threadId) {
? ? return tryAcquireOnceAsync(-1, -1, null, threadId);
}這里是一系列的調(diào)用,可以直接跳過(guò),直接進(jìn)入到 tryAcquireOnceAsync 函數(shù),看看 tryAcquireOnceAsync 函數(shù)的處理邏輯。
private RFuture<Boolean> tryAcquireOnceAsync(long waitTime, long leaseTime, TimeUnit unit, long threadId) {
? ? // 由于我們調(diào)用tryLocak沒(méi)有傳遞任何參數(shù),leaseTime默認(rèn)為-1,不走判斷
? ? if (leaseTime != -1) {
? ? ? ? return tryLockInnerAsync(waitTime, leaseTime, unit, threadId, RedisCommands.EVAL_NULL_BOOLEAN);
? ? }
? ??
? ? // 調(diào)用獲取鎖 枷鎖的主要邏輯在這里
? ? RFuture<Boolean> ttlRemainingFuture = tryLockInnerAsync(waitTime,
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? commandExecutor.getConnectionManager().getCfg().getLockWatchdogTimeout(),
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? TimeUnit.MILLISECONDS, threadId, RedisCommands.EVAL_NULL_BOOLEAN);
? ??
? ? ttlRemainingFuture.onComplete((ttlRemaining, e) -> {
? ? ? ? // 如果發(fā)生異常那么直接放回了
? ? ? ? if (e != null) {
? ? ? ? ? ? return;
? ? ? ? }
? ? ? ? // 鎖續(xù)期
? ? ? ? if (ttlRemaining) {
? ? ? ? ? ? scheduleExpirationRenewal(threadId);
? ? ? ? }
? ? });
? ? // 返回結(jié)果
? ? return ttlRemainingFuture;
}<T> RFuture<T> tryLockInnerAsync(long waitTime, long leaseTime, TimeUnit unit, long threadId, RedisStrictCommand<T> command) {
// 將時(shí)間轉(zhuǎn)化為毫秒
internalLockLeaseTime = unit.toMillis(leaseTime);
// 執(zhí)行腳本
return evalWriteAsync(getName(), LongCodec.INSTANCE, command,
"if (redis.call('exists', KEYS[1]) == 0) then " +
"redis.call('hincrby', KEYS[1], ARGV[2], 1); " +
"redis.call('pexpire', KEYS[1], ARGV[1]); " +
"return nil; " +
"end; " +
"if (redis.call('hexists', KEYS[1], ARGV[2]) == 1) then " +
"redis.call('hincrby', KEYS[1], ARGV[2], 1); " +
"redis.call('pexpire', KEYS[1], ARGV[1]); " +
"return nil; " +
"end; " +
"return redis.call('pttl', KEYS[1]);",
Collections.singletonList(getName()), internalLockLeaseTime, getLockName(threadId));
}Redisson 中存儲(chǔ)鎖的數(shù)據(jù)類型結(jié)構(gòu)采用的的是 hash,Key 為鎖名稱,VALUE的屬性是 Redisson 客戶端ID和線程ID組合而成的字符串,值是鎖的重入次數(shù),采用 hash 計(jì)數(shù)實(shí)現(xiàn)鎖的重入性。

該函數(shù)主要執(zhí)行 lua 腳本,腳本的邏輯為:
1、redis.call(‘exists’, KEYS[1]) == 0 用于判斷鎖是否存在,等于 0 說(shuō)明不存在,表明此時(shí)沒(méi)有客戶端持有鎖,此客戶端獲取鎖成功;走步驟 2,否則走步驟 4;
2、設(shè)置鎖,并且對(duì)鎖進(jìn)行 +1 操作,標(biāo)識(shí)獲取鎖的次數(shù);
3、為鎖設(shè)置過(guò)期時(shí)間,成功返回 nil;
4、redis.call(‘hexists’, KEYS[1], ARGV[2]) == 1 判斷鎖是否本客戶端持有,等于1說(shuō)明是,此時(shí)是再次獲取鎖(重入),走步驟 5,否則走 7;
5、對(duì)鎖進(jìn)行 +1 操作,標(biāo)識(shí)獲取鎖的次數(shù);
6、為鎖設(shè)置過(guò)期時(shí)間,成功返回 nil;
7、如果1和4的判斷都不滿足,那么返回鎖的的剩余時(shí)間;
unLock
@Override
public void unlock() {
? ? try {
? ? ? ? get(unlockAsync(Thread.currentThread().getId()));
? ? } catch (RedisException e) {
? ? ? ? if (e.getCause() instanceof IllegalMonitorStateException) {
? ? ? ? ? ? throw (IllegalMonitorStateException) e.getCause();
? ? ? ? } else {
? ? ? ? ? ? throw e;
? ? ? ? }
? ? }
}
@Override
public RFuture<Void> unlockAsync(long threadId) {
? ? RPromise<Void> result = new RedissonPromise<Void>();
? ? // 釋放鎖的邏輯主要這里
? ? RFuture<Boolean> future = unlockInnerAsync(threadId);
? ? future.onComplete((opStatus, e) -> {
? ? ? ? cancelExpirationRenewal(threadId);
? ? ? ? if (e != null) {
? ? ? ? ? ? result.tryFailure(e);
? ? ? ? ? ? return;
? ? ? ? }
? ? ? ? if (opStatus == null) {
? ? ? ? ? ? IllegalMonitorStateException cause = new IllegalMonitorStateException("attempt to unlock lock, not locked by current thread by node id: "
? ? ? ? ? ? ? ? ? ? + id + " thread-id: " + threadId);
? ? ? ? ? ? result.tryFailure(cause);
? ? ? ? ? ? return;
? ? ? ? }
? ? ? ? result.trySuccess(null);
? ? });
? ? return result;
}這里是一系列的調(diào)用,可以直接跳過(guò),直接進(jìn)入到 unlockInnerAsync 函數(shù),看看 unlockInnerAsync 函數(shù)的處理邏輯。
protected RFuture<Boolean> unlockInnerAsync(long threadId) {
return evalWriteAsync(getName(), LongCodec.INSTANCE, RedisCommands.EVAL_BOOLEAN,
"if (redis.call('hexists', KEYS[1], ARGV[3]) == 0) then " +
"return nil;" +
"end; " +
"local counter = redis.call('hincrby', KEYS[1], ARGV[3], -1); " +
"if (counter > 0) then " +
"redis.call('pexpire', KEYS[1], ARGV[2]); " +
"return 0; " +
"else " +
"redis.call('del', KEYS[1]); " +
"redis.call('publish', KEYS[2], ARGV[1]); " +
"return 1; " +
"end; " +
"return nil;",
Arrays.asList(getName(), getChannelName()), LockPubSub.UNLOCK_MESSAGE, internalLockLeaseTime, getLockName(threadId));
}
該函數(shù)主要執(zhí)行 lua 腳本,腳本的邏輯為:
1、redis.call(‘hexists’, KEYS[1], ARGV[3]) == 0 用于判斷鎖是否為當(dāng)前客戶端持有,等于 0 說(shuō)明不是直接返回 nil,否則說(shuō)明是,走步驟 2;
2、對(duì)鎖進(jìn)行 -1 操作,并且獲取其計(jì)數(shù) counter;
3、判斷 counter > 0,如果大于 0 說(shuō)明該客戶端多次獲取鎖,對(duì)鎖進(jìn)行續(xù)期并且返回 0,因?yàn)榇藭r(shí)業(yè)務(wù)還沒(méi)有執(zhí)行完畢,否則走步驟 4;
4、如果count 小于等于 0 則刪除鎖,發(fā)送釋放鎖的消息,返回 1;
5、如果以上邏輯都不滿足,那么直接返回nil;
總結(jié)
redisson 加鎖解鎖:
1、redisson 使用 lua 腳本保證命令執(zhí)行的原子性;
2、redisson 使用 redis 的 hash 數(shù)據(jù)結(jié)構(gòu)類型來(lái)存儲(chǔ)鎖信息,使用 鎖名稱作為 hash 名稱,使用“客戶端ID:線程ID”作為鍵,使用重入次數(shù)作為值;
3、每次獲取鎖,會(huì)對(duì)值進(jìn)行 +1 操作,并且設(shè)置過(guò)期時(shí)間;
4、每次釋放鎖,會(huì)對(duì)值進(jìn)行 -1 操作,如果沒(méi)有減少為 0,則繼續(xù)設(shè)置鎖的超時(shí)時(shí)間,否則刪除鎖,并且發(fā)送釋放鎖的消息。
到此這篇關(guān)于Redisson 加鎖解鎖的實(shí)現(xiàn)的文章就介紹到這了,更多相關(guān)Redisson 加鎖解鎖內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
Redis延遲隊(duì)列和分布式延遲隊(duì)列的簡(jiǎn)答實(shí)現(xiàn)
在我們的工作中,很多地方使用延遲隊(duì)列,比如訂單到期沒(méi)有付款取消訂單,制訂一個(gè)提醒的任務(wù)等都需要延遲隊(duì)列,那么我們需要實(shí)現(xiàn)延遲隊(duì)列,本文就來(lái)介紹一下如何實(shí)現(xiàn),感興趣的可以了解一下2021-05-05
設(shè)置Redis最大占用內(nèi)存的實(shí)現(xiàn)
本文主要介紹了設(shè)置Redis最大占用內(nèi)存的實(shí)現(xiàn),文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2022-05-05
Redis?生成分布式業(yè)務(wù)單號(hào)的實(shí)現(xiàn)
在業(yè)務(wù)系統(tǒng)中很多場(chǎng)景下需要生成不重復(fù)的ID,本文主要介紹了Redis生成分布式業(yè)務(wù)單號(hào)的實(shí)現(xiàn),具有一定的參考價(jià)值,感興趣的可以了解一下2024-04-04
關(guān)于分布式鎖的三種實(shí)現(xiàn)方式
這篇文章主要介紹了關(guān)于分布式鎖的三種實(shí)現(xiàn)方式,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2022-08-08

