发布确认高级

在生产环境中由于一些不明原因，导致 rabbitmq 重启，在 RabbitMQ 重启期间生产者消息投递失败，导致消息丢失，需要手动处理和恢复。于是，我们开始思考，如何才能进行 RabbitMQ 的消息可靠投递呢？特别是在这样比较极端的情况，RabbitMQ 集群不可用的时候，无法投递的消息该如何处理呢？

发布确认 springboot 版本

确认机制方案

在这里插入图片描述

代码架构图

在这里插入图片描述

配置文件

在配置文件当中需要添加

spring.rabbitmq.publisher-confirm-type=correlated

NONE

禁用发布确认模式，是默认值
CORRELATED

发布消息成功到交换器后会出发回调方法
SIMPLE

经测试有两种效果，其一效果和 CORRELATED 值一样会触发回调方法，

其二在发布消息成功后使用 rabbitTemplate 调用 waitForConfirms 或 waitForConfirmsOrDie 方法等待 broker 节点返回发送结果，根据返回结果来判定下一步的逻辑，要注意的点是 waitForConfirmsOrDie 方法如果返回 false 则会关闭 channel，则接下来无法发送消息到 broker

spring.rabbitmq.template.mandatory=true

在仅开启了生产者确认机制的情况下，交换机接收到消息后，会直接给消息生产者发送确认消息，如果发现该消息不可路由，那么消息会被直接丢弃，此时生产者是不知道消息被丢弃这个事件的。通过设置 mandatory 参数可以在当消息传递过程中不可达目的地时将消息返回给生产者。(可以在配置文件中设置,也可以在接口实现类中设置,这里是在实现接口类中设置的)

配置类

package com.qin.rabbitmq.rabbitmq2.config;

import lombok.extern.slf4j.Slf4j;
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.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
@Slf4j
public class CallBackMessageConfig {
    public final static String CALLBACK_EXCHANGE = "callback.exchange";
    public final static String CALLBACK_QUEUE = "callback.queue";
    public final static String CALLBACK_ROUTING_KEY = "callback.routingKey";

    @Bean
    public DirectExchange callbackExchange(){
        return new DirectExchange(CALLBACK_EXCHANGE);
    }
    @Bean
    public Queue callbackQueue(){
        return new Queue(CALLBACK_QUEUE);
    }
    @Bean
    public Binding binding(@Qualifier("callbackExchange")DirectExchange exchange,@Qualifier("callbackQueue")Queue queue){
        return BindingBuilder.bind(queue).to(exchange).with(CALLBACK_ROUTING_KEY);
    }
}

消息生产者

package com.qin.rabbitmq.rabbitmq2.controller;

import com.qin.rabbitmq.rabbitmq2.config.CallBackMessageConfig;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RequestMapping("/confirm")
@RestController
@Slf4j
public class CallBackProducer {
    @Autowired
    private RabbitTemplate rabbitTemplate;
    @GetMapping("/sendMsg/{msg}")
    public void sendMessage(@PathVariable("msg")String msg){
        //指定消息 id 为 1
        CorrelationData correlationData1=new CorrelationData("1");
        rabbitTemplate.convertAndSend(CallBackMessageConfig.CALLBACK_EXCHANGE,
                CallBackMessageConfig.CALLBACK_ROUTING_KEY,
                msg,
                correlationData1);

        CorrelationData correlationData2=new CorrelationData("2");
        rabbitTemplate.convertAndSend(CallBackMessageConfig.CALLBACK_EXCHANGE,
                CallBackMessageConfig.CALLBACK_ROUTING_KEY+"1",
                msg,
                correlationData2);
        log.info("发送消息内容:{}",msg);
    }
}

回调接口

package com.qin.rabbitmq.rabbitmq2.callback;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.core.ReturnedMessage;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

@Component
@Slf4j
public class MyConfirmCallBack implements RabbitTemplate.ConfirmCallback,RabbitTemplate.ReturnsCallback {

    //注入容器中的 RabbitTemplate
    @Autowired
    private RabbitTemplate rabbitTemplate;

    //将自定义的方法设置进去
    @PostConstruct
    public void init(){
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnsCallback(this);
        rabbitTemplate.setMandatory(true);
    }

    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        String id=correlationData!=null?correlationData.getId():"";
        if(ack){
            log.info("交换机已经收到 id 为:{}的消息",id);
        }else{
            log.info("交换机还未收到 id 为:{}消息,由于原因:{}",id,cause);
        }
    }

    @Override
    public void returnedMessage(ReturnedMessage returned) {
        log.error(" 消 息 {}, 被交换机 {} 退回，退回原因 :{}, 路 由 key:{}",new
                String(returned.getMessage().getBody()),returned.getExchange(),returned.getReplyText(),returned.getRoutingKey());
    }
}

消息消费者

package com.qin.rabbitmq.rabbitmq2.bean;

import com.qin.rabbitmq.rabbitmq2.config.CallBackMessageConfig;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.stereotype.Component;

@Component
@Slf4j
public class CallbackConsumer {
    @RabbitListener(queues = CallBackMessageConfig.CALLBACK_QUEUE)
    public void worker(Message message){
        String msg = new String(message.getBody());
        log.info("接受到队列 confirm.queue 消息:{}",msg);
    }
}

备份交换机

有了 mandatory 参数和回退消息，我们获得了对无法投递消息的感知能力，有机会在生产者的消息无法被投递时发现并处理。但有时候，我们并不知道该如何处理这些无法路由的消息，最多打个日志，然后触发报警，再来手动处理。而通过日志来处理这些无法路由的消息是很不优雅的做法，特别是当生产者所在的服务有多台机器的时候，手动复制日志会更加麻烦而且容易出错。而且设置 mandatory 参数会增加生产者的复杂性，需要添加处理这些被退回的消息的逻辑。如果既不想丢失消息，又不想增加生产者的复杂性，该怎么做呢？前面在设置死信队列的文章中，我们提到，可以为队列设置死信交换机来存储那些处理失败的消息，可是这些不可路由消息根本没有机会进入到队列，因此无法使用死信队列来保存消息。在 RabbitMQ 中，有一种备份交换机的机制存在，可以很好的应对这个问题。什么是备份交换机呢？备份交换机可以理解为 RabbitMQ 中交换机的“备胎”，当我们为某一个交换机声明一个对应的备份交换机时，就是为它创建一个备胎，当交换机接收到一条不可路由消息时，将会把这条消息转发到备份交换机中，由备份交换机来进行转发和处理，通常备份交换机的类型为 Fanout ，这样就能把所有消息都投递到与其绑定的队列中，然后我们在备份交换机下绑定一个队列，这样所有那些原交换机无法被路由的消息，就会都进入这个队列了。当然，我们还可以建立一个报警队列，用独立的消费者来进行监测和报警。

代码架构图

在这里插入图片描述

修改配置类

主要修改:

添加备份交换机
添加警报队列
警报队列绑定备份交换机
确认交换机添加备份交换机

package com.qin.rabbitmq.rabbitmq2.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.*;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
@Slf4j
public class CallBackMessageConfig {
    //消息回退的交换机,队列,routingKey
    public final static String CALLBACK_EXCHANGE = "callback.exchange";
    public final static String CALLBACK_QUEUE = "callback.queue";
    public final static String CALLBACK_ROUTING_KEY = "callback.routingKey";
    //备份交换机
    public final static String BACKUP_EXCHANGE = "backup.exchange";
    public final static String BACKUP_QUEUE = "backup.queue";

    //声明备份交换机
    @Bean
    public FanoutExchange backupExchange(){
        return new FanoutExchange(BACKUP_EXCHANGE);
    }
    //声明警报队列
    @Bean
    public Queue backupQueue(){
        return new Queue(BACKUP_QUEUE);
    }
    //绑定警报队列和备份交换机
    @Bean
    public Binding backupBinding(@Qualifier("backupExchange")FanoutExchange exchange,@Qualifier("backupQueue")Queue queue){
        return BindingBuilder.bind(queue).to(exchange);
    }

    //声明消息回退交换机
    @Bean
    public DirectExchange callbackExchange(){
        return ExchangeBuilder.directExchange(CALLBACK_EXCHANGE).durable(true).
                //设置该交换机的备份交换机
                withArgument("alternate-exchange", BACKUP_EXCHANGE).build();
    }
    @Bean
    //声明消息回退队列
    public Queue callbackQueue(){
        return new Queue(CALLBACK_QUEUE);
    }
    //绑定消息回退的队列和交换机
    @Bean
    public Binding binding(@Qualifier("callbackExchange")DirectExchange exchange,@Qualifier("callbackQueue")Queue queue){
        return BindingBuilder.bind(queue).to(exchange).with(CALLBACK_ROUTING_KEY);
    }


}

警报消费者

package com.qin.rabbitmq.rabbitmq2.bean;

import com.qin.rabbitmq.rabbitmq2.config.CallBackMessageConfig;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.stereotype.Component;

@Component
@Slf4j
public class BackupConsumer {
    @RabbitListener(queues = CallBackMessageConfig.BACKUP_QUEUE)
    public void backupWorker(Message message){
         log.error("报警发现不可路由消息：{}", new String(message.getBody()));
    }
}