写点什么

RabbitMQ 高级使用

  • 2025-05-13
    福建
  • 本文字数:18492 字

    阅读完需:约 61 分钟

概述


在支付场景中,支付成功后利用 RabbitMQ 通知交易服务,更新业务订单状态为已支付。但是大家思考一下,如果这里 MQ 通知失败,支付服务中支付流水显示支付成功,而交易服务中的订单状态却显示未支付,数据出现了不一致。此时前端发送请求查询支付状态时,肯定是查询交易服务状态,会发现业务订单未支付,而用户自己知道已经支付成功,这就导致用户体验不一致。


因此,这里必须尽可能确保 MQ 消息的可靠性,即:消息应该至少被消费者处理 1 次


那么问题来了:

  • 该如何确保 MQ 消息的可靠性

  • 如果真的发送失败,有没有其它的兜底方案?


这些问题,在今天的学习中都会找到答案。

首先,一起分析一下消息丢失的可能性有哪些。消息从发送者发送消息,到消费者处理消息,需要经过的流程是这样的:



消息从生产者到消费者的每一步都可能导致消息丢失:

  • 发送消息时丢失:生产者发送消息时连接 MQ 失败生产者发送消息到达 MQ 后未找到Exchange生产者发送消息到达 MQ 的Exchange后,未找到合适的Queue消息到达 MQ 后,处理消息的进程发生异常

  • MQ 导致消息丢失:消息到达 MQ,保存到队列后,尚未消费就突然宕机

  • 消费者处理消息时:消息接收后尚未处理突然宕机消息接收后处理过程中抛出异常


综上,我们要解决消息丢失问题,保证 MQ 的可靠性,就必须从 3 个方面入手:

  • 确保生产者一定把消息发送到 MQ

  • 确保 MQ 不会将消息弄丢

  • 确保消费者一定要处理消息


发送者的可靠性


生产者重试机制


首先第一种情况,就是生产者发送消息时,出现了网络故障,导致与 MQ 的连接中断。


为了解决这个问题,SpringAMQP 提供的消息发送时的重试机制。即:当RabbitTemplate与 MQ 连接超时后,多次重试。


修改publisher模块的application.yaml文件,添加下面的内容:


spring:  rabbitmq:    connection-timeout: 1s # 设置MQ的连接超时时间    template:      retry:        enabled: true # 开启超时重试机制        initial-interval: 1000ms # 失败后的初始等待时间        multiplier: 1 # 失败后下次的等待时长倍数,下次等待时长 = initial-interval * multiplier        max-attempts: 3 # 最大重试次数
复制代码


我们利用命令停掉 RabbitMQ 服务:


docker stop mq
复制代码


然后测试发送一条消息,会发现会每隔 1 秒重试 1 次,总共重试了 3 次。消息发送的超时重试机制配置成功了!


注意:当网络不稳定的时候,利用重试机制可以有效提高消息发送的成功率。不过 SpringAMQP 提供的重试机制是阻塞式的重试,也就是说多次重试等待的过程中,当前线程是被阻塞的。如果对于业务性能有要求,建议禁用重试机制。如果一定要使用,请合理配置等待时长和重试次数,当然也可以考虑使用异步线程来执行发送消息的代码。


生产者确认机制


一般情况下,只要生产者与 MQ 之间的网路连接顺畅,基本不会出现发送消息丢失的情况,因此大多数情况下我们无需考虑这种问题。不过,在少数情况下,也会出现消息发送到 MQ 之后丢失的现象,比如:


  • MQ 内部处理消息的进程发生了异常

  • 生产者发送消息到达 MQ 后未找到Exchange

  • 生产者发送消息到达 MQ 的Exchange后,未找到合适的Queue,因此无法路由


针对上述情况,RabbitMQ 提供了生产者消息确认机制,包括Publisher ConfirmPublisher Return两种。在开启确认机制的情况下,当生产者发送消息给 MQ 后,MQ 会根据消息处理的情况返回不同的回执。具体如图所示:



总结如下:

  • 当消息投递到 MQ,但是路由失败时,通过 Publisher Return 返回异常信息,同时返回 ack 的确认信息,代表投递成功

  • 临时消息投递到了 MQ,并且入队成功,返回 ACK,告知投递成功

  • 持久消息投递到了 MQ,并且入队完成持久化,返回 ACK ,告知投递成功

  • 其它情况都会返回 NACK,告知投递失败


其中acknack属于 Publisher Confirm 机制,ack是投递成功;nack是投递失败。而return则属于 Publisher Return 机制。默认两种机制都是关闭状态,需要通过配置文件来开启。


实现生产者确认


开启生产者确认


在 publisher 模块的application.yaml中添加配置:


spring:  rabbitmq:    publisher-confirm-type: correlated # 开启publisher confirm机制,并设置confirm类型    publisher-returns: true # 开启publisher return机制
复制代码


这里publisher-confirm-type有三种模式可选:

  • none:关闭 confirm 机制

  • simple:同步阻塞等待 MQ 的回执

  • correlated:MQ 异步回调返回回执

一般我们推荐使用correlated,回调机制。


定义 ReturnCallback


每个RabbitTemplate只能配置一个ReturnCallback,因此我们可以在配置类中统一设置。我们在 publisher 模块定义一个配置类:



内容如下:


package com.itheima.publisher.config;
import lombok.AllArgsConstructor;import lombok.extern.slf4j.Slf4j;import org.springframework.amqp.core.ReturnedMessage;import org.springframework.amqp.rabbit.core.RabbitTemplate;import org.springframework.context.annotation.Configuration;
import javax.annotation.PostConstruct;
@Slf4j@AllArgsConstructor@Configurationpublic class MqConfig { private final RabbitTemplate rabbitTemplate;
@PostConstruct public void init(){ rabbitTemplate.setReturnsCallback(new RabbitTemplate.ReturnsCallback() { @Override public void returnedMessage(ReturnedMessage returned) { log.error("触发return callback,"); log.debug("exchange: {}", returned.getExchange()); log.debug("routingKey: {}", returned.getRoutingKey()); log.debug("message: {}", returned.getMessage()); log.debug("replyCode: {}", returned.getReplyCode()); log.debug("replyText: {}", returned.getReplyText()); } }); }}
复制代码


定义 ConfirmCallback


由于每个消息发送时的处理逻辑不一定相同,因此 ConfirmCallback 需要在每次发消息时定义。具体来说,是在调用 RabbitTemplate 中的 convertAndSend 方法时,多传递一个参数:



这里的 CorrelationData 中包含两个核心的东西:


  • id:消息的唯一标示,MQ 对不同的消息的回执以此做判断,避免混淆

  • SettableListenableFuture:回执结果的 Future 对象


将来 MQ 的回执就会通过这个Future来返回,我们可以提前给CorrelationData中的Future添加回调函数来处理消息回执:



我们新建一个测试,向系统自带的交换机发送消息,并且添加ConfirmCallback


@Testvoid testPublisherConfirm() {    // 1.创建CorrelationData    CorrelationData cd = new CorrelationData();    // 2.给Future添加ConfirmCallback    cd.getFuture().addCallback(new ListenableFutureCallback<CorrelationData.Confirm>() {        @Override        public void onFailure(Throwable ex) {            // 2.1.Future发生异常时的处理逻辑,基本不会触发            log.error("send message fail", ex);        }        @Override        public void onSuccess(CorrelationData.Confirm result) {            // 2.2.Future接收到回执的处理逻辑,参数中的result就是回执内容            if(result.isAck()){ // result.isAck(),boolean类型,true代表ack回执,false 代表 nack回执                log.debug("发送消息成功,收到 ack!");            }else{ // result.getReason(),String类型,返回nack时的异常描述                log.error("发送消息失败,收到 nack, reason : {}", result.getReason());            }        }    });    // 3.发送消息    rabbitTemplate.convertAndSend("hmall.direct", "q", "hello", cd);}
复制代码


执行结果如下:



可以看到,由于传递的RoutingKey是错误的,路由失败后,触发了return callback,同时也收到了 ack。当我们修改为正确的RoutingKey以后,就不会触发return callback了,只收到 ack。而如果连交换机都是错误的,则只会收到 nack。


注意:开启生产者确认比较消耗 MQ 性能,一般不建议开启。而且大家思考一下触发确认的几种情况:

  • 路由失败:一般是因为 RoutingKey 错误导致,往往是编程导致

  • 交换机名称错误:同样是编程错误导致

  • MQ 内部故障:这种需要处理,但概率往往较低。因此只有对消息可靠性要求非常高的业务才需要开启,而且仅仅需要开启 ConfirmCallback 处理 nack 就可以了。


MQ 的可靠性


消息到达 MQ 以后,如果 MQ 不能及时保存,也会导致消息丢失,所以 MQ 的可靠性也非常重要。


数据持久化


为了提升性能,默认情况下 MQ 的数据都是在内存存储的临时数据,重启后就会消失。为了保证数据的可靠性,必须配置数据持久化,包括:

  • 交换机持久化

  • 队列持久化

  • 消息持久化

我们以控制台界面为例来说明。


交换机持久化


在控制台的Exchanges页面,添加交换机时可以配置交换机的Durability参数:



设置为Durable就是持久化模式,Transient就是临时模式。


队列持久化


在控制台的 Queues 页面,添加队列时,同样可以配置队列的Durability参数:



除了持久化以外,你可以看到队列还有很多其它参数,有一些我们会在后期学习。


消息持久化


在控制台发送消息的时候,可以添加很多参数,而消息的持久化是要配置一个properties



说明:在开启持久化机制以后,如果同时还开启了生产者确认,那么 MQ 会在消息持久化以后才发送 ACK 回执,进一步确保消息的可靠性。不过出于性能考虑,为了减少 IO 次数,发送到 MQ 的消息并不是逐条持久化到数据库的,而是每隔一段时间批量持久化。一般间隔在 100 毫秒左右,这就会导致 ACK 有一定的延迟,因此建议生产者确认全部采用异步方式。


LazyQueue


在默认情况下,RabbitMQ 会将接收到的信息保存在内存中以降低消息收发的延迟。但在某些特殊情况下,这会导致消息积压,比如:


  • 消费者宕机或出现网络故障

  • 消息发送量激增,超过了消费者处理速度

  • 消费者处理业务发生阻塞


一旦出现消息堆积问题,RabbitMQ 的内存占用就会越来越高,直到触发内存预警上限。此时 RabbitMQ 会将内存消息刷到磁盘上,这个行为成为PageOutPageOut会耗费一段时间,并且会阻塞队列进程。因此在这个过程中 RabbitMQ 不会再处理新的消息,生产者的所有请求都会被阻塞。


为了解决这个问题,从 RabbitMQ 的 3.6.0 版本开始,就增加了 Lazy Queues 的模式,也就是惰性队列。惰性队列的特征如下:


  • 接收到消息后直接存入磁盘而非内存

  • 消费者要消费消息时才会从磁盘中读取并加载到内存(也就是懒加载)

  • 支持数百万条的消息存储


而在 3.12 版本之后,LazyQueue 已经成为所有队列的默认格式。因此官方推荐升级 MQ 为 3.12 版本或者所有队列都设置为 LazyQueue 模式。


控制台配置 Lazy 模式


在添加队列的时候,添加x-queue-mod=lazy参数即可设置队列为 Lazy 模式:



代码配置 Lazy 模式


在利用 SpringAMQP 声明队列的时候,添加x-queue-mod=lazy参数也可设置队列为 Lazy 模式:


@Beanpublic Queue lazyQueue(){    return QueueBuilder            .durable("lazy.queue")            .lazy() // 开启Lazy模式            .build();}
复制代码


这里是通过QueueBuilderlazy()函数配置 Lazy 模式,底层源码如下:



当然,我们也可以基于注解来声明队列并设置为 Lazy 模式:


@RabbitListener(queuesToDeclare = @Queue(        name = "lazy.queue",        durable = "true",        arguments = @Argument(name = "x-queue-mode", value = "lazy")))public void listenLazyQueue(String msg){    log.info("接收到 lazy.queue的消息:{}", msg);}
复制代码


更新已有队列为 lazy 模式


对于已经存在的队列,也可以配置为 lazy 模式,但是要通过设置 policy 实现。可以基于命令行设置 policy:


rabbitmqctl set_policy Lazy "^lazy-queue$" '{"queue-mode":"lazy"}' --apply-to queues  
复制代码


命令解读:

  • rabbitmqctl :RabbitMQ 的命令行工具

  • set_policy :添加一个策略

  • Lazy :策略名称,可以自定义

  • "^lazy-queue$" :用正则表达式匹配队列的名字

  • '{"queue-mode":"lazy"}' :设置队列模式为 lazy 模式

  • --apply-to queues:策略的作用对象,是所有的队列

当然,也可以在控制台配置 policy,进入在控制台的Admin页面,点击Policies,即可添加配置:



消费者的可靠性


当 RabbitMQ 向消费者投递消息以后,需要知道消费者的处理状态如何。因为消息投递给消费者并不代表就一定被正确消费了,可能出现的故障有很多,比如:


  • 消息投递的过程中出现了网络故障

  • 消费者接收到消息后突然宕机

  • 消费者接收到消息后,因处理不当导致异常

  • ...


一旦发生上述情况,消息也会丢失。因此,RabbitMQ 必须知道消费者的处理状态,一旦消息处理失败才能重新投递消息。但问题来了:RabbitMQ 如何得知消费者的处理状态呢?


本章我们就一起研究一下消费者处理消息时的可靠性解决方案。


消费者确认机制


为了确认消费者是否成功处理消息,RabbitMQ 提供了消费者确认机制(Consumer Acknowledgement)。即:当消费者处理消息结束后,应该向 RabbitMQ 发送一个回执,告知 RabbitMQ 自己消息处理状态。回执有三种可选值:

  • ack:成功处理消息,RabbitMQ 从队列中删除该消息

  • nack:消息处理失败,RabbitMQ 需要再次投递消息

  • reject:消息处理失败并拒绝该消息,RabbitMQ 从队列中删除该消息


一般 reject 方式用的较少,除非是消息格式有问题,那就是开发问题了。因此大多数情况下我们需要将消息处理的代码通过try catch机制捕获,消息处理成功时返回 ack,处理失败时返回 nack.

由于消息回执的处理代码比较统一,因此 SpringAMQP 帮我们实现了消息确认。并允许我们通过配置文件设置 ACK 处理方式,有三种模式:

  • **none**:不处理。即消息投递给消费者后立刻 ack,消息会立刻从 MQ 删除。非常不安全,不建议使用

  • **manual**:手动模式。需要自己在业务代码中调用 api,发送ackreject,存在业务入侵,但更灵活

  • **auto**:自动模式。SpringAMQP 利用 AOP 对我们的消息处理逻辑做了环绕增强,当业务正常执行时则自动返回ack. 当业务出现异常时,根据异常判断返回不同结果:如果是业务异常,会自动返回nack;如果是消息处理或校验异常,自动返回reject;


返回 Reject 的常见异常有:


Starting with version 1.3.2, the default ErrorHandler is now a ConditionalRejectingErrorHandler that rejects (and does not requeue) messages that fail with an irrecoverable error. Specifically, it rejects messages that fail with the following errors:

  • o.s.amqp…MessageConversionException: Can be thrown when converting the incoming message payload using a MessageConverter.

  • o.s.messaging…MessageConversionException: Can be thrown by the conversion service if additional conversion is required when mapping to a @RabbitListener method.

  • o.s.messaging…MethodArgumentNotValidException: Can be thrown if validation (for example, @Valid) is used in the listener and the validation fails.

  • o.s.messaging…MethodArgumentTypeMismatchException: Can be thrown if the inbound message was converted to a type that is not correct for the target method. For example, the parameter is declared as Message<Foo> but Message<Bar> is received.

  • java.lang.NoSuchMethodException: Added in version 1.6.3.

  • java.lang.ClassCastException: Added in version 1.6.3.


通过下面的配置可以修改 SpringAMQP 的 ACK 处理方式:


spring:  rabbitmq:    listener:      simple:        acknowledge-mode: none # 不做处理
复制代码


修改 consumer 服务的 SpringRabbitListener 类中的方法,模拟一个消息处理的异常:


@RabbitListener(queues = "simple.queue")public void listenSimpleQueueMessage(String msg) throws InterruptedException {    log.info("spring 消费者接收到消息:【" + msg + "】");    if (true) {        throw new MessageConversionException("故意的");    }    log.info("消息处理完成");}
复制代码


测试可以发现:当消息处理发生异常时,消息依然被 RabbitMQ 删除了。


我们再次把确认机制修改为 auto:


spring:  rabbitmq:    listener:      simple:        acknowledge-mode: auto # 自动ack
复制代码


在异常位置打断点,再次发送消息,程序卡在断点时,可以发现此时消息状态为unacked(未确定状态):



放行以后,由于抛出的是消息转换异常,因此 Spring 会自动返回reject,所以消息依然会被删除:



我们将异常改为 RuntimeException 类型:


@RabbitListener(queues = "simple.queue")public void listenSimpleQueueMessage(String msg) throws InterruptedException {    log.info("spring 消费者接收到消息:【" + msg + "】");    if (true) {        throw new RuntimeException("故意的");    }    log.info("消息处理完成");}
复制代码


在异常位置打断点,然后再次发送消息测试,程序卡在断点时,可以发现此时消息状态为unacked(未确定状态):



放行以后,由于抛出的是业务异常,所以 Spring 返回ack,最终消息恢复至Ready状态,并且没有被 RabbitMQ 删除:



当我们把配置改为auto时,消息处理失败后,会回到 RabbitMQ,并重新投递到消费者。


失败重试机制


当消费者出现异常后,消息会不断 requeue(重入队)到队列,再重新发送给消费者。如果消费者再次执行依然出错,消息会再次 requeue 到队列,再次投递,直到消息处理成功为止。极端情况就是消费者一直无法执行成功,那么消息 requeue 就会无限循环,导致 mq 的消息处理飙升,带来不必要的压力:



当然,上述极端情况发生的概率还是非常低的,不过不怕一万就怕万一。为了应对上述情况 Spring 又提供了消费者失败重试机制:在消费者出现异常时利用本地重试,而不是无限制的 requeue 到 mq 队列。

修改 consumer 服务的 application.yml 文件,添加内容:


spring:  rabbitmq:    listener:      simple:        retry:          enabled: true # 开启消费者失败重试          initial-interval: 1000ms # 初识的失败等待时长为1秒          multiplier: 1 # 失败的等待时长倍数,下次等待时长 = multiplier * last-interval          max-attempts: 3 # 最大重试次数          stateless: true # true无状态;false有状态。如果业务中包含事务,这里改为false
复制代码


重启 consumer 服务,重复之前的测试。可以发现:

  • 消费者在失败后消息没有重新回到 MQ 无限重新投递,而是在本地重试了 3 次

  • 本地重试 3 次以后,抛出了AmqpRejectAndDontRequeueException异常。查看 RabbitMQ 控制台,发现消息被删除了,说明最后 SpringAMQP 返回的是reject


结论:

  • 开启本地重试时,消息处理过程中抛出异常,不会 requeue 到队列,而是在消费者本地重试

  • 重试达到最大次数后,Spring 会返回 reject,消息会被丢弃


失败处理策略


在之前的测试中,本地测试达到最大重试次数后,消息会被丢弃。这在某些对于消息可靠性要求较高的业务场景下,显然不太合适了。因此 Spring 允许我们自定义重试次数耗尽后的消息处理策略,这个策略是由MessageRecovery接口来定义的,它有 3 个不同实现:


  • RejectAndDontRequeueRecoverer:重试耗尽后,直接reject,丢弃消息。默认就是这种方式

  • ImmediateRequeueMessageRecoverer:重试耗尽后,返回nack,消息重新入队

  • RepublishMessageRecoverer:重试耗尽后,将失败消息投递到指定的交换机


比较优雅的一种处理方案是RepublishMessageRecoverer,失败后将消息投递到一个指定的,专门存放异常消息的队列,后续由人工集中处理。


1)在 consumer 服务中定义处理失败消息的交换机和队列


@Beanpublic DirectExchange errorMessageExchange(){    return new DirectExchange("error.direct");}@Beanpublic Queue errorQueue(){    return new Queue("error.queue", true);}@Beanpublic Binding errorBinding(Queue errorQueue, DirectExchange errorMessageExchange){    return BindingBuilder.bind(errorQueue).to(errorMessageExchange).with("error");}
复制代码


2)定义一个 RepublishMessageRecoverer,关联队列和交换机


@Beanpublic MessageRecoverer republishMessageRecoverer(RabbitTemplate rabbitTemplate){    return new RepublishMessageRecoverer(rabbitTemplate, "error.direct", "error");}
复制代码


完整代码如下:


package com.itheima.consumer.config;
import org.springframework.amqp.core.Binding;import org.springframework.amqp.core.BindingBuilder;import org.springframework.amqp.core.DirectExchange;import org.springframework.amqp.core.Queue;import org.springframework.amqp.rabbit.core.RabbitTemplate;import org.springframework.amqp.rabbit.retry.MessageRecoverer;import org.springframework.amqp.rabbit.retry.RepublishMessageRecoverer;import org.springframework.context.annotation.Bean;
@Configuration@ConditionalOnProperty(name = "spring.rabbitmq.listener.simple.retry.enabled", havingValue = "true")public class ErrorMessageConfig { @Bean public DirectExchange errorMessageExchange(){ return new DirectExchange("error.direct"); } @Bean public Queue errorQueue(){ return new Queue("error.queue", true); } @Bean public Binding errorBinding(Queue errorQueue, DirectExchange errorMessageExchange){ return BindingBuilder.bind(errorQueue).to(errorMessageExchange).with("error"); }
@Bean public MessageRecoverer republishMessageRecoverer(RabbitTemplate rabbitTemplate){ return new RepublishMessageRecoverer(rabbitTemplate, "error.direct", "error"); }}
复制代码


业务幂等性


何为幂等性?幂等是一个数学概念,用函数表达式来描述是这样的:f(x) = f(f(x)),例如求绝对值函数。在程序开发中,则是指同一个业务,执行一次或多次对业务状态的影响是一致的。例如:

  • 根据 id 删除数据

  • 查询数据

  • 新增数据


但数据的更新往往不是幂等的,如果重复执行可能造成不一样的后果。比如:

  • 取消订单,恢复库存的业务。如果多次恢复就会出现库存重复增加的情况

  • 退款业务。重复退款对商家而言会有经济损失。


所以,我们要尽可能避免业务被重复执行。然而在实际业务场景中,由于意外经常会出现业务被重复执行的情况,例如:

  • 页面卡顿时频繁刷新导致表单重复提交

  • 服务间调用的重试

  • MQ 消息的重复投递


我们在用户支付成功后会发送 MQ 消息到交易服务,修改订单状态为已支付,就可能出现消息重复投递的情况。如果消费者不做判断,很有可能导致消息被消费多次,出现业务故障。举例:

  1. 假如用户刚刚支付完成,并且投递消息到交易服务,交易服务更改订单为已支付状态。

  2. 由于某种原因,例如网络故障导致生产者没有得到确认,隔了一段时间后重新投递给交易服务。

  3. 但是,在新投递的消息被消费之前,用户选择了退款,将订单状态改为了已退款状态。

  4. 退款完成后,新投递的消息才被消费,那么订单状态会被再次改为已支付。业务异常。


因此,我们必须想办法保证消息处理的幂等性。这里给出两种方案:

  • 唯一消息 ID

  • 业务状态判断


唯一消息 ID


这个思路非常简单:

  1. 每一条消息都生成一个唯一的 id,与消息一起投递给消费者。

  2. 消费者接收到消息后处理自己的业务,业务处理成功后将消息 ID 保存到数据库

  3. 如果下次又收到相同消息,去数据库查询判断是否存在,存在则为重复消息放弃处理。


我们该如何给消息添加唯一 ID 呢?其实很简单,SpringAMQP 的 MessageConverter 自带了 MessageID 的功能,我们只要开启这个功能即可。以 Jackson 的消息转换器为例:


@Beanpublic MessageConverter messageConverter(){    // 1.定义消息转换器    Jackson2JsonMessageConverter jjmc = new Jackson2JsonMessageConverter();    // 2.配置自动创建消息id,用于识别不同消息,也可以在业务中基于ID判断是否是重复消息    jjmc.setCreateMessageIds(true);    return jjmc;}
复制代码


业务判断


业务判断就是基于业务本身的逻辑或状态来判断是否是重复的请求或消息,不同的业务场景判断的思路也不一样。例如我们当前案例中,处理消息的业务逻辑是把订单状态从未支付修改为已支付。因此我们就可以在执行业务时判断订单状态是否是未支付,如果不是则证明订单已经被处理过,无需重复处理。

相比较而言,消息 ID 的方案需要改造原有的数据库,所以我更推荐使用业务判断的方案。

以支付修改订单的业务为例,我们需要修改OrderServiceImpl中的markOrderPaySuccess方法:


    @Override    public void markOrderPaySuccess(Long orderId) {        // 1.查询订单        Order old = getById(orderId);        // 2.判断订单状态        if (old == null || old.getStatus() != 1) {            // 订单不存在或者订单状态不是1,放弃处理            return;        }        // 3.尝试更新订单        Order order = new Order();        order.setId(orderId);        order.setStatus(2);        order.setPayTime(LocalDateTime.now());        updateById(order);    }
复制代码


上述代码逻辑上符合了幂等判断的需求,但是由于判断和更新是两步动作,因此在极小概率下可能存在线程安全问题。


我们可以合并上述操作为这样:


@Overridepublic void markOrderPaySuccess(Long orderId) {    // UPDATE `order` SET status = ? , pay_time = ? WHERE id = ? AND status = 1    lambdaUpdate()            .set(Order::getStatus, 2)            .set(Order::getPayTime, LocalDateTime.now())            .eq(Order::getId, orderId)            .eq(Order::getStatus, 1)            .update();}
复制代码


注意看,上述代码等同于这样的 SQL 语句:


UPDATE `order` SET status = ? , pay_time = ? WHERE id = ? AND status = 1
复制代码


我们在 where 条件中除了判断 id 以外,还加上了 status 必须为 1 的条件。如果条件不符(说明订单已支付),则 SQL 匹配不到数据,根本不会执行。


兜底方案


虽然我们利用各种机制尽可能增加了消息的可靠性,但也不好说能保证消息 100%的可靠。万一真的 MQ 通知失败该怎么办呢?有没有其它兜底方案,能够确保订单的支付状态一致呢?


其实思想很简单:既然 MQ 通知不一定发送到交易服务,那么交易服务就必须自己主动去查询支付状态。这样即便支付服务的 MQ 通知失败,我们依然能通过主动查询来保证订单状态的一致。流程如下:



图中黄色线圈起来的部分就是 MQ 通知失败后的兜底处理方案,由交易服务自己主动去查询支付状态。

不过需要注意的是,交易服务并不知道用户会在什么时候支付,如果查询的时机不正确(比如查询的时候用户正在支付中),可能查询到的支付状态也不正确。那么问题来了,我们到底该在什么时间主动查询支付状态呢?


这个时间是无法确定的,因此,通常我们采取的措施就是利用定时任务定期查询,例如每隔 20 秒就查询一次,并判断支付状态。如果发现订单已经支付,则立刻更新订单状态为已支付即可。定时任务大家之前学习过,具体的实现这里就不再赘述了。


至此,消息可靠性的问题已经解决了。

综上,支付服务与交易服务之间的订单状态一致性是如何保证的?

  • 首先,支付服务会正在用户支付成功以后利用 MQ 消息通知交易服务,完成订单状态同步。

  • 其次,为了保证 MQ 消息的可靠性,我们采用了生产者确认机制、消费者确认、消费者失败重试等策略,确保消息投递的可靠性

  • 最后,我们还在交易服务设置了定时任务,定期查询订单支付状态。这样即便 MQ 通知失败,还可以利用定时任务作为兜底方案,确保订单支付状态的最终一致性。


延迟消息


在电商的支付业务中,对于一些库存有限的商品,为了更好的用户体验,通常都会在用户下单时立刻扣减商品库存。例如电影院购票、高铁购票,下单后就会锁定座位资源,其他人无法重复购买。


但是这样就存在一个问题,假如用户下单后一直不付款,就会一直占有库存资源,导致其他客户无法正常交易,最终导致商户利益受损!


因此,电商中通常的做法就是:对于超过一定时间未支付的订单,应该立刻取消订单并释放占用的库存

例如,订单支付超时时间为 30 分钟,则我们应该在用户下单后的第 30 分钟检查订单支付状态,如果发现未支付,应该立刻取消订单,释放库存。


但问题来了:如何才能准确的实现在下单后第 30 分钟去检查支付状态呢?


像这种在一段时间以后才执行的任务,我们称之为延迟任务,而要实现延迟任务,最简单的方案就是利用 MQ 的延迟消息了。


在 RabbitMQ 中实现延迟消息也有两种方案:

  • 死信交换机+TTL

  • 延迟消息插件

这一章我们就一起研究下这两种方案的实现方式,以及优缺点。


死信交换机和延迟消息


首先我们来学习一下基于死信交换机的延迟消息方案。


死信交换机


什么是死信?


当一个队列中的消息满足下列情况之一时,可以成为死信(dead letter):

  • 消费者使用basic.reject或 basic.nack声明消费失败,并且消息的requeue参数设置为 false

  • 消息是一个过期消息,超时无人消费

  • 要投递的队列消息满了,无法投递


如果一个队列中的消息已经成为死信,并且这个队列通过**dead-letter-exchange**属性指定了一个交换机,那么队列中的死信就会投递到这个交换机中,而这个交换机就称为死信交换机(Dead Letter Exchange)。而此时加入有队列与死信交换机绑定,则最终死信就会被投递到这个队列中。


死信交换机有什么作用呢?

  1. 收集那些因处理失败而被拒绝的消息

  2. 收集那些因队列满了而被拒绝的消息

  3. 收集因 TTL(有效期)到期的消息


延迟消息


前面两种作用场景可以看做是把死信交换机当做一种消息处理的最终兜底方案,与消费者重试时讲的RepublishMessageRecoverer作用类似。


而最后一种场景,大家设想一下这样的场景:如图,有一组绑定的交换机(ttl.fanout)和队列(ttl.queue)。但是ttl.queue没有消费者监听,而是设定了死信交换机hmall.direct,而队列direct.queue1则与死信交换机绑定,RoutingKey 是 blue:



假如我们现在发送一条消息到ttl.fanout,RoutingKey 为 blue,并设置消息的有效期为 5000 毫秒:



:::warning 注意:尽管这里的ttl.fanout不需要 RoutingKey,但是当消息变为死信并投递到死信交换机时,会沿用之前的 RoutingKey,这样hmall.direct才能正确路由消息。:::


消息肯定会被投递到ttl.queue之后,由于没有消费者,因此消息无人消费。5 秒之后,消息的有效期到期,成为死信:



死信被再次投递到死信交换机hmall.direct,并沿用之前的 RoutingKey,也就是blue



由于direct.queue1hmall.direct绑定的 key 是 blue,因此最终消息被成功路由到direct.queue1,如果此时有消费者与direct.queue1绑定, 也就能成功消费消息了。但此时已经是 5 秒钟以后了:



也就是说,publisher 发送了一条消息,但最终 consumer 在 5 秒后才收到消息。我们成功实现了延迟消息


总结


注意:RabbitMQ 的消息过期是基于追溯方式来实现的,也就是说当一个消息的 TTL 到期以后不一定会被移除或投递到死信交换机,而是在消息恰好处于队首时才会被处理。当队列中消息堆积很多的时候,过期消息可能不会被按时处理,因此你设置的 TTL 时间不一定准确。


DelayExchange 插件


基于死信队列虽然可以实现延迟消息,但是太麻烦了。因此 RabbitMQ 社区提供了一个延迟消息插件来实现相同的效果。官方文档说明:Scheduling Messages with RabbitMQ | RabbitMQ - Blog


下载


插件下载地址:GitHub - rabbitmq/rabbitmq-delayed-message-exchange: Delayed Messaging for RabbitMQ 由于我们安装的 MQ 是3.8版本,因此这里下载3.8.17版本:



安装


因为我们是基于 Docker 安装,所以需要先查看 RabbitMQ 的插件目录对应的数据卷。


docker volume inspect mq-plugins
复制代码


结果如下:


[    {        "CreatedAt": "2024-06-19T09:22:59+08:00",        "Driver": "local",        "Labels": null,        "Mountpoint": "/var/lib/docker/volumes/mq-plugins/_data",        "Name": "mq-plugins",        "Options": null,        "Scope": "local"    }]
复制代码


插件目录被挂载到了/var/lib/docker/volumes/mq-plugins/_data这个目录,我们上传插件到该目录下。


接下来执行命令,安装插件:


docker exec -it mq rabbitmq-plugins enable rabbitmq_delayed_message_exchange
复制代码


运行结果如下:



声明延迟交换机


基于注解方式:


@RabbitListener(bindings = @QueueBinding(        value = @Queue(name = "delay.queue", durable = "true"),        exchange = @Exchange(name = "delay.direct", delayed = "true"),        key = "delay"))public void listenDelayMessage(String msg){    log.info("接收到delay.queue的延迟消息:{}", msg);}
复制代码


基于@Bean的方式:


package com.itheima.consumer.config;
import lombok.extern.slf4j.Slf4j;import org.springframework.amqp.core.*;import org.springframework.context.annotation.Bean;import org.springframework.context.annotation.Configuration;
@Slf4j@Configurationpublic class DelayExchangeConfig {
@Bean public DirectExchange delayExchange(){ return ExchangeBuilder .directExchange("delay.direct") // 指定交换机类型和名称 .delayed() // 设置delay的属性为true .durable(true) // 持久化 .build(); }
@Bean public Queue delayedQueue(){ return new Queue("delay.queue"); } @Bean public Binding delayQueueBinding(){ return BindingBuilder.bind(delayedQueue()).to(delayExchange()).with("delay"); }}
复制代码


发送延迟消息


发送消息时,必须通过 x-delay 属性设定延迟时间:


@Testvoid testPublisherDelayMessage() {    // 1.创建消息    String message = "hello, delayed message";    // 2.发送消息,利用消息后置处理器添加消息头    rabbitTemplate.convertAndSend("delay.direct", "delay", message, new MessagePostProcessor() {        @Override        public Message postProcessMessage(Message message) throws AmqpException {            // 添加延迟消息属性            message.getMessageProperties().setDelay(5000);            return message;        }    });}
复制代码


注意:延迟消息插件内部会维护一个本地数据库表,同时使用 Elang Timers 功能实现计时。如果消息的延迟时间设置较长,可能会导致堆积的延迟消息非常多,会带来较大的 CPU 开销,同时延迟消息的时间会存在误差。因此,不建议设置延迟时间过长的延迟消息


订单状态同步问题


接下来,我们就在交易服务中利用延迟消息实现订单支付状态的同步。其大概思路如下:



假如订单超时支付时间为 30 分钟,理论上说我们应该在下单时发送一条延迟消息,延迟时间为 30 分钟。这样就可以在接收到消息时检验订单支付状态,关闭未支付订单。但是大多数情况下用户支付都会在 1 分钟内完成,我们发送的消息却要在 MQ 中停留 30 分钟,额外消耗了 MQ 的资源。因此,我们最好多检测几次订单支付状态,而不是在最后第 30 分钟才检测。例如:我们在用户下单后的第 10 秒、20 秒、30 秒、45 秒、60 秒、1 分 30 秒、2 分、...30 分分别设置延迟消息,如果提前发现订单已经支付,则后续的检测取消即可。这样就可以有效避免对 MQ 资源的浪费了。


优化后的实现思路如下:



由于我们要多次发送延迟消息,因此需要先定义一个记录消息延迟时间的消息体,处于通用性考虑,我们将其定义到hm-common模块下:



代码如下:


package com.hmall.common.domain;
import com.hmall.common.utils.CollUtils;import lombok.Data;
import java.util.List;
@Datapublic class MultiDelayMessage<T> { /** * 消息体 */ private T data; /** * 记录延迟时间的集合 */ private List<Long> delayMillis;
public MultiDelayMessage(T data, List<Long> delayMillis) { this.data = data; this.delayMillis = delayMillis; } public static <T> MultiDelayMessage<T> of(T data, Long ... delayMillis){ return new MultiDelayMessage<>(data, CollUtils.newArrayList(delayMillis)); }
/** * 获取并移除下一个延迟时间 * @return 队列中的第一个延迟时间 */ public Long removeNextDelay(){ return delayMillis.remove(0); }
/** * 是否还有下一个延迟时间 */ public boolean hasNextDelay(){ return !delayMillis.isEmpty(); }}
复制代码


定义常量


无论是消息发送还是接收都是在交易服务完成,因此我们在trade-service中定义一个常量类,用于记录交换机、队列、RoutingKey 等常量:



内容如下:


package com.hmall.trade.constants;
public interface MqConstants { String DELAY_EXCHANGE = "trade.delay.topic"; String DELAY_ORDER_QUEUE = "trade.order.delay.queue"; String DELAY_ORDER_ROUTING_KEY = "order.query";}
复制代码


抽取共享 mq 配置


我们将 mq 的配置抽取到 nacos 中,方便各个微服务共享配置。在 nacos 中定义一个名为shared-mq.xml的配置文件,内容如下:


spring:  rabbitmq:    host: ${hm.mq.host:192.168.150.101} # 主机名    port: ${hm.mq.port:5672} # 端口    virtual-host: ${hm.mq.vhost:/hmall} # 虚拟主机    username: ${hm.mq.un:hmall} # 用户名    password: ${hm.mq.pw:123} # 密码    listener:      simple:        prefetch: 1 # 每次只能获取一条消息,处理完成才能获取下一个消息
复制代码


这里只添加一些基础配置,至于生产者确认,消费者确认配置则由微服务根据业务自己决定。

trade-service模块添加共享配置:



改造下单业务


接下来,我们改造下单业务,在下单完成后,发送延迟消息,查询支付状态。


1)引入依赖在trade-service模块的pom.xml中引入 amqp 的依赖:


  <!--amqp-->  <dependency>      <groupId>org.springframework.boot</groupId>      <artifactId>spring-boot-starter-amqp</artifactId>  </dependency>
复制代码


2)改造下单业务修改trade-service模块的com.hmall.trade.service.impl.OrderServiceImpl类的createOrder方法,添加消息发送的代码:



编写查询支付状态接口


由于 MQ 消息处理时需要查询支付状态,因此我们要在 pay-service 模块定义一个这样的接口,并提供对应的 FeignClient.首先,在 hm-api 模块定义三个类:



说明:

  • PayOrderDTO:支付单的数据传输实体

  • PayClient:支付系统的 Feign 客户端

  • PayClientFallback:支付系统的 fallback 逻辑


PayOrderDTO 代码如下:


package com.hmall.api.dto;
import io.swagger.annotations.ApiModel;import io.swagger.annotations.ApiModelProperty;import lombok.Data;
import java.time.LocalDateTime;
/** * <p> * 支付订单 * </p> */@Data@ApiModel(description = "支付单数据传输实体")public class PayOrderDTO { @ApiModelProperty("id") private Long id; @ApiModelProperty("业务订单号") private Long bizOrderNo; @ApiModelProperty("支付单号") private Long payOrderNo; @ApiModelProperty("支付用户id") private Long bizUserId; @ApiModelProperty("支付渠道编码") private String payChannelCode; @ApiModelProperty("支付金额,单位分") private Integer amount; @ApiModelProperty("付类型,1:h5,2:小程序,3:公众号,4:扫码,5:余额支付") private Integer payType; @ApiModelProperty("付状态,0:待提交,1:待支付,2:支付超时或取消,3:支付成功") private Integer status; @ApiModelProperty("拓展字段,用于传递不同渠道单独处理的字段") private String expandJson; @ApiModelProperty("第三方返回业务码") private String resultCode; @ApiModelProperty("第三方返回提示信息") private String resultMsg; @ApiModelProperty("支付成功时间") private LocalDateTime paySuccessTime; @ApiModelProperty("支付超时时间") private LocalDateTime payOverTime; @ApiModelProperty("支付二维码链接") private String qrCodeUrl; @ApiModelProperty("创建时间") private LocalDateTime createTime; @ApiModelProperty("更新时间") private LocalDateTime updateTime;}
复制代码


PayClient 代码如下:


package com.hmall.api.client;
import com.hmall.api.client.fallback.PayClientFallback;import com.hmall.api.dto.PayOrderDTO;import org.springframework.cloud.openfeign.FeignClient;import org.springframework.web.bind.annotation.GetMapping;import org.springframework.web.bind.annotation.PathVariable;
@FeignClient(value = "pay-service", fallbackFactory = PayClientFallback.class)public interface PayClient { /** * 根据交易订单id查询支付单 * @param id 业务订单id * @return 支付单信息 */ @GetMapping("/pay-orders/biz/{id}") PayOrderDTO queryPayOrderByBizOrderNo(@PathVariable("id") Long id);}
复制代码


PayClientFallback 代码如下:


package com.hmall.api.client.fallback;
import com.hmall.api.client.PayClient;import com.hmall.api.dto.PayOrderDTO;import lombok.extern.slf4j.Slf4j;import org.springframework.cloud.openfeign.FallbackFactory;
@Slf4jpublic class PayClientFallback implements FallbackFactory<PayClient> { @Override public PayClient create(Throwable cause) { return new PayClient() { @Override public PayOrderDTO queryPayOrderByBizOrderNo(Long id) { return null; } }; }}
复制代码


最后,在 pay-service 模块的 PayController 中实现该接口:


@ApiOperation("根据id查询支付单")@GetMapping("/biz/{id}")public PayOrderDTO queryPayOrderByBizOrderNo(@PathVariable("id") Long id){    PayOrder payOrder = payOrderService.lambdaQuery().eq(PayOrder::getBizOrderNo, id).one();    return BeanUtils.copyBean(payOrder, PayOrderDTO.class);}
复制代码


消息监听


接下来,我们在 trader-service 编写一个监听器,监听延迟消息,查询订单支付状态:



代码如下:


package com.hmall.trade.listener;
import com.hmall.api.client.PayClient;import com.hmall.api.dto.PayOrderDTO;import com.hmall.common.domain.MultiDelayMessage;import com.hmall.trade.constants.MqConstants;import com.hmall.trade.domain.po.Order;import com.hmall.trade.service.IOrderService;import lombok.RequiredArgsConstructor;import lombok.extern.slf4j.Slf4j;import org.springframework.amqp.core.ExchangeTypes;import org.springframework.amqp.rabbit.annotation.Exchange;import org.springframework.amqp.rabbit.annotation.Queue;import org.springframework.amqp.rabbit.annotation.QueueBinding;import org.springframework.amqp.rabbit.annotation.RabbitListener;import org.springframework.amqp.rabbit.core.RabbitTemplate;import org.springframework.stereotype.Component;
@Slf4j@Component@RequiredArgsConstructorpublic class OrderStatusListener {
private final IOrderService orderService;
private final PayClient payClient;
private final RabbitTemplate rabbitTemplate;
@RabbitListener(bindings = @QueueBinding( value = @Queue(name = MqConstants.DELAY_ORDER_QUEUE, durable = "true"), exchange = @Exchange(name = MqConstants.DELAY_EXCHANGE, type = ExchangeTypes.TOPIC), key = MqConstants.DELAY_ORDER_ROUTING_KEY )) public void listenOrderCheckDelayMessage(MultiDelayMessage<Long> msg) { // 1.获取消息中的订单id Long orderId = msg.getData(); // 2.查询订单,判断状态:1是未支付,大于1则是已支付或已关闭 Order order = orderService.getById(orderId); if (order == null || order.getStatus() > 1) { // 订单不存在或交易已经结束,放弃处理 return; } // 3.可能是未支付,查询支付服务 PayOrderDTO payOrder = payClient.queryPayOrderByBizOrderNo(orderId); if (payOrder != null && payOrder.getStatus() == 3) { // 支付成功,更新订单状态 orderService.markOrderPaySuccess(orderId); return; } // 4.确定未支付,判断是否还有剩余延迟时间 if (msg.hasNextDelay()) { // 4.1.有延迟时间,需要重发延迟消息,先获取延迟时间的int值 int delayVal = msg.removeNextDelay().intValue(); // 4.2.发送延迟消息 rabbitTemplate.convertAndSend(MqConstants.DELAY_EXCHANGE, MqConstants.DELAY_ORDER_ROUTING_KEY, msg, message -> { message.getMessageProperties().setDelay(delayVal); return message; }); return; } // 5.没有剩余延迟时间了,说明订单超时未支付,需要取消订单 orderService.cancelOrder(orderId); }}
复制代码


注意,这里要在 OrderServiceImpl 中实现 cancelOrder 方法


文章转载自:程序员Seven

原文链接:https://www.cnblogs.com/seven97-top/p/18863305

体验地址:http://www.jnpfsoft.com/?from=001YH

用户头像

还未添加个人签名 2023-06-19 加入

还未添加个人简介

评论

发布
暂无评论
RabbitMQ高级使用_Java_不在线第一只蜗牛_InfoQ写作社区