发布确认原理

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

发布确认的策略

开启发布确认的方法

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

单个确认发布

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

  1. // 批量发消息个数
  2. private static final int MESSAGE_COUNT = 1000;
  3. public static void publishMessageIndividually() throws Exception {
  4. Channel channel = RabbitMqUtils.getChannel();
  5. String queueName = UUID.randomUUID().toString();
  6. channel.queueDeclare(queueName, true, false, false, null);
  7. //开启发布确认
  8. channel.confirmSelect();
  9. // 开始时间
  10. long begin = System.currentTimeMillis();
  11. for (int i = 0; i < MESSAGE_COUNT; i++) {
  12. String message = i + "";
  13. channel.basicPublish("", queueName, null, message.getBytes());
  14. // 服务端返回 false 或超时时间内未返回,生产者可以消息重发
  15. boolean flag = channel.waitForConfirms();
  16. if (flag) {
  17. System.out.println("消息发送成功:" + i);
  18. }
  19. }
  20. // 结束时间
  21. long end = System.currentTimeMillis();
  22. System.out.println("发布" + MESSAGE_COUNT + "个单独确认消息,耗时" + (end - begin) + "ms");
  23. }

批量确认发布

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

  1. public static void publishMessageBatch() throws Exception {
  2. Channel channel = RabbitMqUtils.getChannel();
  3. String queueName = UUID.randomUUID().toString();
  4. channel.queueDeclare(queueName, true, false, false, null);
  5. //开启发布确认
  6. channel.confirmSelect();
  7. // 开始时间
  8. long begin = System.currentTimeMillis();
  9. // 批量确认消息大小
  10. int batchSize = 100;
  11. for (int i = 0; i < MESSAGE_COUNT; i++) {
  12. String message = i + "";
  13. channel.basicPublish("", queueName, null, message.getBytes());
  14. // 判断达到100条消息的时候 批量确认一次
  15. if (i % batchSize == 0) {
  16. // 服务端返回 false 或超时时间内未返回,生产者可以消息重发
  17. boolean flag = channel.waitForConfirms();
  18. if (flag) {
  19. System.out.println("消息发送成功:" + i);
  20. }
  21. }
  22. }
  23. // 结束时间
  24. long end = System.currentTimeMillis();
  25. System.out.println("发布" + MESSAGE_COUNT + "个批量确认消息,耗时" + (end - begin) + "ms");
  26. }

异步确认发布

异步确认虽然编程逻辑比上两个要复杂,但是性价比最高,无论是可靠性还是效率都没得说, 他是利用回调函数来达到消息可靠性传递的,这个中间件也是通过函数回调来保证是否投递成功, 下面就让我们来详细讲解异步确认是怎么实现的。
05 发布确认 - 图2

  1. public static void publishMessageAsync() throws Exception {
  2. Channel channel = RabbitMqUtils.getChannel();
  3. String queueName = UUID.randomUUID().toString();
  4. channel.queueDeclare(queueName, true, false, false, null);
  5. //开启发布确认
  6. channel.confirmSelect();
  7. // 开始时间
  8. long begin = System.currentTimeMillis();
  9. // 监听成功
  10. ConfirmCallback ackCallback = (deliveryTag, multiple) -> {
  11. System.out.println("确认的消息:" + deliveryTag);
  12. };
  13. // 监听失败
  14. ConfirmCallback nackCallback = (deliveryTag, multiple) -> {
  15. System.out.println("未确认的消息:" + deliveryTag);
  16. };
  17. // 准备消息的监听器 监听哪些消息成功了 哪些消息失败了
  18. channel.addConfirmListener(ackCallback, nackCallback);
  19. for (int i = 0; i < MESSAGE_COUNT; i++) {
  20. String message = i + "";
  21. channel.basicPublish("", queueName, null, message.getBytes());
  22. }
  23. // 结束时间
  24. long end = System.currentTimeMillis();
  25. System.out.println("发布" + MESSAGE_COUNT + "个异步确认消息,耗时" + (end - begin) + "ms");
  26. }

如何处理异步未确认消息

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

  1. public static void publishMessageAsync() throws Exception {
  2. Channel channel = RabbitMqUtils.getChannel();
  3. String queueName = UUID.randomUUID().toString();
  4. channel.queueDeclare(queueName, true, false, false, null);
  5. //开启发布确认
  6. channel.confirmSelect();
  7. // 线程安全有序的哈希表 适用于高并发的情况下
  8. ConcurrentSkipListMap<Long, String> outStandingConfirms = new ConcurrentSkipListMap<>();
  9. // 监听成功
  10. ConfirmCallback ackCallback = (deliveryTag, multiple) -> {
  11. // 2.删除已经确认的消息 剩下的就是未确认的消息
  12. if (multiple) {
  13. ConcurrentNavigableMap<Long, String> map = outStandingConfirms.headMap(deliveryTag);
  14. map.clear();
  15. } else {
  16. outStandingConfirms.remove(deliveryTag);
  17. }
  18. System.out.println("确认的消息:" + deliveryTag);
  19. };
  20. // 监听失败
  21. ConfirmCallback nackCallback = (deliveryTag, multiple) -> {
  22. // 3.打印未确认的消息有哪些
  23. String message = outStandingConfirms.get(deliveryTag);
  24. System.out.println(message + " 未确认的消息tag:" + deliveryTag);
  25. };
  26. // 准备消息的监听器 监听哪些消息成功了 哪些消息失败了
  27. channel.addConfirmListener(ackCallback, nackCallback);
  28. // 开始时间
  29. long begin = System.currentTimeMillis();
  30. for (int i = 0; i < MESSAGE_COUNT; i++) {
  31. String message = i + "";
  32. channel.basicPublish("", queueName, null, message.getBytes());
  33. // 1.记录所有要发送的消息
  34. outStandingConfirms.put(channel.getNextPublishSeqNo(), message);
  35. }
  36. // 结束时间
  37. long end = System.currentTimeMillis();
  38. System.out.println("发布" + MESSAGE_COUNT + "个异步确认消息,耗时" + (end - begin) + "ms");
  39. }

以上 3 种发布确认速度对比

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

  1. public static void main(String[] args) throws Exception {
  2. // 1.单个确认 发布1000个单独确认消息,耗时613ms
  3. // publishMessageIndividually();
  4. // 2.批量确认 发布1000个批量确认消息,耗时113ms
  5. // publishMessageBatch();
  6. // 3.异步批量确认 发布1000个异步确认消息,耗时44ms
  7. publishMessageAsync();
  8. }