分布式锁实战-用户抢单

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

再来看一下项目架构:

java 抢单设计 面试 抢单程序怎么设计_java

首先用户统一调用接口服务 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 来模拟并发抢单。

java 抢单设计 面试 抢单程序怎么设计_java_02

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

java 抢单设计 面试 抢单程序怎么设计_架构_03

java 抢单设计 面试 抢单程序怎么设计_java_04

可以看到,不管是哪个 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("抢单成功");
    }

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

java 抢单设计 面试 抢单程序怎么设计_java 抢单设计 面试_05

java 抢单设计 面试 抢单程序怎么设计_架构_06

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