RabbitMQ发布确认

news2024/12/24 21:48:51

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

发布确认的策略

开启发布确认的方法

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

channel.confirmSelect();

单个确认发布

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

/**
 * 单个分发确认
 */
public static void publicMessageIndividually() throws Exception {
    Channel channel = RabbitMqUtil.getChannel();
    String queue_name = UUID.randomUUID().toString();
    channel.queueDeclare(queue_name,false,false,false,null);
    //开启分发确认
    channel.confirmSelect();
    long start = System.currentTimeMillis();
    for(int i=0;i<100;i++){
        String message=i+"";
        channel.basicPublish("",queue_name,null,message.getBytes(StandardCharsets.UTF_8));
        boolean flag = channel.waitForConfirms();
        if(flag){
            System.out.println("消息分发成功:"+message);
        }
    }
    long end = System.currentTimeMillis();
    System.out.println("耗时："+(end-start)+"ms");
}

批量确认发布

    /**
     * 批量分发确认
     */
    public static void publicMessageBatch() throws Exception {
        Channel channel = RabbitMqUtil.getChannel();
        String queue_name = UUID.randomUUID().toString();
        channel.queueDeclare(queue_name,false,false,false,null);
        //开启分发确认
        channel.confirmSelect();
        long start = System.currentTimeMillis();
        int outstandingMessageCount = 0;
        for(int i=0;i<100;i++){
            String message=i+"";
            channel.basicPublish("",queue_name,null,message.getBytes(StandardCharsets.UTF_8));
            outstandingMessageCount++;
            if(i%10==0) {
                channel.waitForConfirms();
                outstandingMessageCount=0;
            }
        }
        if(outstandingMessageCount>0) channel.waitForConfirms();
        long end = System.currentTimeMillis();
        System.out.println("耗时："+(end-start)+"ms");
    }

缺点：不知道哪一个消息出现错误。
相比于单个确认发布耗时短

异步分发确认

异步确认虽然编程逻辑比上两个要复杂，但是性价比最高，无论是可靠性还是效率都没得说，他是利用回调函数来达到消息可靠性传递的，这个中间件也是通过函数回调来保证是否投递成功

    /**
     * 异步分发确认
     * @throws Exception
     */
    public static void publicMessageAsync() throws Exception {
        Channel channel = RabbitMqUtil.getChannel();
        String queue_name = UUID.randomUUID().toString();
        channel.queueDeclare(queue_name,false,false,false,null);
        //开启分发确认
        channel.confirmSelect();
        //消息分发成功回调 第一个参数 消息序列号，第二个参数 是否是批量分发
        ConfirmCallback askCallback=(deliveryTag,multiple)->{
            System.out.println("消息成功分发："+deliveryTag);
        };
        //消息分发失败回调 第一个参数 消息序列号，第二个参数 是否是批量分发
        ConfirmCallback nackCallback=(deliveryTag,multiple)->{
            System.out.println("消息分发失败："+deliveryTag);
        };
        //添加监听器 第一个参数成功时的回调，第二个参数失败时的回调
        channel.addConfirmListener(askCallback,nackCallback);
        long start = System.currentTimeMillis();
        int outstandingMessageCount = 0;
        for(int i=0;i<100;i++){
            String message=i+"";
            channel.basicPublish("",queue_name,null,message.getBytes(StandardCharsets.UTF_8));
            outstandingMessageCount++;
        }
        long end = System.currentTimeMillis();
        System.out.println("耗时："+(end-start)+"ms");
    }

处理异步未确认消息

最好的解决的解决方案就是把未确认的消息放到一个基于内存的能被发布线程访问的队列

    public static void publicMessageAsync() throws Exception {
        Channel channel = RabbitMqUtil.getChannel();
        String queue_name = UUID.randomUUID().toString();
        channel.queueDeclare(queue_name,false,false,false,null);
        //开启分发确认
        channel.confirmSelect();
        /**
         * 线程安全有序的一个哈希表，适用于高并发的情况
         * 1.轻松的将序号与消息进行关联
         * 2.轻松批量删除条目 只要给到序列号
         * 3.支持并发访问
         */
        ConcurrentSkipListMap<Long, String> outstandingConfirms = new ConcurrentSkipListMap<>();
        //消息分发成功回调
        ConfirmCallback askCallback=(deliveryTag,multiple)->{
            if (multiple) {
                //返回的是小于等于当前序列号的未确认消息 是一个 map
                ConcurrentNavigableMap<Long, String> confirmed = outstandingConfirms.headMap(deliveryTag, true);
                //清除该部分未确认消息
                confirmed.clear();
            }else{
                //只清除当前序列号的消息
                outstandingConfirms.remove(deliveryTag);
            }
            System.out.println("消息成功分发："+deliveryTag);
        };
        //消息分发失败回调
        ConfirmCallback nackCallback=(deliveryTag,multiple)->{
            System.out.println("消息分发失败："+deliveryTag);
        };
        //添加监听器
        channel.addConfirmListener(askCallback,nackCallback);
        long start = System.currentTimeMillis();
        for(int i=0;i<100;i++){
            String message=i+"";
            outstandingConfirms.put(channel.getNextPublishSeqNo(),message);
            channel.basicPublish("",queue_name,null,message.getBytes(StandardCharsets.UTF_8));
        }
        long end = System.currentTimeMillis();
        System.out.println("耗时："+(end-start)+"ms");
    }