Redis分布式鎖
Redis SET 命令如下:
SET key value [EX seconds] [PX milliseconds] [NX|XX]
參數說明:
EX seconds:將鍵的過期時間設置為seconds秒。 執(zhí)行SET key value EX seconds的效果等同于執(zhí)行SETEX key seconds value。
PX milliseconds:將鍵的過期時間設置為milliseconds毫秒。 執(zhí)行SET key value PX milliseconds的效果等同于執(zhí)行 PSETEX key milliseconds value。
NX : 只在鍵不存在時, 才對鍵進行設置操作。 執(zhí)行SET key value NX的效果等同于執(zhí)行SETNX key value。
XX : 只在鍵已經存在時, 才對鍵進行設置操作。
返回值
在Redis 2.6.12版本以前,SET命令總是返回 OK 。
從Redis 2.6.12版本開始,SET命令只在設置操作成功完成時才返回OK; 如果命令使用了NX或者XX選項, 但是因為條件沒達到而造成設置操作未執(zhí)行, 那么命令將返回空批量回復(NULL Bulk Reply)
Redis分布式鎖實現原理
從Redis命令看出,SET命令為原子操作,我們可以用SET key value EX seconds NX來實現分布式鎖,下面為go語言實現:
下面簡單的go代碼
package main
import(
"errors"
"time"
"github.com/go-redis/redis"
)
var(
redisClient *redis.Client
)
type TryLocker interface{
TryLock() error
Unlock() error
}
type redisLock struct{
Name string //key
Expire time.Duration //過期時間
}
func newRedisLock(name string, t time.Duration) *redisLock{
return & redisLock{
Name:name,
Expire:t,
}
}
func(l *redisLock)TryLock() error{
if ok, _ := redisClient.SetNX(l.Name, 1, l.Expire).Result(); !ok {
return errors.New("redis lock: already locked")
}
return nil
}
func(l *redisLock)Unlock() error{
return redisClient.Del(l.Name).Err()
}
func main(){
l:=newRedisLock("cr",time.Second)
err:=l.TryLock()
var wg sync.WaitGroup
for i := 0; i < 100; i++ {
wg.Add(1)
if err != nil {
time.Sleep(50*time.Millisecond)
if i >= 100 {
fmt.Println("任務處理 異常:", i)
wg.Done()
return
}
}
fmt.Println("任務處理中....")
l.Unlock()
fmt.Println("任務處理 ok:", i)
wg.Done()
}
wg.Wait()
}