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

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

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

0、簡(jiǎn)介

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

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

1、簡(jiǎn)單版

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

func MsCode(uuid, prodid string) bool { ?
? ?// 1、對(duì)uuid和prodid進(jìn)行非空判斷 ?
? ?if uuid == "" || prodid == "" { ?
? ? ? return false ?
? ?} ?
??
? ?//2、獲取連接 ?
? ?rdb := DB ?
??
? ?//3、拼接key ?
? ?kcKey := "kc:" + prodid + ":qt" ?
? ?userKey := "sk:" + prodid + ":user" ?
??
? ?//4、獲取庫(kù)存 ?
? ?str, err := rdb.Get(ctx, kcKey).Result() ?
? ?if err != nil { ?
? ? ? fmt.Println(err) ?
? ? ? fmt.Println("秒殺還未開始.......") ?
? ? ? return false ?
? ?} ?
??
? ?// 5、判斷用戶是否重復(fù)秒殺操作 ?
? ?flag, err := rdb.SIsMember(ctx, userKey, userKey).Result() ?
? ?if err != nil { ?
? ? ? fmt.Println(err) ?
? ?} ?
? ?if flag { ?
? ? ? fmt.Println("你已經(jīng)參加了秒殺，無(wú)法再次參加。。。。") ?
? ? ? return false ?
? ?} ?
??
? ?// 6、判斷商品數(shù)量，如果庫(kù)存數(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é)束，請(qǐng)下次再來(lái)吧。。。。") ?
? ? ? return false ?
? ?} ?
??
? ?// 7、秒殺過程 ?
? ?// 7.1、庫(kù)存減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進(jìn)行監(jiān)視key，關(guān)鍵部分如下。但是這樣會(huì)造成一個(gè)問題，就是搶購(gòu)不完，會(huì)有一些庫(kù)存，但是又有人沒有搶到。

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("搶購(gòu)結(jié)束了！請(qǐng)下次早點(diǎn)來(lái)。。。。")  
   }  
   _, 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、解決庫(kù)存問題Lua

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

將復(fù)雜的或者多步的redis操作，寫為一個(gè)腳本，一次提交給redis執(zhí)行，減少反復(fù)連接redis的次數(shù)。提升性能。
luan腳本類似redis事務(wù)，有一定的原子性，不會(huì)被其他命令插隊(duì)，可以完成一些redis事務(wù)性的操作。
redis的lua腳本功能，只有在redis2.6以上的版本才可以使用。
利用lua腳本淘汰用戶，解決超賣問題。
redis2.6版本以后，通過lua腳本解決爭(zhēng)奪問題，實(shí)際上是redis利用其單線程的特性，用任務(wù)隊(duì)列的方式解決多任務(wù)并發(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("搶購(gòu)結(jié)束")
?? ??? ?return false
?? ?case int64(1):
?? ??? ?fmt.Println(userid, "：搶購(gòu)成功")
?? ??? ?return true
?? ?case int64(2):
?? ??? ?fmt.Println(userid, "：已經(jīng)搶購(gòu)了")
?? ??? ?return false
?? ?default:
?? ??? ?fmt.Println("發(fā)生未知錯(cuò)誤！")
?? ??? ?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秒殺中庫(kù)存與超賣問題的文章就介紹到這了,更多相關(guān)Go Lua Redis秒殺中庫(kù)存與超賣內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: