<svg width="80" height="80" viewBox="0 0 250 250" style="fill: #222; color: #fff; position: absolute; top: 0; border: 0; right: 0;" aria-hidden="true"><path d="M0,0 L115,115 L130,115 L142,142 L250,250 L250,0 Z"></path><path d="M128.3,109.0 C113.8,99.7 119.0,89.6 119.0,89.6 C122.0,82.7 120.5,78.6 120.5,78.6 C119.2,72.0 123.4,76.3 123.4,76.3 C127.3,80.9 125.5,87.3 125.5,87.3 C122.9,97.6 130.6,101.9 134.4,103.2" fill="currentColor" style="transform-origin: 130px 106px;" class="octo-arm"></path><path d="M115.0,115.0 C114.9,115.1 118.7,116.5 119.8,115.4 L133.7,101.6 C136.9,99.2 139.9,98.4 142.2,98.6 C133.8,88.0 127.5,74.4 143.8,58.0 C148.5,53.4 154.0,51.2 159.7,51.0 C160.3,49.4 163.2,43.6 171.4,40.1 C171.4,40.1 176.1,42.5 178.8,56.2 C183.1,58.6 187.2,61.8 190.9,65.4 C194.5,69.0 197.7,73.2 200.1,77.6 C213.8,80.2 216.3,84.9 216.3,84.9 C212.7,93.1 206.9,96.0 205.4,96.6 C205.1,102.4 203.0,107.8 198.3,112.5 C181.9,128.9 168.3,122.5 157.7,114.1 C157.9,116.9 156.7,120.9 152.7,124.9 L141.0,136.5 C139.8,137.7 141.6,141.9 141.8,141.8 Z" fill="currentColor" class="octo-body"></path></svg>

Redis分布式锁

基于redis的分布式锁可以基于redis的set命令实现:

  • NX 当不存在时创建
  • EX 设置 n 秒後过期

删除操作需要借助eval命令的原子性 eval命令文档

if redis.call(‘get’, KEYS[1]) == ARGV[1] then return redis.call(‘del’, KEYS[1]) else return 0 end

通过对这段lua脚本求值,防止过期时误解锁

以下是使用golang的实现形式,java可借助jedis,实现原理相同

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

package lock

import (
"github.com/garyburd/redigo/redis"
"errors"
"github.com/satori/go.uuid"
"crypto/md5"
"fmt"
)

const (
PREFIX = "lock->"
script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end"

)
type RedisLock struct {
pool *redis.Pool
expireTime int
key string
value string
}

func NewLock(name string, pool *redis.Pool, expireSeconds int) (*RedisLock, error) {
if name == "" {
return nil, errors.New("illegal argument: name")
}
key := PREFIX + name
valueContent := uuid.NewV1().String() + key
md5sum := md5.New()
md5sum.Write([]byte(valueContent))

lock := RedisLock{
key: key,
value: fmt.Sprintf("%x", md5sum.Sum(nil)),
pool: pool,
expireTime:expireSeconds}
return &lock, nil
}



func (redisLock *RedisLock) TryLock() (bool, error) {
dbConn := redisLock.pool.Get()
rep, err := dbConn.Do("SET",
redisLock.key, redisLock.value, "NX", "EX", redisLock.ExpireTime())
if err != nil {
return false, err
}

if rep == nil {
return false, nil
}
result, err := redis.String(rep, err)
if result == "OK" {
return true, nil
}
return false, err
}

func (redisLock *RedisLock) Unlock() error {
dbConn := redisLock.pool.Get()
_, err := dbConn.Do("EVAL", script, 1, redisLock.key, redisLock.value)
return err
}

func (redisLock *RedisLock) ExpireTime() int {
if redisLock.expireTime == 0 {
// default one day
return 60 * 60 * 24
}
return redisLock.expireTime
}

func (redisLock *RedisLock) SetExpireTime(expireTime int) {
redisLock.expireTime = expireTime
}