RabbitMQ的发布确认机制

一、发布确认原理

• 生产者将信道设置成 confirm 模式，一旦信道进入 confirm 模式，所有在该信道上面发布的消息都将会被指派一个唯一的 ID(从 1 开始)。
• 一旦消息被投递到所有匹配的队列之后，broker就会发送一个确认给生产者(包含消息的唯一 ID)，这就使得生产者知道消息已经正确到达目的队列。
• 如果消息和队列是可持久化的，那么确认消息会在将消息写入磁盘之后发出，broker 回传给生产者的确认消息中 delivery-tag 域包含了确认消息的序列号，此外 broker 也可以设置basic.ack 的 multiple 域，表示到这个序列号之前的所有消息都已经得到了处理。

二、发布确认策略

开启发布确认策略

channel开启发布确认模式

单个发布确认

是一种同步确认发布的方式，也就是发布一个消息之后只有它被确认发布，后续的消息才能继续发布,waitForConfirmsOrDie(long)这个方法只有在消息被确认的时候才返回，如果在指定时间范围内这个消息没有被确认那么它将抛出异常。

public class Producer {

    public static final String QUEUE_NAME = "publish_confirm";

    public static void main(String[] args) throws Exception {
        singleConfirm();

    }

    public static void singleConfirm() throws Exception {
        Channel channel = RabbitMqUtils.getChannel();
        channel.queueDeclare(QUEUE_NAME, false, false, false, null);
        channel.confirmSelect();

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

        //批量发消息
        for (int i = 1; i <= 1000; i++) {
            String message = i + "";
            channel.basicPublish("", QUEUE_NAME, null, message.getBytes());
            //单个消息发布马上进行确认
            boolean flag = channel.waitForConfirms();
            if (flag) {
                System.out.println("消息发送成功：" + message);
            }
        }

        //结束时间
        long end = System.currentTimeMillis();

        System.out.println("发布 1000 个单独确认消息，总耗时：" + (end - begin) + "ms");

    }
}

批量发布确认

当发生故障导致发布出现问题时，不知道是哪个消息出现问题了，必须将整个批处理保存在内存中，以记录重要的信息而后重新发布消息。这种方案仍然是同步的，也一样阻塞消息的发布。

public static void batchConfirm() throws Exception {
        Channel channel = RabbitMqUtils.getChannel();
        channel.queueDeclare(QUEUE_NAME, false, false, false, null);
        channel.confirmSelect();

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

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

        //批量发消息,批量发布确认
        for (int i = 1; i <= 1000; i++) {
            String message = i + "";
            channel.basicPublish("", QUEUE_NAME, null, message.getBytes());
            //每100条消息确认一次
            if (i % batchSize == 0){
                channel.waitForConfirms();
                System.out.println("消息发送成功：" + message);
            }
        }

        //结束时间
        long end = System.currentTimeMillis();

        System.out.println("发布 1000 个批量确认消息，总耗时：" + (end - begin) + "ms");

}

异步确认发布

利用回调函数来达到消息可靠性传递的，这个中间件也是通过函数回调来保证是否投递成功

public static void asyncConfirm() throws Exception {
        Channel channel = RabbitMqUtils.getChannel();
        channel.queueDeclare(QUEUE_NAME, false, false, false, null);
        channel.confirmSelect();

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

        //准备消息监听器，监听成功和失败的消息
        //声明接收成功消息回调函数 和 接收失败的回调函数
        ConfirmCallback ackCallback = (deliveryTag, multiple) -> {
            System.out.println("已确认消息：" + deliveryTag);
        };
        ConfirmCallback nackCallback = (deliveryTag, multiple) -> {
            System.out.println("未确认消息：" + deliveryTag);
        };
        channel.addConfirmListener(ackCallback,nackCallback);

        //批量发消息,异步发布确认
        for (int i = 1; i <= 1000; i++) {
            String message = i + "";
            channel.basicPublish("", QUEUE_NAME, null, message.getBytes());
        }

        //结束时间
        long end = System.currentTimeMillis();

        System.out.println("发布 1000 个异步确认消息，总耗时：" + (end - begin) + "ms");

}

对于异步未确认的消息，可以把未确认的消息放到一个基于内存的能被发布线程访问的队列，使得这个队列在 confirm callbacks回调函数与发布消息线程之间进行消息的传递。两个线程交互可以使用ConcurrentLinkedQueue。