RabbitMq发布确认

RabbitMq发布确认

发布确认原理

生产者将信道设置成 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 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");
}

批量确认发布

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

/**
 * 发布批量消息
 */
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");
}

异步确认发布

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

/**
 * 发布异步消息
 */
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> outstandingConfirm = new ConcurrentSkipListMap<>();

    /*
        消息确认成功回调
        1. 消息的标记
        2. 是否为批量确认
     */
    ConfirmCallback ackConfirmCallback = (deliveryTag, multiple) -> {
        // 判断是否批量确认
        if (multiple) {
            // 删除掉已经确认的消息 剩下的就是未确认的消息
            ConcurrentNavigableMap<Long, String> confirmed =
                    outstandingConfirm.headMap(deliveryTag);
            confirmed.clear();
        } else {
            outstandingConfirm.remove(deliveryTag);
        }
        System.out.println("确认的消息：" + deliveryTag);
    };
    /*
        消息确认失败回调
        1. 消息的标记
        2. 是否为批量确认
     */
    ConfirmCallback nackConfirmCallback = (deliveryTag, multiple) -> {
        // 打印未确认的消息
        String message = outstandingConfirm.get(deliveryTag);
        System.out.println("未确认的消息：" + message + "未确认的消息标记：" + deliveryTag);
    };

    /*
        准备消息的监听器 监听哪些消息成功、哪些消息失败
        1. 监听哪些消息成功
        2. 哪些消息失败
     */
    channel.addConfirmListener(ackConfirmCallback, nackConfirmCallback);

    // 开始时间
    long begin = System.currentTimeMillis();

    // 批量发送消息, 批量发布确认
    for (int i = 0; i < MESSAGE_COUNT; i++) {
        String message = i + "";
        channel.basicPublish("", queueName, null, message.getBytes());
        // 记录所有要发送的消息
        outstandingConfirm.put(channel.getNextPublishSeqNo(), message);
    }

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

如何处理异步未确认消息

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

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

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

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

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