分布式锁实战-用户抢单

分布式锁实战-用户抢单

首先说下业务需求：订单产生以后，用户通过平台进行抢单，只有一个用户最终会拥有订单。

再来看一下项目架构：

首先用户统一调用接口服务 api-order ，再由 api-order 调用 (负载均衡) service-order 集群。

我们先看一下如果不加分布式锁会发生什么情况。

api-order 代码：

    @GetMapping("/acceptOrderByExpert")
    public String acceptOrderByExpert(String orderSn, String expertId) {
            String url = "http://service-order/order/acceptOrderByExpert";

            HttpHeaders headers = new HttpHeaders();
            headers.setContentType(MediaType.APPLICATION_JSON);
            AcceptOrderByExpertRequest para = new AcceptOrderByExpertRequest();
            para.setType(2);
            para.setOrder_sn(orderSn);
            para.setExpert_id(expertId);

            HttpEntity<AcceptOrderByExpertRequest> request = new HttpEntity<>(para, headers);

            Object o = restTemplate.postForEntity(url, request, String.class);

            return "success";
    }

这段代码非常简单，就是通过 RestTemplate 调用 service-order 的服务。

service-order 代码（此时还并没有加入分布式锁）：

    @Override
    public void acceptOrderByExpert(AcceptOrderByExpertRequest request) {
        // 判断订单是否存在
        Order order = orderMapper.findOrderByOrderSn(request.getOrder_sn());
        log.info("开始抢单");
        // 如果是异步的订单
        // 判断订单状态
        if (order.getStatus() != OrderStatus.UPLOAD.getCode()) {
            throw new BadRequestException("订单状态异常");
        }
        orderMapper.changeOrderStatus(request.getOrder_sn(), request.getExpert_id(), OrderStatus.REPORT.getCode());
        log.info("抢单成功");
    }

此处大体逻辑就是，我们查询订单状态，如果订单状态是未接单，那么就改变订单的状态，否则返回订单异常。这里我们如果改变订单状态成功，就在控制台打印一下抢单成功。

这里我们采用 JMeter 来模拟并发抢单。

我们先想一下，我们需要的结果是什么，肯定是只有一个用户抢单成功，所以我们在两个 service-order 的控制台应该只会看见一句 “抢单成功” 的打印日志，我们来看下实际的结果。

可以看到，不管是哪个 service-order 服务，都打印了 “抢单成功“ 的日志，表示不止一个用户抢到了订单，这便是分布式锁需要解决的问题。

这里我们使用 Redisson 来解决分布式锁的问题，Redisson 的原理和使用这里不再赘述，网上有很多相关文章。

需要改的代码只有 service-order：

    @Override
    public void acceptOrderByExpert(AcceptOrderByExpertRequest request) {
        log.info("开始抢单");

        // 生成 key 值
        String key = "order_" + request.getOrder_sn();

        // 获取锁
        RLock lock = redissonClient.getLock(key.intern());

        try {
            // 加锁
            lock.lock();
            // 判断订单是否存在
            Order order = orderMapper.findOrderByOrderSn(request.getOrder_sn());
            if (order == null) {
                throw new BadRequestException("订单不存在");
            }
            // 判断订单状态
            if (order.getStatus() != OrderStatus.UPLOAD.getCode()) {
                throw new BadRequestException("订单状态异常");
            }
            orderMapper.changeOrderStatus(request.getOrder_sn(), request.getExpert_id(), OrderStatus.REPORT.getCode());
        } finally {
            // 释放锁
            lock.unlock();
        }
        log.info("抢单成功");
    }

现在我们来看一下测试结果：

可以看到，现在两个服务当中就只有一个用户抢单成功啦！