在rocketmq总topic下包含多个queue,发送消息时如果不指定会随机发送到不同queue,此时消费者无法保证顺序消费。参考下图,发送时发送了producetMessage0-producetMessage9,但消费时顺序变了。
解决顺序消费的方法比较简单,第一步是在生产者发送时将消息发送到同一个queue。要达到该效果可以实现MessageQueueSelector,在其select方法中自定义发送到哪个queue。代码如下:
public class IdMessageQueueSelector implements MessageQueueSelector {
@Override
public MessageQueue select(List<MessageQueue> mqs, Message msg, Object arg) {
Integer id = (Integer) arg;
int index = id % mqs.size(); // 同一id消息发送到同一queue
return mqs.get(index);
}
}
定义好后生产者在send发送时使用IdMessageQueueSelector发送
public static void main(String[] args) throws MQClientException, InterruptedException {
DefaultMQProducer producer = new DefaultMQProducer("producer_test");
producer.setNamesrvAddr("ip:9876");
producer.setInstanceName(UUID.randomUUID().toString());
producer.start();
IdMessageQueueSelector idMessageQueueSelector = new IdMessageQueueSelector();
int id = 1;
for (int i = 0; i < 10; i++) {
try {
Message msg = new Message("TopicTest",
("producetMessage" + i).getBytes(RemotingHelper.DEFAULT_CHARSET)// body
);
SendResult sendResult = producer.send(msg, idMessageQueueSelector, id);
System.out.println(sendResult);
} catch (Exception e) {
e.printStackTrace();
}
}
producer.shutdown();
}
至此生产者已经保证发送时消息到同一queue中,消费者只需监听MessageListenerOrderly即可
public class IdMessageListener implements MessageListenerOrderly {
@Override
public ConsumeOrderlyStatus consumeMessage(List<MessageExt> msgs, ConsumeOrderlyContext context) {
if (CollectionUtils.isEmpty(msgs)) {
return ConsumeOrderlyStatus.SUCCESS;
}
// 监听处理逻辑
for (Message msg : msgs) {
System.out.println(new String(msg.getBody()) + " === date:" + new Date());
}
return ConsumeOrderlyStatus.SUCCESS;
}
}
监听完成后启动消费者
public static void main(String[] args) throws InterruptedException, MQClientException {
DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("consumer_test");
consumer.setConsumeFromWhere(ConsumeFromWhere.CONSUME_FROM_FIRST_OFFSET);
consumer.setNamesrvAddr("ip:9876");
consumer.setInstanceName(UUID.randomUUID().toString());
consumer.subscribe("TopicTest", "*");
consumer.registerMessageListener(new IdMessageListener());
consumer.start();
System.out.println("start success");
}