Go+Lua解決Redis秒殺中庫存與超賣問題

更新時間：2023年03月01日 09:41:51 作者：ccgkk

本文主要介紹了Go+Lua解決Redis秒殺中庫存與超賣問題，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧

0、簡介

Go語言連接go-redis進行數(shù)據(jù)庫的連接，如果你對這部分尚不了解，建議你先學習這部分知識。
另外，本秒殺主要解決兩個問題，第一個就是超賣問題，另一個就是庫存問題。
沒有設(shè)計專門的頁面來模擬并發(fā)，我們直接使用gorountine，在調(diào)用請求前停留10s。
針對超賣問題，引入go-redis的watch搭配事務處理即可【相當于樂觀鎖】。

而針對庫存的問題較為麻煩一點，需要使用Lua編輯腳本，但是你無需在自己的機器上下載lua的編譯環(huán)境，go提供了其相關(guān)的支持。針對這一部分，不用慌張，其基本架構(gòu)如下：

1、簡單版

面對并發(fā)的情況下會出現(xiàn)超賣的情況，redis數(shù)據(jù)庫中會出現(xiàn)負值的情況。即使你在操作之前進行了數(shù)據(jù)的判斷。

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("你已經(jīng)參加了秒殺，無法再次參加。。。。") ?
? ? ? return false ?
? ?} ?
??
? ?// 6、判斷商品數(shù)量，如果庫存數(shù)量小于1，秒殺結(jié)束 ?
? ?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("秒殺結(jié)束，請下次再來吧。。。。") ?
? ? ? 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() {
?? ?// 并發(fā)的版本
?? ?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)
}

2、解決超賣

使用watch進行監(jiān)視key，關(guān)鍵部分如下。但是這樣會造成一個問題，就是搶購不完，會有一些庫存，但是又有人沒有搶到。

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("搶購結(jié)束了！請下次早點來。。。。")  
   }  
   _, 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)

3、解決庫存問題Lua

Lua操作redis能夠比較好的解決這個問題。因為redis中使用watch是使用了悲觀鎖的形態(tài)，而悲觀鎖會自然得造成庫存問題，因此要使用樂觀鎖。而redis天然不支持樂觀鎖，基于此，需要時lua來編寫相關(guān)腳本。其主要有以下優(yōu)勢：

將復雜的或者多步的redis操作，寫為一個腳本，一次提交給redis執(zhí)行，減少反復連接redis的次數(shù)。提升性能。
luan腳本類似redis事務，有一定的原子性，不會被其他命令插隊，可以完成一些redis事務性的操作。
redis的lua腳本功能，只有在redis2.6以上的版本才可以使用。
利用lua腳本淘汰用戶，解決超賣問題。
redis2.6版本以后，通過lua腳本解決爭奪問題，實際上是redis利用其單線程的特性，用任務隊列的方式解決多任務并發(fā)問題。

import (
?? ?"context"
?? ?"fmt"
?? ?"github.com/go-redis/redis/v8"
?? ?"net"
?? ?"time"
)

func useLua(userid, prodid string) bool {
?? ?//編寫腳本 - 檢查數(shù)值，是否夠用，夠用再減，否則返回減掉后的結(jié)果
?? ?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;
?? ?`)
?? ?//執(zhí)行腳本
?? ?n, err := luaScript.Run(ctx, DB, []string{userid, prodid}).Result()
?? ?if err != nil {
?? ??? ?return false
?? ?}
?? ?switch n {
?? ?case int64(0):
?? ??? ?fmt.Println("搶購結(jié)束")
?? ??? ?return false
?? ?case int64(1):
?? ??? ?fmt.Println(userid, "：搶購成功")
?? ??? ?return true
?? ?case int64(2):
?? ??? ?fmt.Println(userid, "：已經(jīng)搶購了")
?? ??? ?return false
?? ?default:
?? ??? ?fmt.Println("發(fā)生未知錯誤！")
?? ??? ?return false
?? ?}
?? ?return true
}

func main() {
?? ?// 并發(fā)的版本
?? ?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)
}

到此這篇關(guān)于Go+Lua解決Redis秒殺中庫存與超賣問題的文章就介紹到這了,更多相關(guān)Go Lua Redis秒殺中庫存與超賣內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: