redis 腳本介紹
Redis從2.6版本開始，通過內(nèi)嵌支持Lua環(huán)境
好處
- 減少網(wǎng)絡(luò)開銷?？梢詫⒍鄠€(gè)請求通過腳本的形式一次發(fā)送，減少網(wǎng)絡(luò)延遲
- 原子操作。redis將整個(gè)腳本當(dāng)作一個(gè)整體去執(zhí)行，中間不會(huì)被其他命令插入，無需擔(dān)心腳本執(zhí)行過程中會(huì)出現(xiàn)競態(tài)條件
- 復(fù)用。客戶端發(fā)送的腳本會(huì)永久保存在redis中，可以復(fù)用這一腳本
數(shù)據(jù)庫表設(shè)計(jì)
簡單兩張表，一個(gè)紅包表，一個(gè)紅包領(lǐng)取記錄表
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 '數(shù)量(分割成幾分)',
`create_time` datetime DEFAULT NULL COMMENT '創(chuàng)建時(shí)間',
`update_time` datetime DEFAULT NULL COMMENT '更新時(shí)間',
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 '領(lǐng)取到獎(jiǎng)勵(lì)',
`red_envelope_id` bigint(20) DEFAULT NULL COMMENT '紅包id',
`create_time` datetime DEFAULT NULL COMMENT '創(chuàng)建時(shí)間',
`update_time` datetime DEFAULT NULL COMMENT '更新時(shí)間',
PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=11 DEFAULT CHARSET=utf8mb4 COMMENT='紅包領(lǐng)取記錄'
代碼編寫
首先，生成一個(gè)紅包，將其分成指定數(shù)量的隨機(jī)小紅包，以list結(jié)構(gòu)（envelope:redEnvelopeId:紅包id作為key）存儲(chǔ)在reids中（以便搶紅包彈出數(shù)據(jù)）
public Long divideRedEnvelope(int amount, int num) {
/**
* 每個(gè)人至少分到一分錢,如果有2000分，6人，隨機(jī)得到五個(gè)小于1994（2000-6）的數(shù)
* 比如 a1=4，a2=120，a3=324，a4=500，a5=700(隨機(jī)拿到的五個(gè)數(shù)進(jìn)行排序)，那么紅包錢分別為： 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);
/**
* 拿來隨機(jī)分的，按分來算
*/
int totalAmount = amount * 100 - num;
/**
* 隨機(jī)數(shù)
*/
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);
/**
* 條件語句分別分配的第一個(gè)、最后一個(gè)、中間的紅包
*/
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;
}
}
/**
* 產(chǎn)生的小紅包key,以list存儲(chǔ)在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();
}
搶紅包時(shí)，根據(jù)用戶userId和紅包id,生成KEYS[1]、KEYS[2]、KEYS[3] (存儲(chǔ)小紅包的key、領(lǐng)取紅包記錄的key、用戶userId的key)傳入腳本中。
1、先判斷該用戶是否搶過紅包，有則返回-1，沒有則從紅包列表取出一個(gè)小紅包
2、步驟1的小紅包如果為空，則表明紅包已經(jīng)沒搶光，返回 -2
3、否則返回取出的小紅包金額
public String grabRedEnvelope(Long userId, Long redEnvelopeId) {
DefaultRedisScriptString> redisScript = new DefaultRedisScript>();
redisScript.setResultType(String.class);
redisScript.setScriptText(LuaScript.redLua);
ListString> keyList = new ArrayList();
/**
* 產(chǎn)生的小紅包key
*/
keyList.add("envelope:redEnvelopeId:" + redEnvelopeId);
/**
* 紅包領(lǐng)取記錄key
*/
keyList.add("envelope:record:" + redEnvelopeId);
keyList.add("" + userId);
keyList.add(String.valueOf(userId));
/**
* -1 已經(jīng)搶到紅包 -2 紅包已經(jīng)完了 ，其余是搶到紅包并返回紅包余額
*/
String result = stringRedisTemplate.execute(redisScript, keyList);
return result;
}
實(shí)現(xiàn)搶紅包的Lua腳本
public class LuaScript {
/**
* -1 已經(jīng)搶到紅包 -2 紅包被搶光 re 紅包金額 ，keys[1]、keys[2]、keys[3]分別為存儲(chǔ)小紅包的key、紅包領(lǐng)取記錄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[2],KEYS[3],1);\n" +
"return re;\n" +
"else\n" +
"return '-2';\n" +
"end\n" +
"end";
}
測試
首先通過接口分配紅包生成一個(gè)100塊、份額為10份的紅包，并將其mysql數(shù)據(jù)庫和redis
通過jmeter進(jìn)行壓測搶紅包
結(jié)果
github代碼鏈接
鏈接
總結(jié)
到此這篇關(guān)于通過redis的腳本lua如何實(shí)現(xiàn)搶紅包功能的文章就介紹到這了,更多相關(guān)redis的腳本lua實(shí)現(xiàn)搶紅包內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
您可能感興趣的文章:- 詳解利用redis + lua解決搶紅包高并發(fā)的問題
- 簡介Lua腳本與Redis數(shù)據(jù)庫的結(jié)合使用
- Redis執(zhí)行Lua腳本的好處與示例代碼
- redis中如何使用lua腳本讓你的靈活性提高5個(gè)逼格詳解
- 利用Lua定制Redis命令的方法詳解
- Redis如何使用lua腳本實(shí)例教程
- Nginx利用Lua+Redis實(shí)現(xiàn)動(dòng)態(tài)封禁IP的方法
- 詳解Redis中Lua腳本的應(yīng)用和實(shí)踐
- Redis和Lua使用過程中遇到的小問題
