RabbitMQ发布确认

RabbitMQ发布确认

当生产者将信道设置为 confirm 模式后，该信道上发布的每条消息都会被赋予一个唯一的 ID（从 1 开始递增）。当消息成功投递到所有匹配的队列后，broker 会向生产者发送一条确认消息，其中包含消息的唯一 ID。这样一来，生产者就能确认消息已正确到达目的队列。如果消息和队列是持久化的，那么确认消息会在将消息写入磁盘后发出。broker 回传给生产者的确认消息中的 delivery-tag 域包含了确认消息的序列号，此外，broker 还可以通过设置 basic.ack 的 multiple 域来表示这个序列号之前的所有消息都已被处理。

confirm 模式的主要优势在于其异步特性。生产者应用程序在发布一条消息后，可以在等待信道返回确认的同时继续发送下一条消息。当消息最终被确认时，生产者应用程序可以通过回调方法处理该确认消息。如果 RabbitMQ 由于内部错误导致消息丢失，则会发送一条 nack 消息，生产者应用程序同样可以在回调方法中处理该 nack 消息。

开启确认发布的方法

//开启发布确认
channel.confirmSelect();

单个确认发布

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

public static void individual(){
        try {
            Channel channel = RabbitMqUtils.getChannel();
            /**
             * 生成一个队列
             * 1.队列名称
             * 2.durable 队列是否持久化 默认消息存储在内存中
             * 3.该队列是否只供一个消费者进行消费 是否进行共享 true 可以多个消费者消费
             * 4.是否自动删除 最后一个消费者断开连接以后 该队列是否自动删除 true 自动删除
             * 5.其他参数
             */
            //设置队列持久化
            boolean durable = true;
            channel.queueDeclare(QUEUE_NAME,durable,false,false,null);
            //开启发布确认
            channel.confirmSelect();
            long beginTime = System.currentTimeMillis();
//            //从控制台输入消息
            for(int i = 0;i<1000;i++){
                String messageC1 = "C1_"+i;
                channel.basicPublish("",QUEUE_NAME,
                MessageProperties.PERSISTENT_TEXT_PLAIN,
                messageC1.getBytes());
                //服务端返回 false 或超时时间内未返回，生产者可以消息重发
                boolean flag = channel.waitForConfirms();
                if(flag){
                    System.out.println("消息发送完成:"+messageC1);
                }
            }
            long endTime = System.currentTimeMillis();
            long spendTime = endTime - beginTime;
            System.out.println("单个确认发布花费："+spendTime +"ms");
        } catch (IOException e) {
            e.printStackTrace();
        } catch (TimeoutException | InterruptedException e) {
            e.printStackTrace();
        }
    }

批量确认发布

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

public static void batchConfirm(){
        try {
            Channel channel = RabbitMqUtils.getChannel();
            /**
             * 生成一个队列
             * 1.队列名称
             * 2.durable 队列是否持久化 默认消息存储在内存中
             * 3.该队列是否只供一个消费者进行消费 是否进行共享 true
              可以多个消费者消费
             * 4.是否自动删除 最后一个消费者端开连接以后 
             该队列是否自动删除 true 自动删除
             * 5.其他参数
             */
            //设置队列持久化
            boolean durable = true;
            channel.queueDeclare(QUEUE_NAME,durable,false,false,null);
            //开启发布确认
            channel.confirmSelect();
            long beginTime = System.currentTimeMillis();
            int batchSize = 100;
//            //从控制台输入消息
            for(int i = 0;i<1000;i++){
                String messageC1 = "C1_"+i;
                channel.basicPublish("",QUEUE_NAME,
                MessageProperties.PERSISTENT_TEXT_PLAIN,
                messageC1.getBytes());
                //服务端返回 false 或超时时间内未返回，生产者可以消息重发
                if(i % batchSize == 0){
                    boolean flag = channel.waitForConfirms();
                    if(flag){
                         System.out.println("批量消息发送完成:"+messageC1 +"发送数量:"+i);
                    }
                }

            }
            long endTime = System.currentTimeMillis();
            long spendTime = endTime - beginTime;
            System.out.println("批量确认发布花费："+spendTime +"ms");
        } catch (IOException e) {
            e.printStackTrace();
        } catch (TimeoutException | InterruptedException e) {
            e.printStackTrace();
        }
    }

异步确认发布

异步确认虽然编程逻辑比上两个要复杂，但是性价比最高，无论是可靠性还是效率都有很好的保障，异步确认利用回调函数来达到消息可靠性传递的。

public static void confirmAsync(){
        try {
            Channel channel = RabbitMqUtils.getChannel();
            /**
             * 生成一个队列
             * 1.队列名称
             * 2.durable 队列是否持久化 默认消息存储在内存中
             * 3.该队列是否只供一个消费者进行消费
              是否进行共享 true 可以多个消费者消费
             * 4.是否自动删除 
             最后一个消费者端开连接以后 该队列是否自动删除 true 自动删除
             * 5.其他参数
             */
            //设置队列持久化
            boolean durable = true;
            channel.queueDeclare(QUEUE_NAME,durable,false,false,null);
            //开启发布确认
            channel.confirmSelect();

            /**
             * 线程安全有序的一个哈希表，适用于高并发的情况
             * 1.轻松的将序号与消息进行关联
             * 2.轻松批量删除条目 只要给到序列号
             * 3.支持并发访问
             */
            ConcurrentSkipListMap<Long,String> outstandingConfirm
             = new ConcurrentSkipListMap<>();

            ConfirmCallback confirmCallback = (deliveryTag, multiple) -> {
                //是否批量的消息
                if(multiple){
                    /**
                     * 确认收到消息的一个回调
                     * 1.消息序列号
                     * 2.true 可以确认小于等于当前序列号的消息
                     * false 确认当前序列号消息
                     */
                    ConcurrentNavigableMap<Long,String> confirmed 
                    = outstandingConfirm.headMap(deliveryTag,true);
                    confirmed.clear();
                }else {
                    //删除已确认的消息，剩下的就是未确认的消息
                    outstandingConfirm.remove(deliveryTag);
                }
            };

            /**
             * 未确认的消息
             */
            ConfirmCallback neckCallback = (deliveryTag, multiple) -> {
                String neckMessage = outstandingConfirm.get(deliveryTag);
                System.out.println("未确认的消息:"+neckMessage
                +" tag:"+deliveryTag);
            };

            /**
             * 添加一个异步确认的监听器
             * 1.确认收到消息的回调
             * 2.未收到消息的回调
             */
            channel.addConfirmListener(confirmCallback,neckCallback);
            long beginTime = System.currentTimeMillis();
//            //从控制台输入消息
            for(int i = 0;i<1000;i++){
                String messageC1 = "C1_"+i;
//                System.out.println("异步消息发送完成:"+messageC1 +"发送数量:"+i);
                //记录发送消息的
                outstandingConfirm.put(channel.getNextPublishSeqNo(),messageC1);
                channel.basicPublish("",QUEUE_NAME,MessageProperties.PERSISTENT_TEXT_PLAIN,messageC1.getBytes());
            }
            long endTime = System.currentTimeMillis();
            long spendTime = endTime - beginTime;
            System.out.println("异步消息发送花费时间："+spendTime +"ms");
        } catch (IOException e) {
            e.printStackTrace();
        } catch (TimeoutException e) {
            e.printStackTrace();
        }
    }

三种方式的测试时间对比

单个确认发布花费：2571ms
批量确认发布花费：100ms
异步消息发送花费时间：62ms