前言

群里有小伙伴咨詢微信紅包的架構(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'搶到紅包用戶的用戶標識',`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){
                    /**
                    * 雖然能搶到 但是不一定能拆到
                    * 類似于微信的 點擊紅包顯示搶的按鈕
                    */
                    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();
                        //隨機范圍：[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{
                /**
                * 獲取鎖失敗相當于搶紅包失敗
                */
                return Result.error("手慢了，紅包派完了");
            }
      } catch (Exception e) {
            e.printStackTrace();
        }finally {
            if(res){//釋放鎖
                RedissLockUtil.unlock(redPacketId+"");
            }
        }
        return Result.ok(money);
}

演示

在 Application中有接口演示說明，你可以在搶紅包 Red Packet Controller接口中輸入任何參數(shù)進行測試，也可以配合數(shù)據(jù)庫稍加修改即可作為生產(chǎn)環(huán)境的搶紅包功能模塊。

ps:在實際的業(yè)務(wù)中紅包肯定是要提前拆分好的，寫入庫中的，搶紅包其實搶的就是一個紅包記錄的id，只要把這個紅包的id搶到，就算你搶到紅包了