关于golang:用-Go-Redis-实现分布式锁

为什么须要分布式锁

1. 用户下单

锁住 uid，避免反复下单。

2. 库存扣减

锁住库存，避免超卖。

3. 余额扣减

锁住账户，避免并发操作。
分布式系统中共享同一个资源时往往须要分布式锁来保障变更资源一致性。

分布式锁须要具备个性

1. 排他性

锁的根本个性，并且只能被第一个持有者持有。

2. 防死锁

高并发场景下临界资源一旦产生死锁十分难以排查，通常能够通过设置超时工夫到期主动开释锁来躲避。

3. 可重入

锁持有者反对可重入，避免锁持有者再次重入时锁被超时开释。

4. 高性能高可用

锁是代码运行的要害前置节点，一旦不可用则业务间接就报故障了。高并发场景下，高性能高可用是根本要求。

实现 Redis 锁应先把握哪些知识点

1. set 命令

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

• EX second：设置键的过期工夫为 second 秒。SET key value EX second 成果等同于 SETEX key second value。
• PX millisecond：设置键的过期工夫为 millisecond 毫秒。SET key value PX millisecond 成果等同于 PSETEX key millisecond value。
• NX ：只在键不存在时，才对键进行设置操作。SET key value NX 成果等同于 SETNX key value。
• XX ：只在键曾经存在时，才对键进行设置操作。

2. Redis.lua 脚本

应用 redis lua 脚本能将一系列命令操作封装成 pipline 实现整体操作的原子性。

go-zero 分布式锁 RedisLock 源码剖析

core/stores/redis/redislock.go

1. 加锁流程

-- KEYS[1]: 锁 key
-- ARGV[1]: 锁 value, 随机字符串
-- ARGV[2]: 过期工夫
-- 判断锁 key 持有的 value 是否等于传入的 value
-- 如果相等阐明是再次获取锁并更新获取工夫，避免重入时过期
-- 这里阐明是“可重入锁”if redis.call("GET", KEYS[1]) == ARGV[1] then
    -- 设置
    redis.call("SET", KEYS[1], ARGV[1], "PX", ARGV[2])
    return "OK"

else
    -- 锁 key.value 不等于传入的 value 则阐明是第一次获取锁
    -- SET key value NX PX timeout : 当 key 不存在时才设置 key 的值
    -- 设置胜利会主动返回“OK”，设置失败返回“NULL Bulk Reply”-- 为什么这里要加“NX”呢，因为须要避免把他人的锁给笼罩了
    return redis.call("SET", KEYS[1], ARGV[1], "NX", "PX", ARGV[2])
end

2. 解锁流程

-- 开释锁
-- 不能够开释他人的锁
if redis.call("GET", KEYS[1]) == ARGV[1] then
    -- 执行胜利返回“1”return redis.call("DEL", KEYS[1])
else
    return 0
end

3. 源码解析

package redis

import (
    "math/rand"
    "strconv"
    "sync/atomic"
    "time"

    red "github.com/go-redis/redis"
    "github.com/tal-tech/go-zero/core/logx"
)

const (
    letters     = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
    lockCommand = `if redis.call("GET", KEYS[1]) == ARGV[1] then
    redis.call("SET", KEYS[1], ARGV[1], "PX", ARGV[2])
    return "OK"
else
    return redis.call("SET", KEYS[1], ARGV[1], "NX", "PX", ARGV[2])
end`
    delCommand = `if redis.call("GET", KEYS[1]) == ARGV[1] then
    return redis.call("DEL", KEYS[1])
else
    return 0
end`
    randomLen = 16
    // 默认超时工夫，避免死锁
    tolerance       = 500 // milliseconds
    millisPerSecond = 1000
)

// A RedisLock is a redis lock.
type RedisLock struct {
    // redis 客户端
    store *Redis
    // 超时工夫
    seconds uint32
    // 锁 key
    key string
    // 锁 value，避免锁被他人获取到
    id string
}

func init() {rand.Seed(time.Now().UnixNano())
}

// NewRedisLock returns a RedisLock.
func NewRedisLock(store *Redis, key string) *RedisLock {
    return &RedisLock{
        store: store,
        key:   key,
        // 获取锁时，锁的值通过随机字符串生成
        // 实际上 go-zero 提供更加高效的随机字符串生成形式
        // 见 core/stringx/random.go：Randn
        id:    randomStr(randomLen),
    }
}

// Acquire acquires the lock.
// 加锁
func (rl *RedisLock) Acquire() (bool, error) {
    // 获取过期工夫
    seconds := atomic.LoadUint32(&rl.seconds)
    // 默认锁过期工夫为 500ms，避免死锁
    resp, err := rl.store.Eval(lockCommand, []string{rl.key}, []string{rl.id, strconv.Itoa(int(seconds)*millisPerSecond + tolerance),
    })
    if err == red.Nil {return false, nil} else if err != nil {logx.Errorf("Error on acquiring lock for %s, %s", rl.key, err.Error())
        return false, err
    } else if resp == nil {return false, nil}

    reply, ok := resp.(string)
    if ok && reply == "OK" {return true, nil}

    logx.Errorf("Unknown reply when acquiring lock for %s: %v", rl.key, resp)
    return false, nil
}

// Release releases the lock.
// 开释锁
func (rl *RedisLock) Release() (bool, error) {resp, err := rl.store.Eval(delCommand, []string{rl.key}, []string{rl.id})
    if err != nil {return false, err}

    reply, ok := resp.(int64)
    if !ok {return false, nil}

    return reply == 1, nil
}

// SetExpire sets the expire.
// 须要留神的是须要在 Acquire()之前调用
// 不然默认为 500ms 主动开释
func (rl *RedisLock) SetExpire(seconds int) {atomic.StoreUint32(&rl.seconds, uint32(seconds))
}

func randomStr(n int) string {b := make([]byte, n)
    for i := range b {b[i] = letters[rand.Intn(len(letters))]
    }
    return string(b)
}

对于分布式锁还有哪些实现计划

1. etcd
2. redis redlock

我的项目地址

https://github.com/zeromicro/go-zero

欢送应用 go-zero 并 star 反对咱们！

微信交换群

关注『微服务实际 』公众号并点击 交换群 获取社区群二维码。