而针对库存的问题较为麻烦一点,需要使用Lua编辑脚本,但是你无需在自己的机器上下载lua的编译环境,go提供了其相关的支持。针对这一部分,不用慌张,其基本架构如下:
面对并发的情况下会出现超卖的情况,redis数据库中会出现负值的情况。即使你在操作之前进行了数据的判断。
func MsCode(uuid, prodid string) bool { // 1、对uuid和prodid进行非空判断 if uuid == "" || prodid == "" { return false } //2、获取连接 rdb := DB //3、拼接key kcKey := "kc:" + prodid + ":qt" userKey := "sk:" + prodid + ":user" //4、获取库存 str, err := rdb.Get(ctx, kcKey).Result() if err != nil { fmt.Println(err) fmt.Println("秒杀还未开始.......") return false } // 5、判断用户是否重复秒杀操作 flag, err := rdb.SIsMember(ctx, userKey, userKey).Result() if err != nil { fmt.Println(err) } if flag { fmt.Println("你已经参加了秒杀,无法再次参加。。。。") return false } // 6、判断商品数量,如果库存数量小于1,秒杀结束 str, err = rdb.Get(ctx, kcKey).Result() if err != nil { fmt.Println(err) } n, err := strconv.Atoi(str) if err != nil { fmt.Println(err) } if n < 1 { fmt.Println("秒杀结束,请下次再来吧。。。。") return false } // 7、秒杀过程 // 7.1、库存减1 num, err := rdb.Decr(ctx, kcKey).Result() if err != nil { fmt.Println(err) } if num != 0 { // 7.2、添加用户 rdb.SAdd(ctx, userKey, uuid) } return true } func main() { // 并发的版本 for i := 0; i < 20; i++ { go func() { uuid := GenerateUUID() prodid := "1023" time.Sleep(10 * time.Second) MsCode(uuid, prodid) }() } time.Sleep(15 * time.Second) }
使用watch进行监视key,关键部分如下。但是这样会造成一个问题,就是抢购不完,会有一些库存,但是又有人没有抢到。
err = rdb.Watch(ctx, func(tx *redis.Tx) error { n, err := tx.Get(ctx, kcKey).Int() if err != nil && err != redis.Nil { return err } if n <= 0 { return fmt.Errorf("抢购结束了!请下次早点来。。。。") } _, err = tx.TxPipelined(ctx, func(pipeliner redis.Pipeliner) error { err := pipeliner.Decr(ctx, kcKey).Err() if err != nil { return err } err = pipeliner.SAdd(ctx, userKey, uuid).Err() if err != nil { return err } return nil }) return err }, kcKey)
Lua操作redis能够比较好的解决这个问题。因为redis中使用watch是使用了悲观锁的形态,而悲观锁会自然得造成库存问题,因此要使用乐观锁。而redis天然不支持乐观锁,基于此,需要时lua来编写相关脚本。其主要有以下优势:
import ( "context" "fmt" "github.com/go-redis/redis/v8" "net" "time" ) func useLua(userid, prodid string) bool { //编写脚本 - 检查数值,是否够用,够用再减,否则返回减掉后的结果 var luaScript = redis.NewScript(` local userid=KEYS[1]; local prodid=KEYS[2]; local qtKey="sk:"..prodid..":qt"; local userKey="sk:"..prodid..":user"; local userExists=redis.call("sismember",userKey,userid); if tonumber(userExists)==1 then return 2; end local num=redis.call("get",qtKey); if tonumber(num)<=0 then return 0; else redis.call("decr",qtKey); redis.call("SAdd",userKey,userid); end return 1; `) //执行脚本 n, err := luaScript.Run(ctx, DB, []string{userid, prodid}).Result() if err != nil { return false } switch n { case int64(0): fmt.Println("抢购结束") return false case int64(1): fmt.Println(userid, ":抢购成功") return true case int64(2): fmt.Println(userid, ":已经抢购了") return false default: fmt.Println("发生未知错误!") return false } return true } func main() { // 并发的版本 for i := 0; i < 20; i++ { go func() { uuid := GenerateUUID() prodid := "1023" time.Sleep(10 * time.Second) useLua(uuid, prodid) }() } time.Sleep(15 * time.Second) }