通過redis的指令碼lua如何實現搶紅包功能
阿新 • • 發佈:2020-05-27
redis 指令碼介紹
Redis從2.6版本開始,通過內嵌支援Lua環境
好處
- 減少網路開銷。可以將多個請求通過指令碼的形式一次傳送,減少網路延遲
- 原子操作。redis將整個腳本當作一個整體去執行,中間不會被其他命令插入,無需擔心指令碼執行過程中會出現競態條件
- 複用。客戶端傳送的指令碼會永久儲存在redis中,可以複用這一指令碼
資料庫表設計
簡單兩張表,一個紅包表,一個紅包領取記錄表
CREATE TABLE `t_red_envelope` ( `id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT 'ID',`amount` decimal(10,2) DEFAULT NULL COMMENT '金額',`num` int(11) DEFAULT NULL COMMENT '數量(分割成幾分)',`create_time` datetime DEFAULT NULL COMMENT '建立時間',`update_time` datetime DEFAULT NULL COMMENT '更新時間',PRIMARY KEY (`id`) ) ENGINE=InnoDB AUTO_INCREMENT=10 DEFAULT CHARSET=utf8mb4 COMMENT='紅包' CREATE TABLE `t_red_envelope_record` ( `id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT 'id',`user_id` bigint(20) DEFAULT NULL COMMENT '使用者id',`reward` decimal(10,2) DEFAULT NULL COMMENT '領取到獎勵',`red_envelope_id` bigint(20) DEFAULT NULL COMMENT '紅包id',PRIMARY KEY (`id`) ) ENGINE=InnoDB AUTO_INCREMENT=11 DEFAULT CHARSET=utf8mb4 COMMENT='紅包領取記錄'
程式碼編寫
首先,生成一個紅包,將其分成指定數量的隨機小紅包,以list結構(envelope:redEnvelopeId:紅包id作為key)儲存在reids中(以便搶紅包彈出資料)
public Long divideRedEnvelope(int amount,int num) { /** * 每個人至少分到一分錢,如果有2000分,6人,隨機得到五個小於1994(2000-6)的數 * 比如 a1=4,a2=120,a3=324,a4=500,a5=700(隨機拿到的五個數進行排序),那麼紅包錢分別為: a1+1,a2-a1+1,a3-a2+1,a4-a3+1,a5-a4+1,1994-a5+1(總和剛好為2000) */ RedEnvelope redEnvelope = new RedEnvelope(); redEnvelope.setAmount(new BigDecimal(amount)); redEnvelope.setNum(num); redEnvelope.setCreateTime(new Date()); redEnvelope.setUpdateTime(new Date()); redEnvelopeDao.insert(redEnvelope); /** * 拿來隨機分的,按分來算 */ int totalAmount = amount * 100 - num; /** * 隨機數 */ int[] randomNum = new int[num - 1]; /** * 紅包金額 */ int[] redEnvelopeAmount = new int[num]; for (int i = 0; i < num - 1; i++) { int rand = new Random().nextInt(totalAmount); randomNum[i] = rand; } Arrays.sort(randomNum); /** * 條件語句分別分配的第一個、最後一個、中間的紅包 */ for (int i = 0; i < num; i++) { if (i == 0) { redEnvelopeAmount[i] = randomNum[i] + 1; } else if (i == num - 1) { redEnvelopeAmount[i] = totalAmount - randomNum[i - 1] + 1; } else { redEnvelopeAmount[i] = randomNum[i] - randomNum[i - 1] + 1; } } /** * 產生的小紅包key,以list儲存在reids中 */ String key = "envelope:redEnvelopeId:" + redEnvelope.getId(); Boolean flag = stringRedisTemplate.hasKey(key); if (!flag) { for (Integer i : redEnvelopeAmount) { stringRedisTemplate.opsForList().leftPush(key,i + ""); } } return redEnvelope.getId(); }
搶紅包時,根據使用者userId和紅包id,生成KEYS[1]、KEYS[2]、KEYS[3] (儲存小紅包的key、領取紅包記錄的key、使用者userId的key)傳入指令碼中。
1、先判斷該使用者是否搶過紅包,有則返回-1,沒有則從紅包列表取出一個小紅包
2、步驟1的小紅包如果為空,則表明紅包已經沒搶光,返回 -2
3、否則返回取出的小紅包金額
public String grabRedEnvelope(Long userId,Long redEnvelopeId) { DefaultRedisScript<String> redisScript = new DefaultRedisScript<>(); redisScript.setResultType(String.class); redisScript.setScriptText(LuaScript.redLua); List<String> keyList = new ArrayList(); /** * 產生的小紅包key */ keyList.add("envelope:redEnvelopeId:" + redEnvelopeId); /** * 紅包領取記錄key */ keyList.add("envelope:record:" + redEnvelopeId); keyList.add("" + userId); keyList.add(String.valueOf(userId)); /** * -1 已經搶到紅包 -2 紅包已經完了 ,其餘是搶到紅包並返回紅包餘額 */ String result = stringRedisTemplate.execute(redisScript,keyList); return result; }
實現搶紅包的Lua指令碼
public class LuaScript { /** * -1 已經搶到紅包 -2 紅包被搶光 re 紅包金額 ,keys[1]、keys[2]、keys[3]分別為儲存小紅包的key、紅包領取記錄key、使用者id */ public static String redLua = "if redis.call('hexists',KEYS[2],KEYS[3]) ~=0 then \n" + " return '-1';\n" + " else \n" + "local re=redis.call('rpop',KEYS[1]);\n" + "if re then\n" + "redis.call('hset',KEYS[3],1);\n" + "return re;\n" + "else\n" + "return '-2';\n" + "end\n" + "end"; }
測試
首先通過介面分配紅包生成一個100塊、份額為10份的紅包,並將其mysql資料庫和redis
通過jmeter進行壓測搶紅包
結果
github程式碼連結
連結
總結
到此這篇關於通過redis的指令碼lua如何實現搶紅包功能的文章就介紹到這了,更多相關redis的指令碼lua實現搶紅包內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!