怎么使用Go和Lua解決Redis秒殺中庫(kù)存與超賣(mài)問(wèn)題

這篇文章主要介紹“怎么使用Go和Lua解決Redis秒殺中庫(kù)存與超賣(mài)問(wèn)題”，在日常操作中，相信很多人在怎么使用Go和Lua解決Redis秒殺中庫(kù)存與超賣(mài)問(wèn)題問(wèn)題上存在疑惑，小編查閱了各式資料，整理出簡(jiǎn)單好用的操作方法，希望對(duì)大家解答”怎么使用Go和Lua解決Redis秒殺中庫(kù)存與超賣(mài)問(wèn)題”的疑惑有所幫助！接下來(lái)，請(qǐng)跟著小編一起來(lái)學(xué)習(xí)吧！

0、簡(jiǎn)介

• Go語(yǔ)言連接go-redis進(jìn)行數(shù)據(jù)庫(kù)的連接，如果你對(duì)這部分尚不了解，建議你先學(xué)習(xí)這部分知識(shí)。

• 另外，本秒殺主要解決兩個(gè)問(wèn)題，第一個(gè)就是超賣(mài)問(wèn)題，另一個(gè)就是庫(kù)存問(wèn)題。

• 沒(méi)有設(shè)計(jì)專(zhuān)門(mén)的頁(yè)面來(lái)模擬并發(fā)，我們直接使用gorountine，在調(diào)用請(qǐng)求前停留10s。

• 針對(duì)超賣(mài)問(wèn)題，引入go-redis的watch搭配事務(wù)處理即可【相當(dāng)于樂(lè)觀鎖】。

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

1、簡(jiǎn)單版

面對(duì)并發(fā)的情況下會(huì)出現(xiàn)超賣(mài)的情況，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("秒殺還未開(kāi)始.......")  
      return false  
   }  
  
   // 5、判斷用戶(hù)是否重復(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、秒殺過(guò)程  
   // 7.1、庫(kù)存減1  
   num, err := rdb.Decr(ctx, kcKey).Result()  
   if err != nil {  
      fmt.Println(err)  
   }  
   if num != 0 {  
      // 7.2、添加用戶(hù)  
      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、解決超賣(mài)

使用watch進(jìn)行監(jiān)視key，關(guān)鍵部分如下。但是這樣會(huì)造成一個(gè)問(wèn)題，就是搶購(gòu)不完，會(huì)有一些庫(kù)存，但是又有人沒(méi)有搶到。

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ù)存問(wèn)題Lua

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

• 將復(fù)雜的或者多步的redis操作，寫(xiě)為一個(gè)腳本，一次提交給redis執(zhí)行，減少反復(fù)連接redis的次數(shù)。提升性能。

• luan腳本類(lèi)似redis事務(wù)，有一定的原子性，不會(huì)被其他命令插隊(duì)，可以完成一些redis事務(wù)性的操作。

• redis的lua腳本功能，只有在redis2.6以上的版本才可以使用。

• 利用lua腳本淘汰用戶(hù)，解決超賣(mài)問(wèn)題。

• redis2.6版本以后，通過(guò)lua腳本解決爭(zhēng)奪問(wèn)題，實(shí)際上是redis利用其單線程的特性，用任務(wù)隊(duì)列的方式解決多任務(wù)并發(fā)問(wèn)題。

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

func useLua(userid, prodid string) bool {
    //編寫(xiě)腳本 - 檢查數(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ù)存與超賣(mài)問(wèn)題”的學(xué)習(xí)就結(jié)束了，希望能夠解決大家的疑惑。理論與實(shí)踐的搭配能更好的幫助大家學(xué)習(xí)，快去試試吧！若想繼續(xù)學(xué)習(xí)更多相關(guān)知識(shí)，請(qǐng)繼續(xù)關(guān)注億速云網(wǎng)站，小編會(huì)繼續(xù)努力為大家?guī)?lái)更多實(shí)用的文章！