消息中间件Rabbit(五)-保证消息不丢失之发布确认

悠悠 2023-10-06 21:09 39阅读 0赞

## 一、发布确认原理 ##

生产者将信道设置成 confirm 模式，一旦信道进入 confirm 模式，所有在该信道上面发布的消息都将会被指派一个唯一的 ID(从 1 开始)，一旦消息被投递到所有匹配的队列之后，broker 就会发送一个确认给生产者(包含消息的唯一 ID)，这就使得生产者知道消息已经正确到达目的队列了，如果消息和队列是可持久化的，那么确认消息会在将消息写入磁盘之后发出，broker 回传给生产者的确认消息中 delivery-tag 域包含了确认消息的序列号，此外 broker 也可以设置basic.ack 的 multiple 域，表示到这个序列号之前的所有消息都已经得到了处理。 confirm 模式最大的好处在于他是异步的，一旦发布一条消息，生产者应用程序就可以在等信道返回确认的同时继续发送下一条消息，当消息最终得到确认之后，生产者应用便可以通过回调方法来处理该确认消息，如果 RabbitMQ 因为自身内部错误导致消息丢失，就会发送一条 nack 消息， 生产者应用程序同样可以在回调方法中处理该 nack 消息。

## 二、发布确认的策略 ##

### 2.1 开启发布确认的方法 ###

发布确认默认是没有开启的，如果要开启需要调用方法 confirmSelect，每当你要想使用发布确认，都需要在 channel 上调用该方法。

Channel channel = connection.createChannel();
    //开启发布确认
    channel.confirmSelect();

### 2.2 单个发布确认 ###

这是一种简单的确认方式，它是一种同步确认发布的方式，也就是发布一个消息之后只有它 被确认发布，后续的消息才能继续发布,waitForConfirmsOrDie(long)这个方法只有在消息被确认 的时候才返回，如果在指定时间范围内这个消息没有被确认那么它将抛出异常。 这种确认方式有一个最大的缺点就是:发布速度特别的慢，因为如果没有确认发布的消息就会 阻塞所有后续消息的发布，这种方式最多提供每秒不超过数百条发布消息的吞吐量。当然对于某 些应用程序来说这可能已经足够了。

public static void publishMessageIndividually() throws Exception {
        Connection connection= RabbitMqUtils.getConnection();
        Channel channel = connection.createChannel();
        String queueName = UUID.randomUUID().toString();
        channel.queueDeclare(queueName, false, false, false, null);
        //开启发布确认
         channel.confirmSelect();
         long begin = System.currentTimeMillis();
         for (int i = 0; i < MESSAGE_COUNT; i++)
          {
                String message = i + "";
                channel.basicPublish("", queueName, null, message.getBytes());
                //服务端返回 false 或超时时间内未返回，生产者可以消息重发
                boolean flag = channel.waitForConfirms();
              if(flag){
                    System.out.println("消息发送成功");
               }
          }
        long end = System.currentTimeMillis();
        System.out.println("发布" + MESSAGE_COUNT + "个单独确认消息,耗时" + (end -       begin)+"ms");
      RabbitMqUtils.closeConnectionAndChanel(connection,channel);
    }

### 2.3 批量确认发布 ###

上面那种方式非常慢，与单个等待确认消息相比，先发布一批消息然后一起确认可以极大地提高吞吐量，当然这种方式的缺点就是:当发生故障导致发布出现问题时，不知道是哪个消息出现问题了，我们必须将整个批处理保存在内存中，以记录重要的信息而后重新发布消息。当然这种方案仍然是同步的，也一样阻塞消息的发布。

### 2.4 异步确认发布 ###

异步确认无论是可靠性还是效率都比较高， 他是利用回调函数来达到消息可靠性传递的，这个中间件也是通过函数回调来保证是否投递成功， 下面就让我们来详细讲解异步确认是怎么实现的。

public static void publishMessageAsync() throws Exception{
            Connection connection= RabbitMqUtils.getConnection();
            Channel channel = connection.createChannel();
            String queueName = UUID.randomUUID().toString();
            channel.queueDeclare(queueName,false,false,false,null);
            //1. 开启发布确认
            channel.confirmSelect();
    
            //3. 使用线程安全的跳表类，保存信息
            /**
             *线程安全有序的一个哈希表，适用于高并发的情况
             *  * 1.轻松的将序号与消息进行关联
             *  * 2.轻松批量删除条目 只要给到序列号
             *  * 3.支持并发访问
             */
            ConcurrentSkipListMap<Long, String> outstandingConfirms = new ConcurrentSkipListMap<>();
    
            /**
             * 2.1 确认收到消息的一个回调
             * 参数 1.消息序列号
             *      2. 是否是批量处理
             */
            ConfirmCallback ackCallback = (sequenceNumber, multiple) -> {
                if (multiple) {
                //ConcurrentNavigableMap返回的是小于等于当前序列号的确认消息 是一个 map
                    ConcurrentNavigableMap<Long, String> confirmed =
                            outstandingConfirms.headMap(sequenceNumber, true);
                    //清除该部分已确认消息,outstandingConfirms的值就只有>sequenceNumber的值了
                    confirmed.clear();
                }else{
                    //只清除当前序列号的消息
                    outstandingConfirms.remove(sequenceNumber);
                }
            };
            /**消息确认失败 回调函数
             * 
             */
            ConfirmCallback nackCallback = (sequenceNumber, multiple) -> {
                String message = outstandingConfirms.get(sequenceNumber);
                System.out.println("发布的消息"+message+"未被确认，序列号"+sequenceNumber);
            };
        /**
        * 2. 添加一个异步确认的监听器，可以用来监听哪些消息成功啦，哪些消息失败了,异步通知
        * 参数：1.确认收到消息的回调
        * 2.未收到消息的回调
        */
            channel.addConfirmListener(ackCallback, nackCallback);
    
            long begin = System.currentTimeMillis();
            for (int i = 0; i < MESSAGE_COUNT; i++)
            {
                String message = i + "";
                /** 3.1
                 * channel.getNextPublishSeqNo()获取下一个消息的序列号
                 * 通过序列号与消息体进行一个关联
                 * 全部都是未确认的消息体
                 */
                outstandingConfirms.put(channel.getNextPublishSeqNo(),message);
                //4. 发布消息
                channel.basicPublish("", queueName, null, message.getBytes());
            }
    
    
            long end = System.currentTimeMillis();
            System.out.println("发布" + MESSAGE_COUNT + "个批量确认消息,耗时" + (end - begin) + "ms");
        
        RabbitMqUtils.closeConnectionAndChanel(connection,channel);
    }

### 2.5. 如何处理异步未确认消息 ###

最好的解决方案就是把未确认的消息放到一个基于内存的能被发布线程访问的队列， 比如说用 ConcurrentLinkedQueue 这个队列在 confirm callbacks 与发布线程之间进行消息的传递。2.4中使用的 ConcurrentLinkedQueue 队列就是ConcurrentSkipListMap。

### 2.6 3 种发布确认速度对比 ###

**单独发布消息** 同步等待确认，简单，但吞吐量非常有限。

**批量发布消息** 批量同步等待确认，简单，合理的吞吐量，一旦出现问题但很难推断出是那条 消息出现了问题。

**异步处理**： 最佳性能和资源使用，在出现错误的情况下可以很好地控制，但是实现起来稍微难些