通知短信+运营短信,5秒速达,支持群发助手一键发送🚀高效触达和通知客户 广告
[TOC] ## 概述 网上有很多golang操作redis的例子,我使用`github.com/gomodule/redigo/redis`包写一个关于redis分布式锁的问题。 ## redis分布式锁 锁需要的几个组件 * 获取锁 * 删除锁 分布式锁还需要 * 给锁加上唯一id (只能获取和删除自己的锁) * 给锁加上过期时间 (防止死锁) ## golang代码 推荐使用redis的连接池 ```golang pool := redis.Pool{ MaxIdle: 8, MaxActive: 0, IdleTimeout: 100, Dial: func() (redis.Conn, error) { return redis.Dial("tcp", "127.0.0.1:6100") }, } ``` 通过lua脚本保证`获取锁/加过期时间`和`获取锁的值/删除锁`为原子操作 ```golang var ( updateLockExpireUidScript = redis.NewScript(1, ` local res = redis.call("SETNX", KEYS[1], ARGV[1]) if res == 1 then return redis.call("EXPIRE", KEYS[1], ARGV[2]) end return res `) deleteLockByUidScript = redis.NewScript(1, ` local res = redis.call("GET", KEYS[1]) if res == ARGV[1] then return redis.call("DEL", KEYS[1]) end return res `) ) ``` golang完整代码 ```golang package main import ( "fmt" "github.com/gomodule/redigo/redis" "time" ) func getLock(conn redis.Conn, key []string, uid int, expire int) bool { lock := false for !lock { res, err := updateLockExpireUidScript.Do(conn, key, uid, expire) if err != nil { fmt.Println(err.Error()) break } if res.(int64) == 1 { lock = true fmt.Println("获取锁成功") return true } } fmt.Println("获取锁失败") return false } func delLock(conn redis.Conn, key []string, uid int) bool { _, err := deleteLockByUidScript.Do(conn, key, uid) if err != nil { fmt.Println("删除锁失败") fmt.Println(err.Error()) return false } fmt.Println("删除锁成功") return true } var ( updateLockExpireUidScript = redis.NewScript(1, ` local res = redis.call("SETNX", KEYS[1], ARGV[1]) if res == 1 then return redis.call("EXPIRE", KEYS[1], ARGV[2]) end return res `) deleteLockByUidScript = redis.NewScript(1, ` local res = redis.call("GET", KEYS[1]) if res == ARGV[1] then return redis.call("DEL", KEYS[1]) end return res `) ) func main() { pool := redis.Pool{ MaxIdle: 8, MaxActive: 0, IdleTimeout: 100, Dial: func() (redis.Conn, error) { return redis.Dial("tcp", "127.0.0.1:6100") }, } key := []string{"distribute"} uid := 1 expire := 20 fmt.Println(time.Now()) for i:= 1; i< 100; i++ { go func() { conn := pool.Get() if getLock(conn, key, uid, expire) { fmt.Println("开始执行逻辑...") fmt.Println("逻辑结束,删除锁") delLock(conn, key, uid) } }() } fmt.Println(time.Now()) time.Sleep(time.Second * 20) } ``` ## LUA **Lua脚本的限制** * Redis不提供引入额外的包,例如os等,只有redis这一个包可用。 * Lua脚本将会在一个函数中运行,所有变量必须使用local声明 * return返回多个值时,Redis将会只给你第一个 **脚本中的类型限制** * 脚本返回nil时,Go中得到的是`err = redis.Nil`(与Get找不到值相同) * 脚本返回false时,Go中得到的是nil,脚本返回true时,Go中得到的是int64类型的1 * 脚本返回{"ok": ...}时,Go中得到的是redis的status类型(true/false) * 脚本返回{"err": ...}时,Go中得到的是err值,也可以通过`return redis.error_reply("My Error")`达成 * 脚本返回number类型时,Go中得到的是int64类型 * 传入脚本的KEYS/ARGV中的值一律为string类型,要转换为数字类型应当使用to\_number **如果脚本运行了很久会发生什么?** Lua脚本运行期间,为了避免被其他操作污染数据,这期间将不能执行其它命令,一直等到执行完毕才可以继续执行其它请求。当Lua脚本执行时间超过了lua-time-limit时,其他请求将会收到Busy错误,除非这些请求是SCRIPT KILL(杀掉脚本)或者SHUTDOWN NOSAVE(不保存结果直接关闭Redis) 更多内容参考以下地址,我这里主要是根据使用Go的经验提供一些总结。[https://redis.io/commands/eval](https://redis.io/commands/eval) 一段更“复杂”的脚本,它要求在获取一个key值时,如果该值访问较多,就延长生存周期。此外还要比较更新时间,如果不需要更新,则直接返回取到的值,否则返回redis.Nil ck: https://cloud.tencent.com/developer/article/1830969