SpringBoot + Redis:模擬 10w 人的秒殺搶單!
SpringBoot + Redis:模擬 10w 人的秒殺搶單!
本篇內(nèi)容主要講解的是redis分布式鎖,這個在各大廠面試幾乎都是必備的,下面結(jié)合模擬搶單的場景來使用她;本篇不涉及到的redis環(huán)境搭建,快速搭建個人測試環(huán)境,這里建議使用docker;本篇內(nèi)容節(jié)點(diǎn)如下:
Jedis的nx生成鎖
如何刪除鎖
模擬搶單動作(10w個人開搶)
jedis的nx生成鎖
對于java中想操作redis,好的方式是使用jedis,首先pom中引入依賴:
對于分布式鎖的生成通常需要注意如下幾個方面:
創(chuàng)建鎖的策略:?redis的普通key一般都允許覆蓋,A用戶set某個key后,B在set相同的key時同樣能成功,如果是鎖場景,那就無法知道到底是哪個用戶set成功的;這里jedis的setnx方式為我們解決了這個問題,簡單原理是:當(dāng)A用戶先set成功了,那B用戶set的時候就返回失敗,滿足了某個時間點(diǎn)只允許一個用戶拿到鎖。
鎖過期時間:?某個搶購場景時候,如果沒有過期的概念,當(dāng)A用戶生成了鎖,但是后面的流程被阻塞了一直無法釋放鎖,那其他用戶此時獲取鎖就會一直失敗,無法完成搶購的活動;當(dāng)然正常情況一般都不會阻塞,A用戶流程會正常釋放鎖;過期時間只是為了更有保障。
下面來上段setnx操作的代碼:
這里注意點(diǎn)在于jedis的set方法,其參數(shù)的說明如:
NX:是否存在key,存在就不set成功
PX:key過期時間單位設(shè)置為毫秒(EX:單位秒)
setnx如果失敗直接封裝返回false即可,下面我們通過一個get方式的api來調(diào)用下這個setnx方法:
訪問如下測試url,正常來說第一次返回了true,第二次返回了false,由于第二次請求的時候redis的key已存在,所以無法set成功

由上圖能夠看到只有一次set成功,并key具有一個有效時間,此時已到達(dá)了分布式鎖的條件。
如何刪除鎖
上面是創(chuàng)建鎖,同樣的具有有效時間,但是我們不能完全依賴這個有效時間,場景如:有效時間設(shè)置1分鐘,本身用戶A獲取鎖后,沒遇到什么特殊情況正常生成了搶購訂單后,此時其他用戶應(yīng)該能正常下單了才對,但是由于有個1分鐘后鎖才能自動釋放,那其他用戶在這1分鐘無法正常下單(因?yàn)殒i還是A用戶的),因此我們需要A用戶操作完后,主動去解鎖:
這里也使用了jedis方式,直接執(zhí)行l(wèi)ua腳本:根據(jù)val判斷其是否存在,如果存在就del;
其實(shí)個人認(rèn)為通過jedis的get方式獲取val后,然后再比較value是否是當(dāng)前持有鎖的用戶,如果是那最后再刪除,效果其實(shí)相當(dāng);只不過直接通過eval執(zhí)行腳本,這樣避免多一次操作了redis而已,縮短了原子操作的間隔。(如有不同見解請留言探討);同樣這里創(chuàng)建個get方式的api來測試:
注意的是delnx時,需要傳遞創(chuàng)建鎖時的value,因?yàn)橥ㄟ^et的value與delnx的value來判斷是否是持有鎖的操作請求,只有value一樣才允許del;
模擬搶單動作(10w個人開搶)
有了上面對分布式鎖的粗略基礎(chǔ),我們模擬下10w人搶單的場景,其實(shí)就是一個并發(fā)操作請求而已,由于環(huán)境有限,只能如此測試;如下初始化10w個用戶,并初始化庫存,商品等信息,如下代碼:
有了上面10w個不同用戶,我們設(shè)定商品只有10個庫存,然后通過并行流的方式來模擬搶購,如下?lián)屬彽膶?shí)現(xiàn):
這里實(shí)現(xiàn)的邏輯是:
1、parallelStream():并行流模擬多用戶搶購
2、(startTime + timeout) >= System.currentTimeMillis():判斷未搶成功的用戶,timeout秒內(nèi)繼續(xù)獲取鎖
3、獲取鎖前和后都判斷庫存是否還足夠
4、jedisCom.setnx(shangpingKey, b):用戶獲取搶購鎖
5、獲取鎖后并下單成功,最后釋放鎖:jedisCom.delnx(shangpingKey, b)
再來看下記錄的日志結(jié)果:

最終返回?fù)屬彸晒Φ挠脩簦?/p>