RabbitMQ消息确认机制

科技 08-14 来源：程序那点事

前言

上一章讲解了RabbitMq的三种Exchange消息发送的模式，但是在默认情况下RabbitMQ并不能保证消息是否发送成功，以及是否能被成功消费，为了保证消息在传递过程中不丢失，需要对消息进行确认机制，来提高消息的可靠性。

消息确认

基本流程

说明：

生产者发送消息到RabbitMQ Server后，RabbitMQ Server需要对生产者进行消息Confirm确认。
消费者消费消息后需要对 RabbitMQ Server进行消息ACK确认。

消息确认模式

RabbitMq提供了两种消息发送者确认模式分别为： ConfirmCallback确认模式和 ReturnCallback退回模式。

ConfirmCallback确认模式

@Componentpublic class RabbitConfirmConfig implements ConfirmCallback{    private Logger logger = LoggerFactory.getLogger(RabbitConfirmConfig.class);        public void confirm(CorrelationData correlationData, boolean ack,            String cause)    {        logger.info("数据内容:{}",correlationData);        logger.info("是否确认成功:{}",ack);        logger.info("错误原因:{}",cause);        if (!ack)         {            logger.info("exchange produce confirm message send error" + cause);        }        else         {            logger.info("exchange produce confirm message send success");        }    }}复制代码

说明：ConfirmCallback模式确认，需要重写confirm接方法，此方法的三个参数分别为：CorrelationData、ack、cause

CorrelationData:对象内部只有一个id属性，用来表示当前消息的唯一性。
ack:消息投递状态，true表示投递成功
cause: 消息投递失败原因

虽然消息被broker接收到只能表示已经到达MQ服务器，但是并不能保证消息一定会被投递到目标 queue里。所以我们需要实现returnCallback来进行相关处理。

ReturnCallback退回模式

@Componentpublic class RabbitReturnConfig implements ReturnCallback{    private Logger logger = LoggerFactory.getLogger(RabbitReturnConfig.class);        public void returnedMessage(Message message, int replyCode,            String replyText, String exchange, String routingKey)    {       logger.info("消息发送送到队列信息：");       logger.info("发生消息：{}",message);       logger.info("回应码：{}",replyCode);       logger.info("回应信息：{}",replyText);       logger.info("交换机：{}",exchange);       logger.info("路由键：{}",routingKey);    }}复制代码

说明：实现接口ReturnCallback重写returnedMessage()方法，方法有五个参数message（消息体）、replyCode（响应code）、replyText（响应内容）、exchange（交换机）、routingKey（路由键）。

消息发送者确认

@Componentpublic class MqConfirmProduce{    @Autowired    private RabbitTemplate rabbitTemplate;        @Autowired    private RabbitConfirmConfig rabbitConfirmConfig;        @Autowired    private RabbitReturnConfig rabbitReturnConfig;        /**     *      * @param exchange 消息交互机名称     * @param routeKey 消息路由键的名称     * @param message  消息内容     */    public void sendMessage(String exchange ,String routeKey,Object msg)    {        //确保消息发送失败后可以重新返回到队列中        rabbitTemplate.setMandatory(true);                // 消费者确认收到消息后，手动ack回执回调处理        rabbitTemplate.setConfirmCallback(rabbitConfirmConfig);        //消息投递到队列失败回调处理        rabbitTemplate.setReturnCallback(rabbitReturnConfig);        //保证消息唯一性        CorrelationData correlationData =new CorrelationData(UUID.randomUUID().toString());        //发送消息        rabbitTemplate.convertAndSend(exchange,routeKey,msg,                message -> {                    message.getMessageProperties().setDeliveryMode(MessageDeliveryMode.PERSISTENT);                    return message;                },                correlationData);    }}复制代码

说明：注意需要开启消息确认的配置：

  rabbitmq:    host: localhost    port: 5672    username: guest    password: guest    virtual-host: /    #开启发送确认    publisher-confirms: true    # 开启发送失败退回    publisher-returns: true    listener:      simple:       # 手动确认        acknowledge-mode: manual        retry:           enabled: true

消息接收者确认

@Component@RabbitListener(queues = "testQueue")public class MqConfirmConsumer{    private static final Logger logger = LoggerFactory.getLogger(MqConfirmConsumer.class);        @RabbitHandler    public void receive(String msg, Channel channel, Message message) throws IOException     {        logger.info("receive message content:{}",message);                try        {            logger.info("开始消息确认");            channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);            logger.info("消息确认成功");        }        catch (Exception e)        {             logger.error("消息确认失败，即将再次返回队列中");               channel.basicNack(message.getMessageProperties().getDeliveryTag(), true, true);                     }    }}

说明：消息者确认消息有三种模式，分别为basicAck、basicNack、basicReject。

basicAck模式

表示成功确认，使用此回执方法后，消息会被rabbitmq broker删除。

void basicAck(long deliveryTag, boolean multiple) 复制代码

deliveryTag：消息投递序号，
multiple：是否批量确认，值为 true则会一次性ack所有小于当前消息deliveryTag的消息。

basicNack模式

表示失败确认，一般在消费消息异常时用到此方法，可以将消息重新投递入队列。

void basicNack(long deliveryTag, boolean multiple, boolean requeue)复制代码

deliveryTag：表示消息投递序号。
requeue：表示消息是否重新入队列，true表示重新投入队列中。
multiple：是否批量确认，true表示会一次性ack所有小于当前消息deliveryTag的消息。

basicReject模式

basicReject：拒绝消息，与basicNack区别在于不能进行批量操作，其他用法很相似。

void basicReject(long deliveryTag, boolean requeue)复制代码

deliveryTag：消息投递序号。
requeue：值为true表示消息重新入队列

测试

测试发送消息，消息发送者的确认信息如下：

c.s.f.r.config.RabbitConfirmConfig - exchange produce confirm message send successc.s.f.r.config.RabbitConfirmConfig - 数据内容:CorrelationData [id=88ea47a5-726d-44c5-9839-1f2a6bf942ed]c.s.f.r.config.RabbitConfirmConfig - 是否确认成功:truec.s.f.r.config.RabbitConfirmConfig - 错误原因:nullc.s.f.r.config.RabbitConfirmConfig - exchange produce confirm message send success复制代码

消费者的确认信息如下：

receive message content:(Body:'this is test message' MessageProperties [headers={spring_listener_return_correlation=0fcefb6d-acea-4eb2-8484-e3a82f8c584f, spring_returned_message_correlation=88ea47a5-726d-44c5-9839-1f2a6bf942ed}, contentType=text/plain, contentEncoding=UTF-8, contentLength=0, receivedDeliveryMode=PERSISTENT, priority=0, redelivered=false, receivedExchange=testDirect, receivedRoutingKey=testDirectRouting, deliveryTag=2, consumerTag=amq.ctag-dOwkSPuI1e0HR_1Ufu3Erw, consumerQueue=testQueue]) c.s.f.r.consumer.MqConfirmConsumer - 开始消息确认c.s.f.r.consumer.MqConfirmConsumer - 消息确认成功复制代码

消费者确认失败

如果消息确认在消费者确认失败，那么消息将会重写投递导导消息队列的首部。模拟消费者确认失败场景：

@Component@RabbitListener(queues = "testQueue")public class MqConfirmConsumer{    private static final Logger logger = LoggerFactory.getLogger(MqConfirmConsumer.class);        @RabbitHandler    public void receive(String msg, Channel channel, Message message) throws IOException     {        logger.info("receive message content:{}",message);                try        {            logger.info("开始消息确认");            int c=1/0;            channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);            logger.info("消息确认成功");        }        catch (Exception e)        {          logger.error("消息确认失败，即将再次返回队列中");                   channel.basicNack(message.getMessageProperties().getDeliveryTag(), true, true);          }    }}复制代码

查看执行结果：

c.s.f.r.consumer.MqConfirmConsumer - receive message content:(Body:'this is test message' MessageProperties [headers={spring_listener_return_correlation=0fcefb6d-acea-4eb2-8484-e3a82f8c584f, spring_returned_message_correlation=39d4cdd1-cbeb-4090-91ea-9e5d0bed785c}, contentType=text/plain, contentEncoding=UTF-8, contentLength=0, receivedDeliveryMode=PERSISTENT, priority=0, redelivered=false, receivedExchange=testDirect, receivedRoutingKey=testDirectRouting, deliveryTag=1, consumerTag=amq.ctag-e5GtG455pkm7eWfY3xGleg, consumerQueue=testQueue])c.s.f.r.consumer.MqConfirmConsumer - 开始消息确认c.s.f.r.consumer.MqConfirmConsumer - 消息确认失败，即将再次返回队列中复制代码

消息已经重新返回队列中。我们查看队列信息具体如下：

说明：我们可以看到消息为Unacked状态，消息又会重新会被消费，然后确认失败，又重新被消费，导致死循环。

解决办法

针对这种情况，我们将如何处理呢？我们手动确认失败后，并将消息持久入到MySQL中通过定时任务做补偿。然后删除消息队列。具体修改如下：

 @RabbitHandler    public void receive(String msg, Channel channel, Message message) throws IOException     {        logger.info("receive message content:{}",message);                try        {            logger.info("开始消息确认");            int c=1/0;            channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);            logger.info("消息确认成功");        }        catch (Exception e)        {            if (message.getMessageProperties().getRedelivered())             {                logger.error("消息确认失败,拒绝处理");              //执行持久化处理                channel.basicReject(message.getMessageProperties().getDeliveryTag(), false);           }            else             {                logger.error("消息确认失败，即将再次返回队列中");                channel.basicNack(message.getMessageProperties().getDeliveryTag(), false, true);             }        }    }复制代码

修改后执行结果如下：