前言

群裡有小夥伴諮詢微信紅包的架構，對於我來說，顯然是不知道的，但是寫一個相對高併發的搶紅包案例還是完全可以的。

架構設計

業務流程

• 老闆發紅包，此時快取初始化紅包個數，紅包金額(單位分)，並非同步入庫。

• 搶紅包，判斷快取剩餘紅包金額，剩餘金額大於零則搶到紅包，否則手慢了，紅包派完了

• 拆紅包，根據 redPacketId 獲取分散式鎖，如果獲取到鎖，紅包個數減一，如果剩餘紅包個數大於零搶紅包成功、否則失敗。成功則計算紅包金額，快取總紅包金額減去搶到的紅包金額，非同步入庫、非同步到賬。

資料庫設計

• 紅包資訊表

CREATE TABLE `red_racket` (
   `id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '自增主鍵',
   `red_packet_id` bigint(20) NOT NULL COMMENT '紅包唯一ID',
   `total_amount` int(11) NOT NULL COMMENT '紅包金額單位分',
   `total_packet` int(11) NOT NULL COMMENT '紅包個數',
   `type` int(11) NOT NULL COMMENT '紅包型別',
   `create_time` datetime DEFAULT NULL COMMENT '建立時間',
   `version` int(11) NOT NULL COMMENT '版本號',
   PRIMARY KEY (`id`)
 ) ENGINE=InnoDB AUTO_INCREMENT=2 DEFAULT CHARSET=utf8 CHECKSUM=1 DELAY_KEY_WRITE=1 ROW_FORMAT=DYNAMIC COMMENT='紅包資訊表'
 

• 搶紅包記錄表

CREATE TABLE `red_packet_record` (
   `id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '自增主鍵',
   `amount` int(11) NOT NULL COMMENT '搶到紅包的金額',
   `red_packet_id` bigint(20) NOT NULL COMMENT '紅包ID',
   `uid` int(11) NOT NULL COMMENT '搶到紅包使用者的使用者標識',
   `create_time` datetime DEFAULT NULL COMMENT '建立時間',
   PRIMARY KEY (`id`)
 ) ENGINE=InnoDB AUTO_INCREMENT=10 DEFAULT CHARSET=utf8 CHECKSUM=1 DELAY_KEY_WRITE=1 ROW_FORMAT=DYNAMIC COMMENT='搶紅包記錄表'
 

程式碼案例

老闆發了10個紅包一共200人民幣，100個人同時搶紅包，虛擬碼分別為拆紅包和搶紅包相關業務邏輯。

模擬搶紅包虛擬碼：

   /**
     * 搶紅包 拆紅包 搶到不一定能拆到
     * @param redPacketId
     * @return
     */
    @ApiOperation(value="搶紅包二",nickname="爪哇筆記")
    @PostMapping("/startTwo")
    public Result startTwo(long redPacketId){
        int skillNum = 100;
        final CountDownLatch latch = new CountDownLatch(skillNum);//N個搶紅包
        /**
         * 初始化紅包資料，搶紅包攔截
         */
        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();
    }
 

業務層拆紅包：

    @Override
    @Transactional
    public Result startTwoSeckil(long redPacketId, int userId) {
        Integer money = 0;
        boolean res=false;
        try {
            /**
             * 獲取鎖 保證紅包數量和計算紅包金額的原子性操作
             */
            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介面中輸入任何引數進行測試，也可以配合資料庫稍加修改即可作為生產環境的搶紅包功能模組。

原始碼

https://gitee.com/52itstyle/spring-boot-seck