RabbitMQ生产者确认机制

1.问题分析

publisher生产者将消息发送到exchange交换机，再由exchange路由到指定的queue队列。

其中每一步都可能会导致数据丢失：

• 生产者发送的消息未送达exchange
• 消息到达exchange后未到达queue

2.实现方案

 RabbitMQ提供了publisher confirm机制来避免消息发送到MQ过程中丢失。消息发送到MQ以后，会返回一个结果给发送者，表示消息是否处理成功。

返回结果有两种方式

1.publisher-confirm：发送者确认，消息成功发送到交换机返回ack，失败返回nack

2.publisher-return：消息发送到交换机，但没有路由到队列，返回ack，及路由失败原因

注意：确认机制发送消息的时候，需要给每个消息指定一个全局唯一id，以区分不同的消息，避免冲突

3.实现

3.1修改配置

修改publisher服务配置文件

spring:
  rabbitmq:
    publisher-confirm-type: correlated
    publisher-returns: true
    template:
      mandatory: true

publish-confirm-type：开启publisher-confirm，这里支持两种类型 

• simple：同步等待confirm结果，直到超时
• correlated：异步回调，定义ConfirmCallback，MQ返回结果时会回调这个ConfirmCallback

publish-returns：开启publish-return功能，同样是基于callback机制，不过是定义ReturnCallback

template.mandatory：定义消息路由失败时的策略。true，则调用ReturnCallback；false：则直接丢弃消息

3.2定义Return回调

@Configuration
@Slf4j
public class CommonConfig implements ApplicationContextAware {


    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        //获取RabbitTemplate
        RabbitTemplate bean = applicationContext.getBean(RabbitTemplate.class);
        //设置ReturnCallback
        bean.setReturnCallback(new RabbitTemplate.ReturnCallback() {
            @Override
            public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
                // 投递失败，记录日志
                log.info("消息发送失败，应答码{}，原因{}，交换机{}，路由键{},消息{}",
                        replyCode, replyText, exchange, routingKey, message.toString());
                // 如果有业务需要，可以重发消息
            }
        });

    }
}

3.3定义ConfirmCallback

 @Test
    public void testSendMessage2SimpleQueue() throws InterruptedException {

        //添加collback
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String s) {
                if (ack) {
                    //ack，消息发送成功
                    log.info("消息成功发送" + correlationData.getId());
                } else {
                    //nack，消息发送失败
                    log.error("消息发送失败" + correlationData.getId());
                }
            }
        });

        String routingKey = "simple";
        String message = "hello, spring amqp!";
        //全局唯一的消息id封装到CorrelationData里面
        CorrelationData correlationData = new CorrelationData(UUID.randomUUID().toString());
        rabbitTemplate.convertAndSend("camq.topic", routingKey, message,correlationData);
    }