思路步骤
- 获取锁, 使用 SETNX 命令设置一个key.
- 如果没获取到,从新拿锁,返回步骤1 ; 从新拿锁可以设置等待时间;也可以记录拿锁次数为了做“避免死循环”.
- 如果获取到,使用 EXPIRE 给锁加存活时间;
- 接步骤3,执行业务,执行完释放锁, 使用 DEL 命令
代码实现
package main
import (
"fmt"
"github.com/garyburd/redigo/redis"
gt "github.com/mangenotwork/gathertool"
"log"
"time"
)
var red = gt.NewRedis("192.168.0.197", "6379", "love0021$%", 14)
func main(){
_ = red.RedisPool() // 初始化redis连接池
for i:=0; i<20; i++ {
go Doing(i)
}
select {}
}
func Doing(i int){
log.Println(fmt.Sprintf("[%d] 开始Doing... ", i))
key := "lock"
rd := red.GetConn()
lockTime := 0 // 获取锁次数
Lo:
if lockTime > 999 { //避免死循环
log.Println("避免死循环")
return
}
// 获取锁
is, err := redis.Bool(rd.Do("SETNX", redis.Args{}.Add(key).AddFlat(1)...))
if err != nil {
panic(err)
}
//没有拿到锁
if !is {
//log.Println("没拿到锁")
lockTime++
time.Sleep(1*time.Second)
goto Lo
}
// 给锁添加时间
_,err = rd.Do("EXPIRE", redis.Args{}.Add(key).Add(5)...)
if err != nil {
log.Println("给锁添加时间 err = ", err)
}
// 执行
log.Println(fmt.Sprintf("[%d]执行", i))
time.Sleep(1*time.Second)
// 释放锁
log.Println(fmt.Sprintf("[%d]释放锁", i))
_,err = rd.Do("DEL", key)
if err != nil {
log.Println("释放锁 err = ", err)
}
}
运行结果
一些问题探讨
如果突然服务器宕机,那么必然造成锁无法释放,即造成死锁?
解决方案:设置超时时间。
加锁和设置超时时间中间引起服务器宕机,则一样会导致死锁。
解决方案:原子性操作,即同时加锁和设置超时时间;
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-14Tju5bW-1670470670608)(http://mange1.oss-cn-beijing.aliyuncs.com/test/a8ea5c06b619cb643a891497b15aee86.png#pic_center)]
总结
设计实现分布式锁需要满足一下条件:
- 互斥性;在任意时刻,只有一个客户端能持有锁。
- 不能发生死锁;即使存在一个线程持有锁的期间崩溃而没有主动解锁,也能保证后续其他线程能加锁。
- 加锁和解锁必须是同一个线程。