RabbitMQ发布确认

发布确认

• 1、发布确认原理
• 2、发布确认策略
• • 2.1 开启发布确认的方法
  • 2.2 单个确认发布
  • 2.3 批量确认发布
  • 2.4 异步确认发布
• 3、三种发布方式确认速度对比

1、发布确认原理

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

2、发布确认策略

2.1 开启发布确认的方法

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

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

本文所有代码都会用到一个工具类，如下：

/**
 * 连接工厂创建信道的工具类
 */
public class RabbitMqUtils {
   //得到一个连接的 channel
   public static Channel getChannel() throws Exception{
         //创建一个连接工厂
         ConnectionFactory factory = new ConnectionFactory();
         factory.setHost("192.168.159.33");
         factory.setUsername("admin");
         factory.setPassword("123");
         Connection connection = factory.newConnection();
         Channel channel = connection.createChannel();
         return channel;
   }
}

2.2 单个确认发布

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

 //单个确认
    public static void publishMessageIndividually() throws Exception{
        Channel channel = RabbitMqUtils.getChannel();
        //队列的声明
        String queueName = UUID.randomUUID().toString();
        channel.queueDeclare(queueName,true,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());
            //单个消息就马上进行发布确认
            boolean flag = channel.waitForConfirms();
            if(flag){
                System.out.println("消息发送成功");
            }
        }
        //结束时间
        long end = System.currentTimeMillis();
        System.out.println("发布"+MESSAGE_COUNT+"个单独确认消息，耗时:"+(end-begin)+"ms");
    }

发布1000个单独确认消息，耗时:1517ms

2.3 批量确认发布

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

 //批量发布确认
    public static void publishMessageBatch() throws Exception{
        Channel channel = RabbitMqUtils.getChannel();
        //队列的声明
        String queueName = UUID.randomUUID().toString();
        channel.queueDeclare(queueName,true,false,false,null);

        //开启发布确认
        channel.confirmSelect();
        //开始时间
        long begin = System.currentTimeMillis();


        //批量确认消息的大小
        int batchSize=100;

        //批量发送消息  批量发布确认
        for (int i = 0; i < MESSAGE_COUNT; i++) {
            String message=i+"";
            channel.basicPublish("",queueName,null,message.getBytes());

            //判断达到100条消息的时候  批量确认一次
            if(i%batchSize==0){
                //发布确认
                channel.waitForConfirms();
            }
        }
        //结束时间
        long end = System.currentTimeMillis();
        System.out.println("发布"+MESSAGE_COUNT+"个批量确认消息，耗时:"+(end-begin)+"ms");
    }

发布1000个批量确认消息，耗时:138ms

2.4 异步确认发布

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

//异步发布确认
    public static void publishMessageAsync() throws Exception{
        Channel channel = RabbitMqUtils.getChannel();
        //队列的声明
        String queueName = UUID.randomUUID().toString();
        channel.queueDeclare(queueName,true,false,false,null);

        //开启发布确认
        channel.confirmSelect();
        /**
         * 线程安全有序的一份哈希表 适用于高并发的情况下
         * 1、轻松的将序号与消息进行关联
         * 2、轻松的批量删除条目  只要给到序号
         * 3、支持高并发(多线程)
         */
        ConcurrentSkipListMap<Long,String> outstandingConfirms=new ConcurrentSkipListMap<>();



        //消息确认成功 回调函数
        ConfirmCallback ackCallback=(deliveryTag, multiple) -> {
            if(multiple){   //批量
                //2、删除掉已经确认的消息  剩下的就是未确认的消息
                ConcurrentNavigableMap<Long, String> confirmed =
                        outstandingConfirms.headMap(deliveryTag);
                confirmed.clear();
            }else{          //单个
                outstandingConfirms.remove(deliveryTag);
            }
            System.out.println("确认的消息："+deliveryTag);
        };
        //消息确认失败 回调函数
        /**
         * 参数1：消息的标记
         * 参数2：是否为批量确认
         */
        ConfirmCallback nackCallback=(deliveryTag, multiple) -> {
            //3、打印一下未确认的消息都有哪些
            String message = outstandingConfirms.get(deliveryTag);
            System.out.println("未确认的消息是："+message+":::未确认的消息tag："+deliveryTag);
        };

        //准备消息的监听器  监听哪些消息成功了 哪些消息失败了
        /**
         * 参数1：监听哪些消息成功了
         * 参数2：哪些消息失败了
         */
        channel.addConfirmListener(ackCallback,nackCallback);

        //开始时间
        long begin = System.currentTimeMillis();
        //批量发送消息
        for (int i = 0; i < MESSAGE_COUNT; i++) {
            String message="消息"+i;
            channel.basicPublish("",queueName,null,message.getBytes());
            //1、此处记录下所有要发送的消息 消息的总和
            outstandingConfirms.put(channel.getNextPublishSeqNo(),message);
        }


        //结束时间
        long end = System.currentTimeMillis();
        System.out.println("发布"+MESSAGE_COUNT+"个异步发布确认消息，耗时:"+(end-begin)+"ms");
    }

  发布1000个异步发布确认消息，耗时:58ms
  这里还处理了异步未确认的消息，将未确认的消息放到一个基于内存的能被发布线程访问的队列，比如说用 ConcurrentLinkedQueue 这个队列在 confirm callbacks 与发布线程之间进行消息的传递。

3、三种发布方式确认速度对比

单独发布消息
  同步等待确认，简单，但吞吐量非常有限。
批量发布消息
  批量同步等待确认，简单，合理的吞吐量，一旦出现问题但很难推断出是哪条消息出现了问题。
异步处理：
  最佳性能和资源使用，在出现错误的情况下可以很好地控制，但是实现起来稍微难些