Redisson 分散式鎖原始碼 11:Semaphore 和 CountDownLatch

程式設計師小航 發表於 2021-07-15
Redis

前言

Redisson 除了提供了分散式鎖之外,還額外提供了同步元件,Semaphore 和 CountDownLatch。

Semaphore

Redisson 分散式鎖原始碼 11:Semaphore 和 CountDownLatch

意思就是在分散式場景下,只有 3 個憑證,也就意味著同時只會有三個執行緒執行業務。

設定憑證

Redisson 分散式鎖原始碼 11:Semaphore 和 CountDownLatch

引數列表:

  1. KEYS[1]:指定的 key 這裡叫 semaphore
  2. KEYS[2]:redisson_sc:{semaphore}
  3. ARGV[1]:憑證數 3

這塊 lua 指令碼相對簡單,直接設定一個 key 的 value 為 3。

Redisson 分散式鎖原始碼 11:Semaphore 和 CountDownLatch

獲取憑證

Redisson 分散式鎖原始碼 11:Semaphore 和 CountDownLatch

引數列表:

  1. KEYS[1]:指定的 key 這裡叫 semaphore
  2. ARGV[1]:要獲取的憑證數,預設 1

這段 lua 指令碼就是:

  1. 獲取 key semaphore 的值
  2. 如果值大於等於 1(要獲取的憑證數),對值進行遞減
  3. 成功返回 1,失敗返回 0

Redisson 分散式鎖原始碼 11:Semaphore 和 CountDownLatch

如果成功獲取憑證,直接返回,沒有獲取到憑證,則自旋等待。

釋放憑證

Redisson 分散式鎖原始碼 11:Semaphore 和 CountDownLatch

釋放憑證直接對 Redis key 的值進行自增即可。

CountDownLatch

Redisson 分散式鎖原始碼 11:Semaphore 和 CountDownLatch

輸出內容如下:

Redisson 分散式鎖原始碼 11:Semaphore 和 CountDownLatch

原始碼分析

設定門閂數量

Redisson 分散式鎖原始碼 11:Semaphore 和 CountDownLatch

Redisson 分散式鎖原始碼 11:Semaphore 和 CountDownLatch

這塊都大同小異。

減少門閂數量

Redisson 分散式鎖原始碼 11:Semaphore 和 CountDownLatch

就是對這個值進行遞減。

latch.await() 就是自旋獲取門閂的值,直到為 0 了,然後繼續往下執行。

總結

Redisson 實現分散式的 Semaphore 和 CountDownLatch 基本內容就是如此,僅僅通過在 Redis 維護一個公共的變數,然後進行原子更新即可。

相關推薦

相關文章