golang使用redis实现分布式锁

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
}