原理說明
SET key value NX PX 30000
分布式鎖的核心原理是利用 redis 的 set if not exist 命令，確保只有一個命令能成功設(shè)置值囊嘉，也就是能搶到鎖。
其中解鎖的時候需要注意判斷鎖的內(nèi)容革为。因為某個 client 阻塞很長時間的情況下扭粱，可能會錯誤釋放別的 client 的鎖。
加鎖示例
func getLock(ctx context.Context, rdb *redis.Client, key string, value *int64) (ok bool, err error) {
*value = time.Now().UnixNano() / 1000000.0 // 設(shè)置為毫秒
for i := 0; i < 3; i++ { // 最多重試 3 次
ok, err = rdb.SetNX(ctx, key, *value, time.Second*10).Result()
if ok || err != nil {
break
}
}
return
}
為了增加成功率震檩，快速重試三次
常見解鎖示例
func releaseLock(ctx context.Context, rdb *redis.Client, key string, value int64) (ok bool, err error) {
res, err := rdb.Get(ctx, key).Result()
if err == nil && res == strconv.FormatInt(value, 10) {
i, err := rdb.Del(ctx, key).Result()
fmt.Println("del", i, err)
if i > 0 && err == nil {
return true, nil
}
}
return
}
需要注意的邏輯是判斷 value 的值是不是相等琢蛤，相等就解鎖。
但這樣可能會出現(xiàn)問題恳蹲，在 get 的時候是獲取的 res 確實是該鎖的虐块，但是這個時候這個 key 正好過期了，所以別的進(jìn)程就可以搶到這個鎖了嘉蕾。等該線程 del 的時候其實刪除的是新的那個鎖贺奠，所以也會誤刪除了別人的 lock。所以更好的辦法是用 lua 的腳本错忱。
使用 lua 解鎖示例
// 依賴于 redis 的腳本機制
// 可以保證腳本在執(zhí)行的時候不會有命令插進(jìn)來
func releaseLockBetter(ctx context.Context, rdb *redis.Client, key string, value int64) (ok bool, err error) {
script := `
if redis.call("get",KEYS[1]) == ARGV[1] then
return redis.call("del",KEYS[1])
else
return 0
end
`
v, err := rdb.Eval(ctx, script, []string{key},
strconv.FormatInt(value, 10)).Result()
if x, ok := v.(int64); ok && err == nil {
return x > 0, nil
}
return false, err
}
redis 的腳本可以保證腳本在執(zhí)行的時候不會有其他命令插進(jìn)來儡率，所以實現(xiàn)了更安全的解鎖挂据。
完整代碼示例
https://go.dev/play/p/iim_M2SGOQu
進(jìn)階：redis 機器故障
以上機制的安全性是基于 redis 服務(wù)可用的情況。假設(shè)發(fā)生一種情況儿普，master 宕機了崎逃，那么鎖就會丟失了。一種解決方案是采用了AOF持久化眉孩，問題會好一點个绍。但是假設(shè)出現(xiàn)斷電，還是會丟失數(shù)據(jù)浪汪。這種情況必須要配置 fsync=always巴柿，缺點是會影響性能。
進(jìn)階：Redlock
Redlock 是解決單點故障的一個方案死遭，詳細(xì)參考 官方說明文檔广恢、Redlock 的 go 語言實現(xiàn)、Martin 對 Redlock 的評論呀潭、作者回應(yīng) Martin
進(jìn)階：鎖的安全性
redis 實現(xiàn)的分布式鎖只能適用于非強制安全的情況钉迷，因為加鎖過程中可能會有各種意外的情況的發(fā)生。比如獲取鎖后钠署，突然產(chǎn)生了一次 GC糠聪，一次 STW 可能 90 秒。比如某個網(wǎng)絡(luò)請求可能持續(xù)很長時間踏幻，再次回來的時候鎖已經(jīng)被別的進(jìn)程搶掉了枷颊。針對嚴(yán)格安全的情況戳杀，可以嘗試 MySQL 實現(xiàn)分布式鎖（select for update）
參考資料
官方說明文檔该面、Redlock 的 go 語言實現(xiàn)、Martin 對 Redlock 的評論信卡、作者回應(yīng) Martin
文章創(chuàng)建時間：2022/02/20 修改時間：2022/02/20
