golang]使用redis实现简单的分布式锁

源码来自https://github.com/study-only/go-locks

Redis分布式锁

Redis SET 命令

从Redis 2.6.12版本开始，SET命令的行为可以通过一系列参数来修改，详见Redis命令参考：

SET key value [EX seconds] [PX milliseconds] [NX|XX]

可选参数

• EX seconds：将键的过期时间设置为seconds秒。 执行SET key value EX seconds的效果等同于执行SETEX key seconds value。
• PX milliseconds：将键的过期时间设置为milliseconds毫秒。 执行SET key value PX milliseconds的效果等同于执行 PSETEX key milliseconds value。
• NX ： 只在键不存在时， 才对键进行设置操作。 执行SET key value NX的效果等同于执行SETNX key value。
• XX ： 只在键已经存在时， 才对键进行设置操作。

返回值

在Redis 2.6.12版本以前，SET命令总是返回 OK 。
从Redis 2.6.12版本开始，SET命令只在设置操作成功完成时才返回OK； 如果命令使用了NX或者XX选项， 但是因为条件没达到而造成设置操作未执行， 那么命令将返回空批量回复（NULL Bulk Reply）

Redis分布式锁实现原理

从Redis命令看出，SET命令为原子操作，我们可以用SET key value EX seconds NX来实现分布式锁，下面为go语言实现：

import ( "errors" "time" "github.com/go-redis/redis" ) var redisClient *redis.Client type redisLock struct { name string expiry time.Duration } func (l *redisLock) TryLock() error { if ok, _ := redisClient.SetNX(l.name, 1, l.expiry).Result(); !ok { return errors.New("redis lock: already locked") } return nil } func (l *redisLock) Unlock() error { return redisClient.Del(l.name).Err() }