秒殺系統如何保證數據庫不崩潰以及防止商品超賣

2022 年 7 月 18 日
筆記
JAVA, Redis, springboot, 秒殺

1、應用場景

電商商城，商家上架了一個秒殺活動，早上10點開始，商品A參與秒殺，一共有20個庫存，預計10W的人去搶。

2、面臨問題

高並發、庫存不可超賣

3、問題解決

1）高並發，我們不能把所有的請求都去數據庫查商品詳情，查商品庫存，這樣數據庫會頂不住，很容易的我們就想到了用Redis解決；

2）庫存超賣問題，這個問題主要是由於用戶在同時讀取到的庫存均為大於0，從而認為我們該商品還沒被秒完，繼續創建了訂單，導致了商品超賣了。　

4、編碼實現　　

1、數據庫新建兩張表

秒殺訂單

CREATE TABLE `ms_order` (
  `ms_order_id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '訂單ID',
  `created_time` datetime DEFAULT NULL COMMENT '創建時間',
  `order_price` decimal(12,2) DEFAULT NULL COMMENT '訂單總價',
  `state` tinyint(1) DEFAULT '1' COMMENT '訂單狀態 1未支付 2已支付 3已發貨 4已收貨 -1已取消',
  `pay_time` datetime DEFAULT NULL COMMENT '支付時間',
  `fh_time` datetime DEFAULT NULL COMMENT '發貨時間',
  PRIMARY KEY (`ms_order_id`)
) ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8mb4 COMMENT='秒殺訂單';

秒殺商品

CREATE TABLE `ms_product` (
  `ms_product_id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '秒殺商品ID',
  `product_name` varchar(100) DEFAULT NULL COMMENT '商品名稱',
  `origin_price` decimal(12,2) DEFAULT NULL COMMENT '商品原價',
  `ms_price` decimal(12,2) DEFAULT NULL COMMENT '秒殺價',
  `product_img` varchar(255) DEFAULT NULL COMMENT '商品圖片',
  `state` tinyint(1) DEFAULT NULL COMMENT '商品狀態 1已上架 -1已下架',
  `product_summary` varchar(255) DEFAULT NULL COMMENT '商品描述',
  `product_details` text COMMENT '商品詳情',

 PRIMARY KEY (`ms_product_id`) 
) ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8mb4 COMMENT='秒殺商品';

2、設置商品庫存，正式的流程肯定是由後台添加商品時初始化，這邊為了方便，直接用Redis可視化工具插入了商品，秒殺商品ID為1的設置20個庫存，同時數據庫也要設置20個庫存，利於我們分析扣減庫存是否一致

3、敲代碼

1）寫一個下單接口

@PostMapping(value = "/add")
    public ResultMsg add(HttpServletRequest request, MsOrder msOrder,Long ms_product_id) {
        String interfaceName = "下單測試";
        try {
            User user = getUser();
            return new ResultMsg(true, msOrderService.insert(msOrder, user,ms_product_id));
        } catch (ServiceRuntimeException e) {
            return fail(e);
        } catch (Exception e) {
            return error(interfaceName, e, request);
        }
    }

2）邏輯處理

利用lua腳本減庫存，lua腳本如下

local isExist = redis.call('exists', KEYS[1]);
if (tonumber(isExist) > 0) then
    local goodsNumber = redis.call('get', KEYS[1]);
    if (tonumber(goodsNumber) > 0) then
        redis.call('decr',KEYS[1]);
        return 1;
    else
        redis.call('del', KEYS[1]);
        return 0;
        end;
else
return -1;
end;

lua配置類

@Configuration
public class LuaConfiguration {
    @Bean
    public DefaultRedisScript<Long> redisScript() {
        DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();
        redisScript.setScriptSource(new ResourceScriptSource(new ClassPathResource("script/Stock.lua")));
        redisScript.setResultType(Long.class);
        return redisScript;
    }

}

扣減Redis中對應的商品庫存

@Component
public class LuaReduceStock {

    @Resource
    private DefaultRedisScript<Long> redisScript;
    @Resource
    private StringRedisTemplate stringRedisTemplate;

    /**
     * 減庫存
     * @param key
     * @return
     */
    public boolean reduceStock(String key){
        List<String> keys = new ArrayList<>();
        keys.add(key);

        Long result = stringRedisTemplate.execute(redisScript,keys,"100");
        return result  > 0;
    }
}

業務處理

public boolean insert(MsOrder msOrder, User user,Long ms_product_id){
        Assert.notNull(ms_product_id,"購買商品不能為空");

        boolean b = luaReduceStock.reduceStock(RedisConstants.MSSTOCK+ms_product_id);
        if(b){
            //最終搶到庫存的用戶，可以發送一條消息到隊列中，進行異步下單扣減庫存等。
            Map map = new HashMap();
            map.put("ms_product_id",ms_product_id);
            amqpTemplate.convertAndSend(RabbitConstants.MS_QUEUE,map);
            return true;
        }else{
            serviceError("手慢了，商品已被搶光啦！！！");
        }
        return true;
    }

異步下單，扣減庫存

@Component
@RabbitListener(queues = RabbitConstants.MS_QUEUE)
public class MsOrderHandler {


    @Autowired
    MsProductService msProductService;
    @Resource
    MsProductMapper msProductMapper;
    @Resource
    MsOrderMapper msOrderMapper;

    @RabbitHandler
    public void send(Map map){
        try{
            Long ms_product_id = Long.valueOf(map.get("ms_product_id").toString());
            MsProductDTO msProductDTO = msProductService.findById(ms_product_id);
            MsOrder msOrder = new MsOrder();
            msOrder.setCreated_time(new Date());
            msOrder.setOrder_price(msProductDTO.getMs_price());
            msOrder.setState(1);
            msOrderMapper.insert(msOrder);

            MsProduct msProduct = new MsProduct();
            msProduct.setStock(-1);
            msProduct.setMs_product_id(ms_product_id);
            msProductMapper.updateStock(msProduct);
        }catch (Exception e){
            e.printStackTrace();
        }

    }
}