Bootstrap

通过异步使用消息队列优化秒杀


同步秒杀流程

    public Result seckillVoucher(Long voucherId) throws InterruptedException {
        SeckillVoucher seckillVoucher = iSeckillVoucherService.getById(voucherId);
        LocalDateTime beginTime = seckillVoucher.getBeginTime();
        if (LocalDateTime.now().isBefore(beginTime)) {
            return Result.fail("秒杀还未开始");
        }
        LocalDateTime endTime = seckillVoucher.getEndTime();
        if (LocalDateTime.now().isAfter(endTime)) {
            return Result.fail("秒杀已经结束");
        }
//        查看库存
        Integer stock = seckillVoucher.getStock();
        if (stock < 1) {
            return Result.fail("库存不足");
        }
        Long userId = UserHolder.getUser().getId();
//        redis集群实现
        RLock lock = redissonClient.getLock("order:" + userId);
        boolean isLock = lock.tryLock(1L, TimeUnit.SECONDS);
        if (!isLock) {
            return Result.fail("你已经购买优惠卷");
        }
        try {
            IVoucherOrderService proxy = (IVoucherOrderService) AopContext.currentProxy();
            return proxy.createVoucherOrder(voucherId, userId);
        } finally {
//            redisLock.releaseLock();
            lock.unlock();
        }
    }
  @Transactional
    public Result createVoucherOrder(Long voucherId, Long userId) {
        Integer count = query().eq("user_id", userId).eq("voucher_id", voucherId).count();
        if (count > 0) {
            return Result.fail("已经购买优惠卷");
        }
//        基于乐观锁避免超卖问题,在乐观锁的基础上做出了改进,当要修改库存时判断库存是否大于0,但是也给数据库带来了巨大的压力
        boolean success = iSeckillVoucherService.update().setSql("stock=stock-1")
                .eq("voucher_id", voucherId).
                gt("stock", 0).update();
        if (!success) {
            return Result.fail("库存不足");
        }
        VoucherOrder voucherOrder = new VoucherOrder();
        long orderId = redisWorker.getGlobalId("order");
        voucherOrder.setId(orderId);
        voucherOrder.setVoucherId(voucherId);
        voucherOrder.setUserId(userId);
        save(voucherOrder);
        return Result.ok(orderId);
    }

以上是我们同步秒杀的流程,首先会从数据库中查询优惠卷,检查库存和是否在秒杀阶段,然后使用redisson基于Redis实现分布式锁解决一人一单问题,如果都通过后就会修改库存并且生成订单,整体的流程如下:在这里插入图片描述
在这个流程中,我们发现整个步骤都是同步,都是交给主线程来执行,负责检验库存保证一人一单然后来修改数据库数据,我们是否可以再请一个人来优化这个流程呢?比如A来负责检验,检验成功后交给B来完成数据库的修改,这样是不是能够优化秒杀的业务

异步优化秒杀

我们可以这样做,让主线程来负责判断这个人能不能抢杀,如果抢杀成功交给另一个人来负责修改数据库,主线程直接返回订单id。
那怎样让主线程快速的库存和一人一单的判断呢?Redis
我们可以将库存保存在Redis的字符串类型中,Redis的性能高于直接查询数据库的性能,而一人一单的校验我们可以使用Redis的set集合,校验是否一人一单就看Redis的set集合中是否含有用户id,而当主线程通过校验后让Redis中的库存-1并且将用户id添加到set集合中。
而我们怎样让修改数据库的任务交给另一个人来处理呢?使用阻塞队列或者消息队列,为了实现简单,暂且使用阻塞队列,我们将要修改数据库的任务放入阻塞队列中,创建一个线程池,让线程池来负责执行修改数据库的任务。

异步秒杀流程

在这里插入图片描述
从流程中看出我们主线程只负责进行库存校验和确保一人一单后,生成订单添加到阻塞队列或消息队列中就直接返回,大大提高了业务的性能。

基于lua脚本保证Redis操作原子性

如果我们使用以上的操作来实现代码,会发生并发安全问题,因为校验Redis操作和更新Redis操作不具有原子性,就可能发生问题,例如一个线程通过校验Redis库存等还没有进行修改,另一个线程直接进入比较没有修改的库存,就发生了线程安全问题,我们需要保证这些操作具有原子性

代码实现

lua脚本:

--优惠卷id
local voucherId = ARGV[1];
--用户id
local userId = ARGV[2];
--库存键
local stockKey = "voucher:stock:" + voucherId;
--下过订单的用户键
local orderKey = "voucher:order:" + voucherId;
--查看卷库存是否足够
if (tonumber(redis.call("get", stockKey)) <= 0) then
return 1;
end
if (redis.call("sismember",orderKey,userId)==1) then
return 2
end
--扣库存
redis.call("incrby",stockKey,-1);
--添加用户id
redis.call("sadd",orderKey,userId)
return 0

    /**
     * 异步秒杀
     */
    @Override
    public Result seckillVoucher(Long voucherId) throws InterruptedException {
        Long userId = UserHolder.getUser().getId();
        Long result = redisTemplate.execute(SECkILL_SCRIPT, Collections.emptyList(), voucherId.toString(), userId.toString());
        int re = result.intValue();
        if (re != 0) {
            return re == 1 ? Result.fail("库存不足") : Result.fail("不能重复下单");
        }
        long orderId = redisWorker.getGlobalId("voucher:order");
        VoucherOrder voucherOrder = new VoucherOrder();
        voucherOrder.setVoucherId(voucherId);
        voucherOrder.setId(orderId);
        voucherOrder.setUserId(userId);
        VOUCHER_PROXY = (IVoucherOrderService) AopContext.currentProxy();
        BLOCKING_QUEUE.add(voucherOrder);
        return Result.ok(orderId);
    }

// 初始化lua脚本环境
    private static final DefaultRedisScript<Long> SECkILL_SCRIPT;


    static {
        SECkILL_SCRIPT = new DefaultRedisScript<>();
        SECkILL_SCRIPT.setLocation(new ClassPathResource("seckill.lua"));
        SECkILL_SCRIPT.setResultType(Long.class);
    }

    //    在类初始化完之后执行
    @PostConstruct
    private void init() {
        EXECUTOR_SERVICE.execute(new VoucherHandler());
    }
// 阻塞队列获取任务处理
    private class VoucherHandler implements Runnable {

        @Override
        public void run() {
            while (true){
                try {
//                从阻塞队列中获取订单信息
                    VoucherOrder voucherOrder = BLOCKING_QUEUE.take();
//                  更新数据库库存以及创建优惠卷订单
                    VOUCHER_PROXY.createVoucherOrder(voucherOrder.getVoucherId(), voucherOrder.getUserId());
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }
    }
    @Transactional
    public Result createVoucherOrder(Long voucherId, Long userId) {
        Integer count = query().eq("user_id", userId).eq("voucher_id", voucherId).count();
        if (count > 0) {
            return Result.fail("已经购买优惠卷");
        }
//        基于乐观锁避免超卖问题,在乐观锁的基础上做出了改进,当要修改库存时判断库存是否大于0,但是也给数据库带来了巨大的压力
        boolean success = iSeckillVoucherService.update().setSql("stock=stock-1")
                .eq("voucher_id", voucherId).
                gt("stock", 0).update();
        if (!success) {
            return Result.fail("库存不足");
        }
        VoucherOrder voucherOrder = new VoucherOrder();
        long orderId = redisWorker.getGlobalId("order");
        voucherOrder.setId(orderId);
        voucherOrder.setVoucherId(voucherId);
        voucherOrder.setUserId(userId);
        save(voucherOrder);
        return Result.ok(orderId);
    }

阻塞队列的缺点

在我们进行数据库的修改任务时我们使用了阻塞队列来实现,在实际的业务中,我们需要使用一些消息队列来代替阻塞队列,阻塞队列使用的是JVM中的内存,当消息过多时会造成JVM内存爆满,并且功能不够强大,我们可以使用Redis的Stream或者MQ来实现消息队列来代替阻塞队列

;