mq消费源码
依赖

com.aliyun.openservices
ons-client

阿里云rocketmq消息队列

rocketmq消费源码_rocketmq

参考 SpringBoot整合阿里云rocketmq消息队列,创建生产者和消费者实例

进来createConsumer方法

rocketmq消费源码_rocketmq_02

return new ConsumerImpl(ONSUtil.extractProperties(properties));

rocketmq消费源码_客户端_03

可以看到阿里云 opensevices提供许多mq的功能,包括创建生产者、创建消费者、批量创建消费者、批量创建生产者、
创建顺序Consumer(全家顺序和局部顺序,和上面的并发消费区分开来),顺序消费的使用场景是某些业务必须同步完成的时候可以使用,就是这一个业务线的数据都发到同一个queue中并且使用同一个消费者来消费,这样相对于每个queue就是有序的
最后一个是创建事务生产者,所以做rocketmq包括阿里云那套对常见的方法封装的比较好,这也是这个mq的优势之一。

接着回来ConsumerImpl 消费的源码

rocketmq消费源码_客户端_04

这个方法中先调用了父类ONSConsumerAbstract中的构造方法,在ONSConsumerAbstract的构造方法中又调用了ONSClientAbstract的构造方法。

rocketmq消费源码_客户端_05

ONSClientAbstract的构造方法检查参数,包括连接地址、访问密钥等

rocketmq消费源码_.net_06

private final ScheduledExecutorService scheduledExecutorService = new ScheduledThreadPoolExecutor(1, new ThreadFactory() {
public Thread newThread(Runnable r) {
return new Thread(r, “ONSClient-UpdateNameServerThread”);
}
});


然后有个线程池,每90秒检测一次nameServer地址是否变更
这块逻辑其实consumer与producer公用的。ProducerImpl也会去调用父类方法super(properties);

rocketmq消费源码_构造方法_07

DefaultMQPushConsumer,核心代码,ONSConsumerAbstract的构造方法中创建消费者,使用的是push的方式。

回顾一下push pull区别

rocketmq消费源码_客户端_08

push方式,consumer把轮询过程封装了,并注册messagelistener监听器,取到消息后,唤醒messagelister的consumerMessage消费,感觉队列被push过来进行订阅。
pull方式,消息自己去取,首先通过打算消费topic拿到messagequeue集合,遍历messagequeue集合,然后针对每个messgequeue批量取消息,一次取完后,记录该队列下一次要取的offset,直到取完了,再换另一个messagequeue.

到回DefaultMQPushConsumer之后,配置项设置consumerGroup,实例名称,nameServer地址,消费者线程的最小和最大线程数,包括一些在PropertyKeyConst类中的配置(客户端缓存消息数量,客户端缓存最大内容)。

rocketmq消费源码_.net_09

rocketmq消费源码_rocketmq_10


最后setPostSubscriptionWhenPull设置为false,

consumer.setMessageModel(MessageModel.CLUSTERING)集群模式(消费模式分集群、广播)总结:消费方式mq ;设置Properties[图片]获取yml里面的访问密钥、密钥、mq服务器地址、设置发送超时时间,单位毫秒 1000=1s;

rocketmq消费源码_.net_11


rocketmq消费源码_rocketmq_12


rocketmq消费源码_构造方法_13


rocketmq消费源码_客户端_14

然后获取消费组进行遍历,以及groupid,然后就可以创建消费者,创建消费者上面说了,
创建方式构造方法,[图片]把这些参数初始化,然后做一些校验和在赋值。
拉取消息的前置和后置处理类的创建

// 为Consumer增加消息轨迹回发模块
String msgTraceSwitch = properties.getProperty(PropertyKeyConst.MsgTraceSwitch);
if (!UtilAll.isBlank(msgTraceSwitch) && (!Boolean.parseBoolean(msgTraceSwitch))) {
log.info(“MQ Client Disable the Trace Hook!”);
} else {
try {
Properties tempProperties = new Properties();
tempProperties.put(OnsTraceConstants.AccessKey, sessionCredentials.getAccessKey());
tempProperties.put(OnsTraceConstants.SecretKey, sessionCredentials.getSecretKey());
tempProperties.put(OnsTraceConstants.MaxMsgSize, “128000”);
tempProperties.put(OnsTraceConstants.AsyncBufferSize, “2048”);
tempProperties.put(OnsTraceConstants.MaxBatchNum, “100”);
tempProperties.put(OnsTraceConstants.NAMESRV_ADDR, this.getNameServerAddr());
tempProperties.put(OnsTraceConstants.InstanceName, “PID_CLIENT_INNER_TRACE_PRODUCER”);
tempProperties.put(OnsTraceConstants.TraceDispatcherType, OnsTraceDispatcherType.CONSUMER.name());
AsyncArrayDispatcher dispatcher = new AsyncArrayDispatcher(tempProperties, sessionCredentials);
dispatcher.setHostConsumer(defaultMQPushConsumer.getDefaultMQPushConsumerImpl());
traceDispatcher = dispatcher;
this.defaultMQPushConsumer.getDefaultMQPushConsumerImpl().registerConsumeMessageHook(
new OnsConsumeMessageHookImpl(traceDispatcher));
} catch (Throwable e) {
log.error(“system mqtrace hook init failed ,maybe can’t send msg trace data”, e);
}
}
然后回来就是设置pull flase,集群模式。
然后把创建的消费者去订阅消息
consumerBean.subscribe(consumer.getTopic(), consumer.getTag(),
SpringUtil.getBean(consumer.getBeanName()));

rocketmq消费源码_rocketmq_15


rocketmq消费源码_构造方法_16


rocketmq消费源码_rocketmq_17

subscribeTable放入订阅map,[图片]topic和tag过来消费者订阅消息了,[图片],发送的时候用了new ReentrantLock()锁
public void sendHeartbeatToAllBrokerWithLock() {
if (this.lockHeartbeat.tryLock()) {
try {
this.sendHeartbeatToAllBroker();
this.uploadFilterClassSource();
} catch (Exception var5) {
this.log.error(“sendHeartbeatToAllBroker exception”, var5);
} finally {
this.lockHeartbeat.unlock();
}
} else {
this.log.warn(“lock heartBeat, but failed.”);
}}

push:
理下流程:

首先 new DefaultMQPushConsumer 对象,并指定一个消费组名。
然后设置相关参数,例如 nameSrvAdd、消费失败重试次数、线程数等
通过调用 setConsumeFromWhere 方法指定初次启动时从什么地方消费,默认是最新的消息开始消费。
通过调用 setAllocateMessageQueueStrategy 指定队列负载机制,默认平均分配。
通过调用 registerMessageListener 设置消息监听器,即消息处理逻辑,最终返回 CONSUME_SUCCESS(成功消费)或 RECONSUME_LATER(需要重试)。

pull
首先根据 MQConsumer 的 fetchSubscribeMessageQueues 的方法获取 Topic 的所有队列信息
然后遍历所有队列,依次通过 MQConsuemr 的 PULL 方法从 Broker 端拉取消息。
对拉取的消息进行消费处理
通过调用 MQConsumer 的 updateConsumeOffset 方法更新位点,但需要注意的是这个方法并不是实时向 Broker 提交,而是客户端会启用以线程,默认每隔 5s 向 Broker 集中上报一次。

​ 1.集群消费方式
一个ConsumerGroup中的Consumer实例平均分摊消费生产者发送的消息。例如某个Topic有九条消息,其中一个Consumer Group有三个实例(可能是3个进程,或者3台机器),那么每个实例只消费其中的3条消息,Consumer不指定消费方式的话默认是集群消费的,适用于大部分消息的业务
2.广播消费方式
一条消息被多个Consumer消费,几十这些Consumer属于同一个ConsumerGroup,消息也会被ConsumerGroup中的每个Consumer消费一次,广播消费中的ConsumerGroup概念可以认为在消息划分层面没有意义,适用于一些分发消息的场景,比如我订单下单成功了,需要通知财务系统,客服系统等等这种分发的场景,可以通过修改Consumer中的MessageModel来设置消费方式为广播消费

//广播
BROADCASTING(“BROADCASTING”),
//集群
CLUSTERING(“CLUSTERING”);
源码
String messageModel = properties.getProperty(“MessageModel”, “CLUSTERING”);
this.defaultMQPushConsumer.setMessageModel(MessageModel.valueOf(messageModel));