前言
群里有小伙伴咨詢微信紅包的架構(gòu)伟桅,對于我來說退敦,顯然是不知道的,但是寫一個相對高并發(fā)的搶紅包案例還是完全可以的邪锌。
架構(gòu)設(shè)計
image
業(yè)務(wù)流程
老板發(fā)紅包勉躺,此時緩存初始化紅包個數(shù),紅包金額(單位分)觅丰,并異步入庫饵溅。
搶紅包,判斷緩存剩余紅包金額妇萄,剩余金額大于零則搶到紅包蜕企,否則手慢了,紅包派完了
拆紅包冠句,根據(jù) redPacketId 獲取分布式鎖轻掩,如果獲取到鎖,紅包個數(shù)減一懦底,如果剩余紅包個數(shù)大于零搶紅包成功唇牧、否則失敗。成功則計算紅包金額聚唐,緩存總紅包金額減去搶到的紅包金額丐重,異步入庫、異步到賬杆查。
數(shù)據(jù)庫設(shè)計
紅包信息表
CREATETABLE`red_racket`(`id`bigint(20)NOTNULLAUTO_INCREMENTCOMMENT'自增主鍵',`red_packet_id`bigint(20)NOTNULLCOMMENT'紅包唯一ID',`total_amount`int(11)NOTNULLCOMMENT'紅包金額單位分',`total_packet`int(11)NOTNULLCOMMENT'紅包個數(shù)',`type`int(11)NOTNULLCOMMENT'紅包類型',`create_time`datetimeDEFAULTNULLCOMMENT'創(chuàng)建時間',`version`int(11)NOTNULLCOMMENT'版本號',PRIMARYKEY(`id`))ENGINE=InnoDBAUTO_INCREMENT=2DEFAULTCHARSET=utf8CHECKSUM=1DELAY_KEY_WRITE=1ROW_FORMAT=DYNAMICCOMMENT='紅包信息表';
搶紅包記錄表
CREATETABLE`red_packet_record`(`id`bigint(20)NOTNULLAUTO_INCREMENTCOMMENT'自增主鍵',`amount`int(11)NOTNULLCOMMENT'搶到紅包的金額',`red_packet_id`bigint(20)NOTNULLCOMMENT'紅包ID',`uid`int(11)NOTNULLCOMMENT'搶到紅包用戶的用戶標(biāo)識',`create_time`datetimeDEFAULTNULLCOMMENT'創(chuàng)建時間',PRIMARYKEY(`id`))ENGINE=InnoDBAUTO_INCREMENT=10DEFAULTCHARSET=utf8CHECKSUM=1DELAY_KEY_WRITE=1ROW_FORMAT=DYNAMICCOMMENT='搶紅包記錄表';
代碼案例
老板發(fā)了10個紅包一共200人民幣扮惦,100個人同時搶紅包,偽代碼分別為拆紅包和搶紅包相關(guān)業(yè)務(wù)邏輯根灯。
模擬搶紅包偽代碼:
/**
* 搶紅包 拆紅包 搶到不一定能拆到
* @param redPacketId
* @return
*/
@ApiOperation(value="搶紅包二",nickname="爪哇筆記")
@PostMapping("/startTwo")
public Result startTwo(long redPacketId){
int skillNum = 100;
final CountDownLatch latch = new CountDownLatch(skillNum);//N個搶紅包
/**
* 初始化紅包數(shù)據(jù)径缅,搶紅包攔截
*/
redisUtil.cacheValue(redPacketId+"-num",10);
/**
* 初始化紅包金額,單位為分
*/
redisUtil.cacheValue(redPacketId+"-money",20000);
/**
* 模擬100個用戶搶10個紅包
*/
for(int i=1;i<=skillNum;i++){
int userId = i;
Runnable task = () -> {
/**
* 搶紅包 判斷剩余金額
*/
Integer money = (Integer) redisUtil.getValue(redPacketId+"-money");
if(money>0){
/**
* 雖然能搶到 但是不一定能拆到
* 類似于微信的 點(diǎn)擊紅包顯示搶的按鈕
*/
Result result = redPacketService.startTwoSeckil(redPacketId,userId);
if(result.get("code").toString().equals("500")){
LOGGER.info("用戶{}手慢了烙肺,紅包派完了",userId);
}else{
Double amount = DoubleUtil.divide(Double.parseDouble(result.get("msg").toString()), (double) 100);
LOGGER.info("用戶{}搶紅包成功纳猪,金額:{}", userId,amount);
}
}else{
/**
* 直接顯示手慢了,紅包派完了
*/
//LOGGER.info("用戶{}手慢了桃笙,紅包派完了",userId);
}
latch.countDown();
};
executor.execute(task);
}
try {
latch.await();
Integer restMoney = Integer.parseInt(redisUtil.getValue(redPacketId+"-money").toString());
LOGGER.info("剩余金額:{}",restMoney);
} catch (InterruptedException e) {
e.printStackTrace();
}
return Result.ok();
}
業(yè)務(wù)層拆紅包:
@Override
@Transactional
public Result startTwoSeckil(long redPacketId, int userId) {
Integer money = 0;
boolean res=false;
try {
/**
* 獲取鎖 保證紅包數(shù)量和計算紅包金額的原子性操作
*/
res = RedissLockUtil.tryLock(redPacketId+"", TimeUnit.SECONDS, 3, 10);
if(res){
long restPeople = redisUtil.decr(redPacketId+"-num",1);
if(restPeople>0){
/**
* 如果是最后一人
*/
if(restPeople==1){
money = Integer.parseInt(redisUtil.getValue(redPacketId+"-money").toString());
}else{
Integer restMoney = Integer.parseInt(redisUtil.getValue(redPacketId+"-money").toString());
Random random = new Random();
//隨機(jī)范圍:[1,剩余人均金額的兩倍]
money = random.nextInt((int) (restMoney / (restPeople+1) * 2 - 1)) + 1;
}
redisUtil.decr(redPacketId+"-money",money);
/**
* 異步入庫
*/
RedPacketRecord record = new RedPacketRecord();
record.setMoney(money);
record.setRedPacketId(redPacketId);
record.setUid(userId);
record.setCreateTime(new Timestamp(System.currentTimeMillis()));
saveRecord(record);
/**
* 異步入賬
*/
}else{
return Result.error("手慢了氏堤,紅包派完了");
}
}else{
/**
* 獲取鎖失敗相當(dāng)于搶紅包失敗
*/
return Result.error("手慢了,紅包派完了");
}
} catch (Exception e) {
e.printStackTrace();
}finally {
if(res){//釋放鎖
RedissLockUtil.unlock(redPacketId+"");
}
}
return Result.ok(money);
}
演示
在 Application中有接口演示說明搏明,你可以在搶紅包 Red Packet Controller接口中輸入任何參數(shù)進(jìn)行測試鼠锈,也可以配合數(shù)據(jù)庫稍加修改即可作為生產(chǎn)環(huán)境的搶紅包功能模塊。
ps:在實(shí)際的業(yè)務(wù)中紅包肯定是要提前拆分好的星著,寫入庫中的购笆,搶紅包其實(shí)搶的就是一個紅包記錄的id,只要把這個紅包的id搶到虚循,就算你搶到紅包了