分布式锁实战-用户抢单
首先说下业务需求:订单产生以后,用户通过平台进行抢单,只有一个用户最终会拥有订单。
再来看一下项目架构:
首先用户统一调用接口服务 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("抢单成功");
}
现在我们来看一下测试结果:
可以看到,现在两个服务当中就只有一个用户抢单成功啦!