首先引入rocketmq的依赖
<dependency>
<groupId>org.apache.rocketmq</groupId>
<artifactId>rocketmq-client</artifactId>
<version>4.9.2</version>
</dependency>
然后我们编写一个简单的生产者和消费者
@SpringBootTest
public class RocketMQTest {
/**
* 对于生产者 同一组的生产者可以向不同的topic队列发送消息
*/
@Test
public void produce() throws MQClientException, MQBrokerException, RemotingException, InterruptedException {
DefaultMQProducer producer = new DefaultMQProducer("test-producer-group");
producer.setNamesrvAddr(MQConstant.NAMESRV);
producer.start();
Message message = new Message("testTopic","一个简单的消息".getBytes());
SendResult sendResult = producer.send(message);
System.out.println(sendResult.getSendStatus());
producer.shutdown();
}
/**
* 对于消费者 同一组的消费者只能接收同一个topic的消息
* 并且如果存在多个消费者组,他们都监听同一个topic的消息
* 那么就可以选择使用 负载均衡策略 或者 广播策略
*/
@Test
public void consume() throws MQClientException, IOException {
//创建一个消费者
DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("test-producer-group");
consumer.setNamesrvAddr(MQConstant.NAMESRV);
// * 标识订阅这个主题中的所有消息 后期会有消息过滤
consumer.subscribe("testTopic", "*");
//设置一个监听器 (他会一直监听,然后是一个异步回调的机制)
//那么我们就不能让他start之后这个方法就返回结束 需要挂起当前的JVM(test模式得这样子)
//正常运行项目的时候项目的JVM会正常运行的 不需要挂起
consumer.registerMessageListener(new MessageListenerConcurrently() {
@Override
public ConsumeConcurrentlyStatus consumeMessage(List<MessageExt> list, ConsumeConcurrentlyContext context) {
//这个就是对应的消费方法 业务处理
//消息如果消费失败 那么就要重新放入到消费队列
System.out.println("我是消费者");
System.out.println(list.get(0).toString());
System.out.println("消息上下文"+context);
//返回值如果为null/报错/RECONSUMER_LATER 代表消费失败
//消息会重新回到队列 然后过一会在投递给当前消费者或者其他消费者
return ConsumeConcurrentlyStatus.RECONSUME_LATER;
}
});
//启动
consumer.start();
//挂起当前的JVM
System.in.read();
}
}
这里需要注意的是,对于Rocketmq,如果在你的监听器中,也就是这个MessageListenerConcurrently中,你的返回值为null,或者ConsumeConcurrentlyStatus.RECONSUME_LATER,亦或者抛出了一个异常,那么这条消息都会重新的被放回到我们的队列中,等待其他消费者或者当前消费者再一次消费。
MQ的消费模式可以大致分为两种,一种是推Push,一种是拉Pull。
Push是服务端【MQ】主动推送消息给客户端,优点是及时性较好,但如果客户端没有做好流控,一旦服务端推送大量消息到客户端时,就会导致客户端消息堆积甚至崩溃。
Pull是客户端需要主动到服务端取数据,优点是客户端可以依据自己的消费能力进行消费,但拉取的频率也需要用户自己控制,拉取频繁容易造成服务端和客户端的压力,拉取间隔长又容易造成消费不及时。
Push模式也是基于pull模式的,只能客户端内部封装了api,一般场景下,上游消息生产量小或者均速的时候,选择push模式。在特殊场景下,例如电商大促,抢优惠券等场景可以选择pull模式
上面的快速入门就是发送同步消息,发送过后会有一个返回值,也就是mq服务器接收到消息后返回的一个确认,这种方式非常安全,但是性能上并没有这么高,而且在mq集群中,也是要等到所有的从机都复制了消息以后才会返回,所以针对重要的消息可以选择这种方式
异步消息通常用在对响应时间敏感的业务场景,即发送端不能容忍长时间地等待Broker的响应。发送完以后会有一个异步消息通知。
@Test
public void testAsyncProducer() throws Exception {
// 创建默认的生产者
DefaultMQProducer producer = new DefaultMQProducer("test-producer-group");
// 设置nameServer地址
producer.setNamesrvAddr(MQConstant.NAMESRV);
// 启动实例
producer.start();
Message msg = new Message("testTopic", ("异步消息").getBytes());
producer.send(msg, new SendCallback() {
@Override
public void onSuccess(SendResult sendResult) {
System.out.println("发送成功");
}
@Override
public void onException(Throwable e) {
System.out.println("发送失败");
}
});
System.out.println("看看谁先执行");
// 挂起jvm 因为回调是异步的不然测试不出来
System.in.read();
// 关闭实例
producer.shutdown();
}
这种方式主要用在不关心发送结果的场景,这种方式吞吐量很大,但是存在消息丢失的风险,例如日志信息的发送。
@Test
public void testOnewayProducer() throws Exception {
// 创建默认的生产者
DefaultMQProducer producer = new DefaultMQProducer("test-producer-group");
// 设置nameServer地址
producer.setNamesrvAddr(MQConstant.NAMESRV);
// 启动实例
producer.start();
Message msg = new Message("testTopic", ("单向消息").getBytes());
// 发送单向消息
producer.sendOneway(msg);
// 关闭实例
producer.shutdown();
}
消息放入mq后,过一段时间,才会被监听到,然后消费
比如下订单业务,提交了一个订单就可以发送一个延时消息,30min后去检查这个订单的状态,如果还是未付款就取消订单释放库存。
这里注意的是RocketMQ不支持任意时间的延时
只支持以下几个固定的延时等级,等级1就对应1s,以此类推,最高支持2h延迟
private String messageDelayLevel = “1s 5s 10s 30s 1m 2m 3m 4m 5m 6m 7m 8m 9m 10m 20m 30m 1h 2h”;
@Test
public void testDelayProducer() throws Exception {
// 创建默认的生产者
DefaultMQProducer producer = new DefaultMQProducer("test-group");
// 设置nameServer地址
producer.setNamesrvAddr("localhost:9876");
// 启动实例
producer.start();
Message msg = new Message("TopicTest", ("延迟消息").getBytes());
// 给这个消息设定一个延迟等级
// messageDelayLevel = "1s 5s 10s 30s 1m 2m 3m 4m 5m 6m 7m 8m 9m 10m 20m 30m 1h 2h
msg.setDelayTimeLevel(3);
// 发送单向消息
producer.send(msg);
// 打印时间
System.out.println(new Date());
// 关闭实例
producer.shutdown();
}
批量消息就是一次性发送一个消息集合出去。
@Test
public void testBatchProducer() throws Exception {
// 创建默认的生产者
DefaultMQProducer producer = new DefaultMQProducer("test-producer-group");
// 设置nameServer地址
producer.setNamesrvAddr(MQConstant.NAMESRV);
// 启动实例
producer.start();
List<Message> messages = Arrays.asList(
new Message("testTopic", "批量消息1".getBytes()),
new Message("testTopic", "批量消息2".getBytes()),
new Message("testTopic", "批量消息3".getBytes())
);
producer.send(messages);
System.out.println("批量执行任务");
// 挂起jvm 因为回调是异步的不然测试不出来
System.in.read();
// 关闭实例
producer.shutdown();
}
我们知道一个topic中可以有多个队列,那么如果我们的消息发送到多个队列中去,那么很明显我们的消息消费就是并行消费的,也就是没有了顺序性。
因此如果我们需要发送顺序消息,也就是希望MQ那边的消费者顺序的消费一些消息,我们就得按照如下方式发送顺序消息。
消息有序指的是可以按照消息的发送顺序来消费(FIFO)。RocketMQ可以严格的保证消息有序,可以分为:分区有序或者全局有序。
可能大家会有疑问,mq不就是FIFO吗?
rocketMq的broker的机制,导致了rocketMq会有这个问题
因为一个broker中对应了四个queue。
不同的queue(分区队列);而消费消息的时候从多个queue上拉取消息,这种情况发送和消费是不能保证顺序。但是如果控制发送的顺序消息只依次发送到同一个queue中,消费的时候只从这个queue上依次拉取,则就保证了顺序。当发送和消费参与的queue只有一个,则是全局有序;如果多个queue参与,则为分区有序,即相对每个queue,消息都是有序的。
下面用订单进行分区有序的示例。一个订单的顺序流程是:下订单、发短信通知、物流、签收。订单顺序号相同的消息会被先后发送到同一个队列中,消费时,同一个顺序获取到的肯定是同一个队列。
package zhang.blossom.seckillbyrocketmq;
import org.apache.rocketmq.client.consumer.DefaultMQPushConsumer;
import org.apache.rocketmq.client.consumer.listener.*;
import org.apache.rocketmq.client.exception.MQBrokerException;
import org.apache.rocketmq.client.exception.MQClientException;
import org.apache.rocketmq.client.producer.DefaultMQProducer;
import org.apache.rocketmq.client.producer.MessageQueueSelector;
import org.apache.rocketmq.common.message.Message;
import org.apache.rocketmq.common.message.MessageExt;
import org.apache.rocketmq.common.message.MessageQueue;
import org.apache.rocketmq.remoting.exception.RemotingException;
import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;
import zhang.blossom.seckillbyrocketmq.constant.MQConstant;
import zhang.blossom.seckillbyrocketmq.entity.MsgModel;
import java.io.IOException;
import java.util.Arrays;
import java.util.List;
/**
* @author: 张锦标
* @date: 2023/8/17 9:58
* OrderedRocketMQTest类
*/
@SpringBootTest
public class OrderedRocketMQTest {
private List<MsgModel> msgModels = Arrays.asList(
new MsgModel("qwer", 1L, "下单"),
new MsgModel("qwer", 1L, "短信"),
new MsgModel("qwer", 1L, "物流"),
new MsgModel("zxcv", 2L, "下单"),
new MsgModel("zxcv", 2L, "短信"),
new MsgModel("zxcv", 2L, "物流")
);
//发送顺序消息
@Test
public void orderedProducer() throws MQClientException, MQBrokerException, RemotingException, InterruptedException {
DefaultMQProducer producer = new DefaultMQProducer("test-producer-group");
producer.setNamesrvAddr(MQConstant.NAMESRV);
producer.start();
//发送顺序消息 发送时要确保有序 并且要发送到同一个队列下面去
msgModels.forEach(msgModel -> {
Message message = new Message("testTopic",msgModel.toString().getBytes());
try {
//发送 相同的订单号应该去相同的队列
producer.send(message, new MessageQueueSelector() {
//这里的send方法的第三个参数arg 就是这个队列选择器的第三个参数 会传递过来
@Override
public MessageQueue select(List<MessageQueue> list, Message message, Object arg) {
//这个方法的返回值就是要选择的队列
//这里可以用hash的方式就可以选择到同样的队列了
int hash = arg.toString().hashCode();
int index = hash % list.size();
return list.get(index);
}
}, msgModel.getOrderSn());
} catch (MQClientException e) {
throw new RuntimeException(e);
} catch (RemotingException e) {
throw new RuntimeException(e);
} catch (MQBrokerException e) {
throw new RuntimeException(e);
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
});
producer.shutdown();
System.out.println("发送完毕");
}
@Test
public void orderedConsumer() throws MQClientException, IOException {
DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("test-producer-group");
consumer.setNamesrvAddr(MQConstant.NAMESRV);
consumer.subscribe("testTopic", "*");
//MessageListenerConcurrently 并发模式 多线程的 失败后最多重试16次 然后放入死信队列
//MessageListenerOrderly 顺序模式 单线程的 失败后无限次重试 Integer.MAX_VALUE
consumer.registerMessageListener(new MessageListenerOrderly() {
//顺序模式只有一个线程来执行消费
@Override
public ConsumeOrderlyStatus consumeMessage(List<MessageExt> list,
ConsumeOrderlyContext consumeOrderlyContext) {
//这里的一个线程是一个队列一个线程
System.out.println(new String(list.get(0).getBody()));
return ConsumeOrderlyStatus.SUCCESS;
}
});
consumer.start();
System.in.read();
}
}
一般我们不使用RocketMQ的事务消息,所以有兴趣的可以看看其他的实现。
Rocketmq提供消息过滤功能,通过tag或者key进行区分。
我们往一个主题里面发送消息的时候,根据业务逻辑,可能需要区分,比如带有tagA标签的被A消费,带有tagB标签的被B消费,还有在事务监听的类里面,只要是事务消息都要走同一个监听,我们也需要通过过滤才区别对待。
@Test
public void tagProducer() throws Exception {
DefaultMQProducer producer = new DefaultMQProducer("test-producer-group");
producer.setNamesrvAddr(MQConstant.NAMESRV);
producer.start();
Message message1 = new Message("testTopic", "test1","test1的消息".getBytes() );
Message message2 = new Message("testTopic", "test2","test2的消息".getBytes() );
producer.send(message1);
producer.send(message2);
producer.shutdown();
System.out.println("消息发送成功");
}
@Test
public void test1Consumer() throws MQClientException, IOException {
DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("test-producer-group");
consumer.setNamesrvAddr(MQConstant.NAMESRV);
consumer.subscribe("testTopic", "test1");
consumer.registerMessageListener(new MessageListenerConcurrently() {
@Override
public ConsumeConcurrentlyStatus consumeMessage(List<MessageExt> list,
ConsumeConcurrentlyContext consumeConcurrentlyContext) {
System.out.println("消费test1的消息"+new String(list.get(0).getBody()));
return ConsumeConcurrentlyStatus.CONSUME_SUCCESS;
}
});
consumer.start();
System.in.read();
}
@Test
public void test2Consumer() throws MQClientException, IOException {
DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("test-producer-group");
consumer.setNamesrvAddr(MQConstant.NAMESRV);
consumer.subscribe("testTopic", "test1 || test2");
consumer.registerMessageListener(new MessageListenerConcurrently() {
@Override
public ConsumeConcurrentlyStatus consumeMessage(List<MessageExt> list,
ConsumeConcurrentlyContext consumeConcurrentlyContext) {
System.out.println("消费test1/test2的消息"+new String(list.get(0).getBody()));
return ConsumeConcurrentlyStatus.CONSUME_SUCCESS;
}
});
consumer.start();
System.in.read();
}
什么时候该用Topic,什么时候该用 Tag?
总结:不同的业务应该使用不同的Topic如果是相同的业务里面有不同表的表现形式,那么我们要使用tag进行区分
可以从以下几个方面进行判断:
1.消息类型是否一致:如普通消息、事务消息、定时(延时)消息、顺序消息,不同的消息类型使用不同的 Topic,无法通过 Tag 进行区分。
2.业务是否相关联:没有直接关联的消息,如淘宝交易消息,京东物流消息使用不同的 Topic 进行区分;而同样是天猫交易消息,电器类订单、女装类订单、化妆品类订单的消息可以用 Tag 进行区分。
3.消息优先级是否一致:如同样是物流消息,盒马必须小时内送达,天猫超市 24 小时内送达,淘宝物流则相对会慢一些,不同优先级的消息用不同的 Topic 进行区分。
4.消息量级是否相当:有些业务消息虽然量小但是实时性要求高,如果跟某些万亿量级的消息使用同一个 Topic,则有可能会因为过长的等待时间而“饿死”,此时需要将不同量级的消息进行拆分,使用不同的 Topic。
总的来说,针对消息分类,您可以选择创建多个Topic,或者在同一个 Topic 下创建多个 Tag。但通常情况下,不同的 Topic 之间的消息没有必然的联系,而 Tag 则用来区分同一个 Topic 下相互关联的消息,例如全集和子集的关系、流程先后的关系。
在rocketmq中的消息,默认会有一个messageId当做消息的唯一标识,我们也可以给消息携带一个key,用作唯一标识或者业务标识,包括在控制面板查询的时候也可以使用messageId或者key来进行查询。
@Test
public void testKeyProducer() throws Exception {
// 创建默认的生产者
DefaultMQProducer producer = new DefaultMQProducer("test-producer-group");
// 设置nameServer地址
producer.setNamesrvAddr(MQConstant.NAMESRV);
// 启动实例
producer.start();
Message msg = new Message("testTopic","test1","key", "我是一个带标记和key的消息".getBytes());
SendResult send = producer.send(msg);
System.out.println(send);
// 关闭实例
producer.shutdown();
}
@Test
public void testKeyConsumer() throws Exception {
// 创建默认消费者组
DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("test-producer-group");
// 设置nameServer地址
consumer.setNamesrvAddr(MQConstant.NAMESRV);
// 订阅一个主题来消费 表达式,默认是*,支持"tagA || tagB || tagC" 这样或者的写法 只要是符合任何一个标签都可以消费
consumer.subscribe("testTopic", "test1 || test2 || test3");
// 注册一个消费监听 MessageListenerConcurrently是并发消费
// 默认是20个线程一起消费,可以参看 consumer.setConsumeThreadMax()
consumer.registerMessageListener(new MessageListenerConcurrently() {
@Override
public ConsumeConcurrentlyStatus consumeMessage(List<MessageExt> msgs,
ConsumeConcurrentlyContext context) {
// 这里执行消费的代码 默认是多线程消费
System.out.println(Thread.currentThread().getName() + "----" + new String(msgs.get(0).getBody()));
System.out.println(msgs.get(0).getTags());
System.out.println(msgs.get(0).getKeys());
return ConsumeConcurrentlyStatus.CONSUME_SUCCESS;
}
});
consumer.start();
System.in.read();
}