redis使用Lua腳本解決多線程下的超賣問題及原因解析

更新時間：2023年05月08日 09:23:05 作者：David.Meng

這篇文章主要介紹了redis使用Lua腳本解決多線程下的超賣問題，本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下

一.多線程下引起的超賣問題呈現(xiàn)

1.1.我先初始化庫存數(shù)量為1、訂單數(shù)量為0

1.2.開啟3個線程去執(zhí)行業(yè)務

業(yè)務為：判斷如果說庫存數(shù)量大于0，則庫存減1，訂單數(shù)量加1

結果為：庫存為-2，訂單數(shù)量為3

原因：如下圖所示，這是因為分別有6個指令（3個庫存減1指令，3個訂單數(shù)量加1指令）在redis服務端執(zhí)行導致的。

namespace MengLin.Shopping.Redis.LuaScript
{
    public class SecKillOriginal
    {
        static SecKillOriginal()
        {
            using (RedisClient client = new RedisClient("127.0.0.1", 6379))
            {
                //刪除當前數(shù)據(jù)庫中的所有Key, 默認刪除的是db0
                client.FlushDb();
                //刪除所有數(shù)據(jù)庫中的key 
                client.FlushAll();
                //初始化庫存數(shù)量為1和訂單數(shù)量為0
                client.Set("inventoryNum", 1);
                client.Set("orderNum", 0);
            }
        }
        public static void Show()
        {
            for (int i = 0; i < 3; i++)
            {
                Task.Run(() =>
                {
                    using (RedisClient client = new RedisClient("127.0.0.1", 6379))
                    {
                        int inventoryNum = client.Get<int>("inventoryNum");
                        //如果庫存數(shù)量大于0
                        if (inventoryNum > 0)
                        {
                            //給庫存數(shù)量-1
                            var inventoryNum2 = client.Decr("inventoryNum");
                            Console.WriteLine($"給庫存數(shù)量-1后的數(shù)量-inventoryNum: {inventoryNum2}");
                            //給訂單數(shù)量+1
                            var orderNum = client.Incr("orderNum");
                            Console.WriteLine($"給訂單數(shù)量+1后的數(shù)量-orderNum: {orderNum}");
                        }
                        else
                        {
                            Console.WriteLine($"搶購失敗: 原因是因為沒有庫存");
                        }
                    }
                });
            }
        }
    }
}

二.使用Lua腳本解決多線程下超賣的問題以及為什么

2.1.修改后的代碼如下

結果為：如下圖所示，庫存為0、訂單數(shù)量為1，并沒有出現(xiàn)超賣的問題且有2個線程搶不到。

namespace MengLin.Shopping.Redis.LuaScript
{
    public class SecKillLua
    {
        /// <summary>
        /// 使用Lua腳本解決多線程下變賣的問題
        /// </summary>
        static SecKillLua()
        {
            using (RedisClient client = new RedisClient("127.0.0.1", 6379))
            {
                //刪除當前數(shù)據(jù)庫中的所有Key, 默認刪除的是db0
                client.FlushDb();
                //刪除所有數(shù)據(jù)庫中的key 
                client.FlushAll();
                //初始化庫存數(shù)量為1和訂單數(shù)量為0
                client.Set("inventoryNum", 1);
                client.Set("orderNum", 0);
            }
        }
        public static void Show()
        {
            for (int i = 0; i < 3; i++)
            {
                Task.Run(() =>
                {
                    using (RedisClient client = new RedisClient("127.0.0.1", 6379))
                    {
                        //如果庫存數(shù)量大于0，則給庫存數(shù)量-1，給訂單數(shù)量+1
                        var lua = @"local count = redis.call('get',KEYS[1])
                                        if(tonumber(count)>0)
                                        then
                                            --return count
                                            redis.call('INCR',ARGV[1])
                                            return redis.call('DECR',KEYS[1])
                                        else
                                            return -99
                                        end";
                        Console.WriteLine(client.ExecLuaAsString(lua, keys: new[] { "inventoryNum" }, args: new[] { "orderNum" }));
                    }
                });
            }
        }
    }
}