redis可以使用string数据结构来实现分布式锁。这里记录一些注意点:
- setnx:利用setnx的特性,当key不存在时,可以设置成功;当key存在时,setnx返回nil。
- 设置expire:为了防止客户端获取到锁后忘记解锁,这里设置了锁的过期时间。并且由redis来保证setnx和expire是原子操作,否则可能出现在setnx后可能未设置expire的情况。
- 超时问题:由于设置了expire时间,当进程获取到锁的时间太长以至于超过在redis中设置的expire时间时,就不能保证互斥性。这里要仔细对比expire时间和处理时间,防止出现这种情况。
package redis
import (
"github.com/garyburd/redigo/redis"
)
const (
redisLockTimeout = 10 // 10 seconds
)
func Lock(key string) (isLock bool, err error) {
//从连接池中娶一个con链接,pool可以自己定义。
con := pool.Get()
defer con.Close()
//这里需要redis.String包一下,才能返回redis.ErrNil
_, err = redis.String(con.Do("set", key, 1, "ex", redisLockTimeout, "nx"))
if err != nil {
if err == redis.ErrNil {
err = nil
return
}
return
}
isLock = true
return
}
func Unlock(key string) (err error) {
con := pool.Get()
defer con.Close()
_, err = con.Do("del", key)
if err != nil {
return
}
return
}