五张图带你理解 RocketMQ 延时消息机制

最新推荐文章于 2024-08-08 18:04:59 发布

肥肥技术宅

最新推荐文章于 2024-08-08 18:04:59 发布

阅读量403

点赞数 1

分类专栏： java 文章标签： java kafka 分布式

本文链接：https://blog.csdn.net/m0_71777195/article/details/125259397

版权

本文深入探讨了RocketMQ实现延时消息的机制，包括生产者如何设置延时级别，Broker如何处理和调度延时消息，以及遇到超过最大延时级别时的处理策略。延时消息通过修改消息属性并利用特定的调度任务来实现在预设时间后的投递，整个过程涉及到CommitLog、ConsumeQueue和延时队列的交互。

摘要由CSDN通过智能技术生成

今天来聊一聊 RocketMQ 的延时消息是怎么实现的。

延时消息是指发送到 RocketMQ 后不会马上被消费者拉取到，而是等待固定的时间，才能被消费者拉取到。

延时消息的使用场景很多，比如电商场景下关闭超时未支付的订单，某些场景下需要在固定时间后发送提示消息。

1.生产者

首先看一个生产者发送延时消息的官方示例代码：

public static void main(String[] args) throws Exception {
 // Instantiate a producer to send scheduled messages
 DefaultMQProducer producer = new DefaultMQProducer("ExampleProducerGroup");
 // Launch producer
 producer.start();
 int totalMessagesToSend = 100;
 for (int i = 0; i < totalMessagesToSend; i++) {
  Message message = new Message("TestTopic", ("Hello scheduled message " + i).getBytes());
  // This message will be delivered to consumer 10 seconds later.
  message.setDelayTimeLevel(3);
  // Send the message
  producer.send(message);
 }
 
 // Shutdown producer after use.
 producer.shutdown();
}

从上面的代码可以看到，跟普通消息不一样的是，消息设置 setDelayTimeLevel 属性值，这里设置为 3，这里最终将 3 这个延时级别复制给了 DELAY 属性。

关于延时级别，可以看下面这个定义：

//MessageStoreConfig类
private String messageDelayLevel = "1s 5s 10s 30s 1m 2m 3m 4m 5m 6m 7m 8m 9m 10m 20m 30m 1h 2h";

这里延时级别有 18 个，上面的示例代码中延迟级别是 3，消息会延迟 10s 后消费者才能拉取。

2.Broker 处理

2.1 写入消息

Broker 收到消息后，会将消息写入 CommitLog。在写入时，会判断消息 DELAY 属性是否大于 0。代码如下：

//CommitLog 类
if (msg.getDelayTimeLevel() > 0) {
 if (msg.getDelayTimeLevel() > this.defaultMessageStore.getScheduleMessageService().getMaxDelayLevel()) {
  msg.setDelayTimeLevel(this.defaultMes