RabbitMQ高级特性 - 生产者消息确认机制

文章目录

生产者消息确认机制
- 概述
- [confirm 代码实现](#confirm 代码实现)
- [return 代码实现](#return 代码实现)

生产者消息确认机制

概述

为了保证信息从生产者发送到队列，因此引入了生产者的消息确认机制.

RabbitMQ 提供了两种解决方案：

通过事务机制实现.
通过发送确认机制（confirm 和 return）实现.

因为事务机制比较消耗性能，在实际工作中用的也不多，因此这里主要介绍 confirm 和 return 机制来实现发送放的确认.

a）confirm 确认模式

如上图，confirm 确认模式主要保障于生产者到交换机的消息可靠性.

具体的，在生产者发送消息之前，给 RabbitTemplate 设置一个 ConfirmCallback 回调监听：

如果 Exchange 成功收到消息，那么 ConfirmCallback 这个回调 ack 参数就为 true
如果 Exchange 没有收到消息，那么 ConfirmCallback 这个回调 ack 参数就为 false

b）return 退回模式

如上图，confirm 确认模式主要保障于交换机到队列的消息可靠性.

具体的，在生产者发送消息之前，给 RabbitTemplate 设置一个 ReturnsCallback 回调监听：

如果 Queue 成功收到 Exchange 的消息，那么 ReturnsCallback 回调监听 就不会收到任何消息.
如果 Queue 没有收到 Exchange 的消息，那么 ReturnsCallback 回调监听 就会收到该消息.

confirm 代码实现

a）配置文件

yml 复制代码

spring:
  application:
    name: rabbitmq
  rabbitmq:
    host: env-base
    port: 5672
    username: root
    password: 1111
    publisher-confirm-type: correlated # 开启发送方确认机制

b）交换机、队列、绑定配置

kotlin 复制代码

    @Bean("confirmExchange")
    fun confirmExchange() = DirectExchange(MQConst.CONFIRM_EXCHANGE)
    @Bean("confirmQueue")
    fun confirmQueue() = Queue(MQConst.CONFIRM_QUEUE)
    @Bean("confirmBinding")
    fun confirmBinding(
        @Qualifier("confirmExchange") exchange: DirectExchange,
        @Qualifier("confirmQueue") queue: Queue,
    ): Binding = BindingBuilder
        .bind(queue)
        .to(exchange)
        .with(MQConst.CONFIRM_BINDING)

c）confirmRabbitTemplate Bean 配置

kotlin 复制代码

@Configuration
class MQTemplateConfig {

    /**
     * 这个配置一定要有！！！(或者有大于等于 2 个的 RabbitTemplate Bean)
     *
     * 这是由于 Autowired 注解自身的原因(以 rabbitmq 为例):
     * 如果配置文件中配置 rabbitmq 相关连接信息，那么 spring 会自动为其创建 RabbitTemplate Bean 对象
     * 如果配置文件中配置 rabbitmq 相关连接信息，而且代码中也配置了一个 RabbitTemplate 的 Bean(名称为 confirmRabbitTemplate)，那么 Spring 将不会自动配置默认的 RabbitTemplate Bean 对象
     * 这就导致，我们无论代码写的注入的是 rabbitTemplate 还是 confirmRabbitTemplate，但实际上注入的都是 confirmRabbitTemplate
     */
    @Bean("rabbitTemplate")
    fun rabbitTemplate(
        connectionFactory: ConnectionFactory
    ): RabbitTemplate {
        return RabbitTemplate(connectionFactory)
    }

    @Bean("confirmRabbitTemplate")
    fun confirmRabbitTemplate(
        connectionFactory: ConnectionFactory
    ): RabbitTemplate {
        val tpl = RabbitTemplate(connectionFactory)
        tpl.setConfirmCallback(RabbitTemplate.ConfirmCallback { correlationData, ack, cause ->
            println("执行了 confirm ...")
            if (ack) {
                println("confirm ack: { 消息id: ${correlationData?.id} }")
            } else {
                println("confirm nack: { 消息id: ${correlationData?.id}, cause: $cause }")
                //进行相应的业务处理...
            }
        })
        return tpl
    }

}

d）生产者接口

kotlin 复制代码

@RestController
@RequestMapping("/mq")
class MQApi(
    val confirmRabbitTemplate: RabbitTemplate
) {

    @RequestMapping("/confirm")
    fun confirm(): String {
        val data = CorrelationData("1")
        confirmRabbitTemplate.convertAndSend(MQConst.CONFIRM_EXCHANGE, MQConst.CONFIRM_BINDING, "confirm msg 1", data)
        return "ok"
    }

}

此处演示无需消费者...

e）消息正确的路由到交换机，效果如下：

f）消息没有找到交换机（发送消息时，写了一个不存在的交换机的名字），效果如下：

return 代码实现

a）配置文件

yml 复制代码

spring:
  application:
    name: rabbitmq
  rabbitmq:
    host: env-base
    port: 5672
    username: root
    password: 1111
    publisher-confirm-type: correlated # 开启发送方确认机制

b）bean 的配置

kotlin 复制代码

@Configuration
class MQTemplateConfig {

    /**
     * 这个配置一定要有！！！(或者有大于等于 2 个的 RabbitTemplate Bean)
     *
     * 这是由于 Autowired 注解自身的原因(以 rabbitmq 为例):
     * 如果配置文件中配置 rabbitmq 相关连接信息，那么 spring 会自动为其创建 RabbitTemplate Bean 对象
     * 如果配置文件中配置 rabbitmq 相关连接信息，而且代码中也配置了一个 RabbitTemplate 的 Bean(名称为 confirmRabbitTemplate)，那么 Spring 将不会自动配置默认的 RabbitTemplate Bean 对象
     * 这就导致，我们无论代码写的注入的是 rabbitTemplate 还是 confirmRabbitTemplate，但实际上注入的都是 confirmRabbitTemplate
     */
    @Bean("rabbitTemplate")
    fun rabbitTemplate(
        connectionFactory: ConnectionFactory
    ): RabbitTemplate {
        return RabbitTemplate(connectionFactory)
    }

    @Bean("confirmRabbitTemplate")
    fun confirmRabbitTemplate(
        connectionFactory: ConnectionFactory
    ): RabbitTemplate {
        val tpl = RabbitTemplate(connectionFactory)
        tpl.setConfirmCallback(RabbitTemplate.ConfirmCallback { correlationData, ack, cause ->
            println("执行了 confirm ...")
            if (ack) {
                println("confirm ack: { 消息id: ${correlationData?.id} }")
            } else {
                println("confirm nack: { 消息id: ${correlationData?.id}, cause: $cause }")
                //进行相应的业务处理...
            }
        })
        //这里可以和 confirm模式 一起配置
        //mandatory = true 属性是在告诉 rabbitmq，如果一个消息无法被任何队列消费，那么该消息就会返回给发送者，此时 ReturnCallback 就会被触发
        //mandatory 相当于是开启 ReturnsCallback 前提
        tpl.setMandatory(true)
        tpl.setReturnsCallback(RabbitTemplate.ReturnsCallback { returned ->
            println("执行了 return ...")
            println("return: $returned")
        })
        return tpl
    }

}

c）生产者接口

kotlin 复制代码

@RestController
@RequestMapping("/mq")
class MQApi(
    val confirmRabbitTemplate: RabbitTemplate
) {

    @RequestMapping("/confirm")
    fun confirm(): String {
        val data = CorrelationData("1")
        confirmRabbitTemplate.convertAndSend(MQConst.CONFIRM_EXCHANGE, MQConst.CONFIRM_BINDING, "confirm msg 1", data)
        return "ok"
    }

}

d）正确的路由到队列，效果如下：

可以看到只有 confirm模式被触发.

e）没有路由到队列（发送消息时，我改成了一个不存在的 routingKey 名字），效果如下：