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
}