RocketMQ通过ReputMessageService线程实现实时消息转发，更新ConsumeQueue和IndexFile。在异常退出时，通过检查abort文件判断恢复策略，正常退出执行正常恢复，异常退出执行异常恢复，确保消息不丢失但可能重复。
目录
一、消息转发
消息消费队列文件、索引文件都是基于CommitLog文件构建的，当消息Commit操作（将MappedFile.writeBuffer数据提交到该FileChannel文件通道内，即：提交到文件内存映射）时，ConsumeQueue、IndexFile需要及时更新，否则消息无法及时被消费，根据消息属性查找消息也会出现较大延迟。
RocketMQ通过开启一个线程ReputMessageServcie来实时转发CommitLog文件更新事件，通过各自的任务处理器根据转发的消息及时更新ConsumeQueue、IndexFile文件。如下所示是的UML图。org.apache.rocketmq.store.CommitLogDispatcher是转发的顶层接口。需要注意的是消息转发只是将消息写入ConsumeQueue、IndexFile文件内存映射中，不是写入磁盘。
1. ReputMessageService线程初始化
Broker启动时会启动ReputMessageService线程，并初始化关键参数reputFromOffset（开始转发消息的物理偏移量）。如果允许重复转发，reputFromOffset设置为CommitLog的提交指针；如果不允许重复转发，reputFromOffset设置为Commitlog的内存中最大偏移量。如下代码所示，org.apache.rocketmq.store.DefaultMessageStore#start。
/**
* broker启动时，消息存储线程
* BrokerController#startBasicService()
* @throws Exception
*/
@Override
public void start() throws Exception {
......
/*
启动ReputMessageService服务线程
重要参数reputFromOffset：从哪个物理偏移量转发消息给ConsumeQueue、IndexFile
允许重复转发：提交指针
不允许重复转发：最大偏移量
启动后，调用{@link ReputMessageService} 的run()
*/
// 允许重复转发：提交指针
if (this.getMessageStoreConfig().isDuplicationEnable()) {
this.reputMessageService.setReputFromOffset(this.commitLog.getConfirmOffset());
}
// 不允许重复转发：最大偏移量
else {
// It is [recover]'s responsibility to fully dispatch the commit log data before the max offset of commit log.
this.reputMessageService.setReputFromOffset(this.commitLog.getMaxOffset());
}
// DefaultMessageStore.ReputMessageService.run()
this.reputMessageService.start();
......
}下图所示是org.apache.rocketmq.store.DefaultMessageStore.ReputMessageService#run方法的调用链。其核心方法是org.apache.rocketmq.store.DefaultMessageStore.ReputMessageService#doReput，代码如下。注意转发消息DefaultMessageStore#doDispatch(DispatchRequest)方法并没有返回结果，而是通过重复写入次数控制，详细见下小节。
/**
* 消息提交到Commitlog时消息转发，构建ConsumeQueue、index文件服务 的核心方法
* step1：消息转发偏移量 > 最小偏移量时，则最小偏移量赋值给消息转发偏移量
* step2：消息转发偏移量 <= 最小偏移量时，获取转发偏移量开始的全部有效数据
* step3：循环转发每条消息，获取每条消息的转发请求对象{@link DispatchRequest}
* step4：消息解析成功后，转发消息{@link DefaultMessageStore#doDispatch(DispatchRequest)}
* 注意：转发消息时遍历LinkedList<CommitLogDispatcher> dispatcherList，集合中有消息队列、索引文件执行转发的实现类
* step5：更新消息转发偏移量
*/
private void doReput() {
// 消息转发偏移量 > 最小偏移量时，则最小偏移量赋值给消息转发偏移量
if (this.reputFromOffset < DefaultMessageStore.this.commitLog.getMinOffset()) {
LOGGER.warn("The reputFromOffset={} is smaller than minPyOffset={}, this usually indicate that the dispatch behind too much and the commitlog has expired.",
this.reputFromOffset, DefaultMessageStore.this.commitLog.getMinOffset());
this.reputFromOffset = DefaultMessageStore.this.commitLog.getMinOffset();
}
// 消息转发偏移量 <= 最小偏移量时，则循环转发每条消息
for (boolean doNext = true; this.isCommitLogAvailable() && doNext; ) {
// 获取从消息转发偏移量开始的全部有效数据
SelectMappedBufferResult result = DefaultMessageStore.this.commitLog.getData(reputFromOffset);
if (result != null) {
try {
this.reputFromOffset = result.getStartOffset();
// 循环转发每条消息
for (int readSize = 0; readSize < result.getSize() && reputFromOffset < DefaultMessageStore.this.getConfirmOffset() && doNext; ) {
// 获取每条消息
DispatchRequest dispatchRequest =
DefaultMessageStore.this.commitLog.checkMessageAndReturnSize(result.getByteBuffer(), false, false, false);
// 消息大小
int size = dispatchRequest.getBufferSize() == -1 ? dispatchRequest.getMsgSize() : dispatchRequest.getBufferSize();
// 读取的消息是否超出提交偏移量
if (reputFromOffset + size > DefaultMessageStore.this.getConfirmOffset()) {
doNext = false;
break;
}
// 解析消息成功
if (dispatchRequest.isSuccess()) {
if (size > 0) {
// 消息转发
DefaultMessageStore.this.doDispatch(dispatchRequest);
// 开启长轮询时，唤醒PullRequestHoldService线程，执行被挂起的拉取消息请求
if (DefaultMessageStore.this.brokerConfig.isLongPollingEnable()
&& DefaultMessageStore.this.messageArrivingListener != null) {
// 唤醒被挂起的拉取消息请求，再次拉取消息
DefaultMessageStore.this.messageArrivingListener.arriving(dispatchRequest.getTopic(),
dispatchRequest.getQueueId(), dispatchRequest.getConsumeQueueOffset() + 1,
dispatchRequest.getTagsCode(), dispatchRequest.getStoreTimestamp(),
dispatchRequest.getBitMap(), dispatchRequest.getPropertiesMap());
// 多个消息队列，唤醒被挂起的拉取消息请求，再次拉取消息
notifyMessageArrive4MultiQueue(dispatchRequest);
}
this.reputFromOffset += size;
readSize += size;
if (!DefaultMessageStore.this.getMessageStoreConfig().isDuplicationEnable() &&
DefaultMessageStore.this.getMessageStoreConfig().getBrokerRole() == BrokerRole.SLAVE) {
DefaultMessageStore.this.storeStatsService
.getSinglePutMessageTopicTimesTotal(dispatchRequest.getTopic()).add(1);
DefaultMessageStore.this.storeStatsService
.getSinglePutMessageTopicSizeTotal(dispatchRequest.getTopic())
.add(dispatchRequest.getMsgSize());
}
} else if (size == 0) {
this.reputFromOffset = DefaultMessageStore.this.commitLog.rollNextFile(this.reputFromOffset);
readSize = result.getSize();
}
}
// 解析消息失败
else if (!dispatchRequest.isSuccess()) {
if (size > 0) {
LOGGER.error("[BUG]read total count not equals msg total size. reputFromOffset={}", reputFromOffset);
this.reputFromOffset += size;
} else {
doNext = false;
// If user open the dledger pattern or the broker is master node,
// it will not ignore the exception and fix the reputFromOffset variable
if (DefaultMessageStore.this.getMessageStoreConfig().isEnableDLegerCommitLog() ||
DefaultMessageStore.this.brokerConfig.getBrokerId() == MixAll.MASTER_ID) {
LOGGER.error("[BUG]dispatch message to consume queue error, COMMITLOG OFFSET: {}",
this.reputFromOffset);
this.reputFromOffset += result.getSize() - readSize;
}
}
}
}
} finally {
result.release();
}
} else {
doNext = false;
}
}
}2. 消息转发更新ConsumeQueue
org.apache.rocketmq.store.DefaultMessageStore.CommitLogDispatcherBuildConsumeQueue是转发任务执行更新到消费队列实现类，CommitLogDispatcherBuildConsumeQueue#dispatch的执行转发消息方法，如下所示是调用链。
org.apache.rocketmq.store.ConsumeQueue#putMessagePositionInfoWrapper是执行转发核心方法。其中：写入消费队列最大重试次数maxRetries，默认30次，超出该值则打印异常日志；写入消费队列，只是写入到消费队列文件内存映射，并没有执行刷盘。
/**
* 转发消息到ConsumeQueue的核心方法
* step1：写入消费队列最大重试次数maxRetries，默认30次
* step2：判定消费队列是否允许写入
* step3：写入消费队列，只是写入到消费队列文件内存映射，不是写入磁盘
* step4：超出maxRetries，打印异常日志
*/
@Override
public void putMessagePositionInfoWrapper(DispatchRequest request) {
// 写入消费队列最大重试次数
final int maxRetries = 30;
// 是否能写
boolean canWrite = this.messageStore.getRunningFlags().isCQWriteable();
for (int i = 0; i < maxRetries && canWrite; i++) {
long tagsCode = request.getTagsCode();
if (isExtWriteEnable()) {
ConsumeQueueExt.CqExtUnit cqExtUnit = new ConsumeQueueExt.CqExtUnit();
cqExtUnit.setFilterBitMap(request.getBitMap());
cqExtUnit.setMsgStoreTime(request.getStoreTimestamp());
cqExtUnit.setTagsCode(request.getTagsCode());
long extAddr = this.consumeQueueExt.put(cqExtUnit);
if (isExtAddr(extAddr)) {
tagsCode = extAddr;
} else {
log.warn("Save consume queue extend fail, So just save tagsCode! {}, topic:{}, queueId:{}, offset:{}", cqExtUnit,
topic, queueId, request.getCommitLogOffset());
}
}
// 写入消费队列
boolean result = this.putMessagePositionInfo(request.getCommitLogOffset(),
request.getMsgSize(), tagsCode, request.getConsumeQueueOffset());
if (result) {
if (this.messageStore.getMessageStoreConfig().getBrokerRole() == BrokerRole.SLAVE ||
this.messageStore.getMessageStoreConfig().isEnableDLegerCommitLog()) {
this.messageStore.getStoreCheckpoint().setPhysicMsgTimestamp(request.getStoreTimestamp());
}
this.messageStore.getStoreCheckpoint().setLogicsMsgTimestamp(request.getStoreTimestamp());
if (checkMultiDispatchQueue(request)) {
multiDispatchLmqQueue(request, maxRetries);
}
return;
} else {
// XXX: warn and notify me
log.warn("[BUG]put commit log position info to " + topic + ":" + queueId + " " + request.getCommitLogOffset()
+ " failed, retry " + i + " times");
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
log.warn("", e);
}
}
}
// XXX: warn and notify me
log.error("[BUG]consume queue can not write, {} {}", this.topic, this.queueId);
this.messageStore.getRunningFlags().makeLogicsQueueError();
}org.apache.rocketmq.store.ConsumeQueue#putMessagePositionInfo是消息写入消费队列的文件内存映射的核心方法。代码如下所示。
/**
* 写入消费队列
* 注意：此次写入消息，只是追加到映射内存，固定刷盘方式：异步
* @param offset 消息偏移量
* @param size 消息大小
* @param tagsCode 消息Tag的哈希码
* @param cqOffset 消费队列的偏移量
* @return
*/
private boolean putMessagePositionInfo(final long offset, final int size, final long tagsCode,
final long cqOffset) {
if (offset + size <= this.maxPhysicOffset) {
log.warn("Maybe try to build consume queue repeatedly maxPhysicOffset={} phyOffset={}", maxPhysicOffset, offset);
return true;
}
// 组装消费队列条目：20字节 = 8字节的Commitlog offset + 4字节的消息大小 + 8字节的Tag的哈希码
this.byteBufferIndex.flip();
this.byteBufferIndex.limit(CQ_STORE_UNIT_SIZE);
this.byteBufferIndex.putLong(offset);
this.byteBufferIndex.putInt(size);
this.byteBufferIndex.putLong(tagsCode);
final long expectLogicOffset = cqOffset * CQ_STORE_UNIT_SIZE;
/*
* 更新消费队列参数
*/
MappedFile mappedFile = this.mappedFileQueue.getLastMappedFile(expectLogicOffset);
if (mappedFile != null) {
if (mappedFile.isFirstCreateInQueue() && cqOffset != 0 && mappedFile.getWrotePosition() == 0) {
this.minLogicOffset = expectLogicOffset;
this.mappedFileQueue.setFlushedWhere(expectLogicOffset);
this.mappedFileQueue.setCommittedWhere(expectLogicOffset);
this.fillPreBlank(mappedFile, expectLogicOffset);
log.info("fill pre blank space " + mappedFile.getFileName() + " " + expectLogicOffset + " "
+ mappedFile.getWrotePosition());
}
if (cqOffset != 0) {
long currentLogicOffset = mappedFile.getWrotePosition() + mappedFile.getFileFromOffset();
if (expectLogicOffset < currentLogicOffset) {
log.warn("Build consume queue repeatedly, expectLogicOffset: {} currentLogicOffset: {} Topic: {} QID: {} Diff: {}",
expectLogicOffset, currentLogicOffset, this.topic, this.queueId, expectLogicOffset - currentLogicOffset);
return true;
}
if (expectLogicOffset != currentLogicOffset) {
LOG_ERROR.warn(
"[BUG]logic queue order maybe wrong, expectLogicOffset: {} currentLogicOffset: {} Topic: {} QID: {} Diff: {}",
expectLogicOffset,
currentLogicOffset,
this.topic,
this.queueId,
expectLogicOffset - currentLogicOffset
);
}
}
this.maxPhysicOffset = offset + size;
// 追加消息到映射，并没有刷盘（固定刷盘方式：异步）
return mappedFile.appendMessage(this.byteBufferIndex.array());
}
return false;
}3. 消息转发更新IndexFile
org.apache.rocketmq.store.DefaultMessageStore.CommitLogDispatcherBuildIndex是转发任务执行更新到索引文件实现类，CommitLogDispatcherBuildIndex#dispatch的执行转发消息方法，如下所示是调用链。
org.apache.rocketmq.store.index.IndexService#buildIndex是构建索引的核心方法，如下代码所示。RocketMQ支持同一个消息建立多个索引，多个索引键空格分开，如："key1 key2"。
/**
* 构建消息索引文件
* 注意：索引追加到映射内存
* step1：获取或创建索引文件
* step2：获取索引文件最大偏移量
* step3：判断消息的偏移量 < 索引文件最大偏移量，说明索引文件已构建（数据重复），则返回
* step4：消息唯一键不为空，则首先添加索引（唯一键索引）
* step5：消息key不为空，MQ支持同一消息多个索引，用空格隔开
* step6：
* @param req 消息信息
*/
public void buildIndex(DispatchRequest req) {
// 获取或创建索引文件
IndexFile indexFile = retryGetAndCreateIndexFile();
if (indexFile != null) {
// 索引文件最大偏移量
long endPhyOffset = indexFile.getEndPhyOffset();
DispatchRequest msg = req;
String topic = msg.getTopic();
String keys = msg.getKeys();
// 消息的偏移量 < 索引文件最大偏移量，说明索引文件已构建（数据重复），则返回
if (msg.getCommitLogOffset() < endPhyOffset) {
return;
}
final int tranType = MessageSysFlag.getTransactionValue(msg.getSysFlag());
switch (tranType) {
case MessageSysFlag.TRANSACTION_NOT_TYPE:
case MessageSysFlag.TRANSACTION_PREPARED_TYPE:
case MessageSysFlag.TRANSACTION_COMMIT_TYPE:
break;
case MessageSysFlag.TRANSACTION_ROLLBACK_TYPE:
return;
}
// 消息唯一键不为空，则首先添加索引（唯一键索引）
if (req.getUniqKey() != null) {
// 写入索引文件缓冲
indexFile = putKey(indexFile, msg, buildKey(topic, req.getUniqKey())); // buildKey()：格式：主题#key
if (indexFile == null) {
LOGGER.error("putKey error commitlog {} uniqkey {}", req.getCommitLogOffset(), req.getUniqKey());
return;
}
}
// 消息key不为空，MQ支持同一消息多个索引，用空格隔开
if (keys != null && keys.length() > 0) {
String[] keyset = keys.split(MessageConst.KEY_SEPARATOR);
for (int i = 0; i < keyset.length; i++) {
String key = keyset[i];
if (key.length() > 0) {
indexFile = putKey(indexFile, msg, buildKey(topic, key));
if (indexFile == null) {
LOGGER.error("putKey error commitlog {} uniqkey {}", req.getCommitLogOffset(), req.getUniqKey());
return;
}
}
}
}
} else {
LOGGER.error("build index error, stop building index");
}
}org.apache.rocketmq.store.index.IndexFile#putKey是索引写入IndexFile文件内存映射中的核心方法，如下代码所示。
/**
* 写入IndexFile文件
* 追加到映射内存
* @param key 消息的key
* @param phyOffset 消息的物理偏移量
* @param storeTimestamp 消息的存储时间戳
* @return true写入成功；false写入失败或已满
*/
public boolean putKey(final String key, final long phyOffset, final long storeTimestamp) {
// 当前条目数量是否大于允许最大条目数量
if (this.indexHeader.getIndexCount() < this.indexNum) {
// key的hash值
int keyHash = indexKeyHashMethod(key);
// key的hash槽的下标
int slotPos = keyHash % this.hashSlotNum;
// hash槽的物理地址
int absSlotPos = IndexHeader.INDEX_HEADER_SIZE + slotPos * hashSlotSize;
try {
/*
* 相同hash，存储多个数据（槽中是否有数据）
*/
// 获取hash槽存储的数据（最新条目的地址）
int slotValue = this.mappedByteBuffer.getInt(absSlotPos);
if (slotValue <= invalidIndex || slotValue > this.indexHeader.getIndexCount()) {
slotValue = invalidIndex;
}
// 计算待存储条目 与 第一条消息时间戳 的差值，并转换成秒
long timeDiff = storeTimestamp - this.indexHeader.getBeginTimestamp();
timeDiff = timeDiff / 1000;
if (this.indexHeader.getBeginTimestamp() <= 0) {
timeDiff = 0;
} else if (timeDiff > Integer.MAX_VALUE) {
timeDiff = Integer.MAX_VALUE;
} else if (timeDiff < 0) {
timeDiff = 0;
}
// 存储位置absIndexPos = 头部40字节 + hash槽数量 * 单个槽的字节大小（4字节） + 当前存储条目数量 * 单个条目字节大小（20字节）
int absIndexPos =
IndexHeader.INDEX_HEADER_SIZE + this.hashSlotNum * hashSlotSize
+ this.indexHeader.getIndexCount() * indexSize;
// 组装条目 = hash码 + Commitlog偏移量 + 存储时间差值 + 当前hash槽的值（待存储条目的前一个条目的偏移量）
this.mappedByteBuffer.putInt(absIndexPos, keyHash);
this.mappedByteBuffer.putLong(absIndexPos + 4, phyOffset);
this.mappedByteBuffer.putInt(absIndexPos + 4 + 8, (int) timeDiff);
this.mappedByteBuffer.putInt(absIndexPos + 4 + 8 + 4, slotValue);
// 替换hash槽的值（待存储条目的偏移量absIndexPos）
this.mappedByteBuffer.putInt(absSlotPos, this.indexHeader.getIndexCount());
// 更新文件头信息
if (this.indexHeader.getIndexCount() <= 1) {
this.indexHeader.setBeginPhyOffset(phyOffset);
this.indexHeader.setBeginTimestamp(storeTimestamp);
}
if (invalidIndex == slotValue) {
this.indexHeader.incHashSlotCount();
}
this.indexHeader.incIndexCount();
this.indexHeader.setEndPhyOffset(phyOffset);
this.indexHeader.setEndTimestamp(storeTimestamp);
return true;
} catch (Exception e) {
log.error("putKey exception, Key: " + key + " KeyHashCode: " + key.hashCode(), e);
}
} else {
log.warn("Over index file capacity: index count = " + this.indexHeader.getIndexCount()
+ "; index max num = " + this.indexNum);
}
// 当前条目数量 >= 允许最大数量，则为false
return false;
}二、恢复机制
RocketMQ存储异步（默认）或同步提交到Commitlog文件内存映射，再异步（默认）或同步刷盘到磁盘。当消息写入Commitlog文件内存映射时，然后异步生成转发任务更新ConsumeQueue、Index 文件。如果消息成功存储到Commitlog文件中，转发任务未成功执行，此时Broker由于某个原因宕机，导致Commitlog、ConsumeQueue、IndexFile文件数据不一致。如果没有人工修复，则导致会有一部分消息即便在Commitlog文件中，但由于并没有转发到ConsumeQueue，这部分消息将永远不会被消费者消费。
RocketMQ是如何使Commitlog、ConsumeQueue达到最终一致性的呢？RocketMQ的存储文件的加载流程中，判断上一次退出是否正常，实现机制：Broker启动时判断${ROCKETMQ_HOME}/store/abort文件是否存在，正常退出时JVM钩子函数删除abort文件，若存在abort文件，说明Broker是异常退出。根据Broker是否正常退出，来采取不同的恢复机制。
文件恢复主要完成flushedPosition、committedWhere指针设置，消息消费队列最大偏移量加载到内存，并删除flushedPosition之后所有的文件。 如果Broker异常退出，在文件恢复过程中，RocketMQ会将最后一个有效文件中的所有消息重新转发到消息消费队列与索引文件，确保不丢失消息，但同时会带来消息重复的问题。纵观RocktMQ的整体设计思想，RocketMQ保证消息不丢失但不保证消息不会重复消费，故消息消费业务方需要实现消息消费的幕等设计。
1. Broker加载存储文件
org.apache.rocketmq.store.DefaultMessageStore#load是Broker启动时加载存储文件Commitlog、ConsumeQueue、IndexFile、Checkpoint文件的核心方法，其调用链、代码如下所示。
/**
* broker控制器启动初始化时，加载消息存储：如Commitlog文件、ConsumerQueue文件、IndexFile文件等
* step1：判断broker上次是否正常退出，是否存在${ROCKETMQ_HOME}/store/abort文件，存在：Broker退出异常
* step2：加载Commitlog文件、ConsumerQueue文件、Checkpoint文件、IndexFile索引文件
* step3：恢复文件，根据Broker是否退出异常，选择不同的恢复策略
* {@link DefaultMessageStore#recover(boolean)}
*/
@Override
public boolean load() {
boolean result = true;
try {
// 判断broker上次是否正常退出
boolean lastExitOK = !this.isTempFileExist();
LOGGER.info("last shutdown {}, root dir: {}", lastExitOK ? "normally" : "abnormally", messageStoreConfig.getStorePathRootDir());
// load Commit Log 加载消息存储文件
result = result && this.commitLog.load();
// load Consume Queue 加载消息消费队列文件
result = result && this.consumeQueueStore.load();
if (result) {
// 加载检查点Checkpoint文件
this.storeCheckpoint =
new StoreCheckpoint(StorePathConfigHelper.getStoreCheckpoint(this.messageStoreConfig.getStorePathRootDir()));
// 获取检查点中的刷盘时间点
this.masterFlushedOffset = this.storeCheckpoint.getMasterFlushedOffset();
// 加载IndexFile索引文件
this.indexService.load(lastExitOK);
// 恢复文件
this.recover(lastExitOK);
LOGGER.info("load over, and the max phy offset = {}", this.getMaxPhyOffset());
}
long maxOffset = this.getMaxPhyOffset();
this.setBrokerInitMaxOffset(maxOffset);
LOGGER.info("load over, and the max phy offset = {}", maxOffset);
} catch (Exception e) {
LOGGER.error("load exception", e);
result = false;
}
if (!result) {
this.allocateMappedFileService.shutdown();
}
return result;
}org.apache.rocketmq.store.DefaultMessageStore#recover是ConsumeQueue、IndexFile文件根据Broker是否正常提出采用不同恢复策略的方法入口，如下代码所示。恢复ConsumeQueue文件后，保存每个消费队列的当前存储逻辑偏移量。
/**
* 恢复文件
* @param lastExitOK broker上次是否正常退出，true正常退出；false异常退出
*/
private void recover(final boolean lastExitOK) {
long recoverCqStart = System.currentTimeMillis();
// 获取消息队列中的最大的消息物理偏移量
long maxPhyOffsetOfConsumeQueue = this.recoverConsumeQueue();
long recoverCqEnd = System.currentTimeMillis();
// 正常退出
if (lastExitOK) {
this.commitLog.recoverNormally(maxPhyOffsetOfConsumeQueue);
}
// 异常退出
else {
this.commitLog.recoverAbnormally(maxPhyOffsetOfConsumeQueue);
}
long recoverClogEnd = System.currentTimeMillis();
// 恢复ConsumeQueue文件后，保存每个消费队列的当前存储逻辑偏移量
this.recoverTopicQueueTable();
long recoverOffsetEnd = System.currentTimeMillis();
LOGGER.info("Recover end total:{} recoverCq:{} recoverClog:{} recoverOffset:{}",
recoverOffsetEnd - recoverCqStart, recoverCqEnd - recoverCqStart, recoverClogEnd - recoverCqEnd, recoverOffsetEnd - recoverClogEnd);
}2. Broker正常退出的文件恢复
org.apache.rocketmq.store.CommitLog#recoverNormally是Broker正常退出（abort文件不存在）时的恢复核心方法，其调用链、代码如下所示。
恢复是从ConsumeQueue文件集合的倒数第三个文件开始恢复，若没有3个则从第一个开始恢复。需要注意的是没有恢复IndexFile文件，原因是有异常的IndexFile在加载过程中被销毁，当恢复时再次转发消息来构建IndexFile，详细见org.apache.rocketmq.store.index.IndexService#load是IndexFile文件加载的核心处理方法，这里不作介绍。
/**
* Broker正常停止，文件恢复
* step1：倒数第三个文件开始恢复，若没有则从第一个开始恢复
* step2：遍历Commitlog文件，逐个取出每一条消息进行恢复（再次转发）
* step3：更新MappedFileQueue的刷盘、提交指针 并移除processOffset之后的Commitlog文件
* When the normal exit, data recovery, all memory data have been flush
*/
public void recoverNormally(long maxPhyOffsetOfConsumeQueue) {
// 文件恢复时，查找消息是否验证CRC，可配置
boolean checkCRCOnRecover = this.defaultMessageStore.getMessageStoreConfig().isCheckCRCOnRecover();
boolean checkDupInfo = this.defaultMessageStore.getMessageStoreConfig().isDuplicationEnable();
final List<MappedFile> mappedFiles = this.mappedFileQueue.getMappedFiles();
if (!mappedFiles.isEmpty()) {
// 倒数第三个文件开始恢复，若没有则从第一个开始恢复
// Began to recover from the last third file
int index = mappedFiles.size() - 3;
if (index < 0) {
index = 0;
}
MappedFile mappedFile = mappedFiles.get(index);
ByteBuffer byteBuffer = mappedFile.sliceByteBuffer();
// 已恢复的消息物理偏移量 = mappedFile.getFileFromOffset() + mappedFileOffset
long processOffset = mappedFile.getFileFromOffset();
// 当前已校验通过的offset
long mappedFileOffset = 0; // 从0开始，说明该Commitlog文件的第一个消息开始
long lastValidMsgPhyOffset = this.getConfirmOffset();
// normal recover doesn't require dispatching
boolean doDispatch = false; // 是否正常恢复
// 遍历Commitlog文件，逐个取出每一条消息进行恢复
while (true) {
// 检查消息并返回消息大小
DispatchRequest dispatchRequest = this.checkMessageAndReturnSize(byteBuffer, checkCRCOnRecover, checkDupInfo);
int size = dispatchRequest.getMsgSize();
// Normal data
if (dispatchRequest.isSuccess() && size > 0) {
lastValidMsgPhyOffset = processOffset + mappedFileOffset;
mappedFileOffset += size;
// 消息转发到ConsumeQueue文件、IndexFile文件
this.getMessageStore().onCommitLogDispatch(dispatchRequest, doDispatch, mappedFile, true, false);
}
// 消息验证通过，且消息大小为0，说明已到Commitlog文件末尾，读取下一个Commitlog文件
// Come the end of the file, switch to the next file Since the
// return 0 representatives met last hole,
// this can not be included in truncate offset
else if (dispatchRequest.isSuccess() && size == 0) {
this.getMessageStore().onCommitLogDispatch(dispatchRequest, doDispatch, mappedFile, true, true);
// 读取下一文件
index++;
// 下一文件没有
if (index >= mappedFiles.size()) {
// Current branch can not happen
log.info("recover last 3 physics file over, last mapped file " + mappedFile.getFileName());
break;
}
// 若有下个Commitlog，重新设置变量
else {
mappedFile = mappedFiles.get(index);
byteBuffer = mappedFile.sliceByteBuffer();
processOffset = mappedFile.getFileFromOffset();
mappedFileOffset = 0;
log.info("recover next physics file, " + mappedFile.getFileName());
}
}
// Intermediate file read error
else if (!dispatchRequest.isSuccess()) {
if (size > 0) {
log.warn("found a half message at {}, it will be truncated.", processOffset + mappedFileOffset);
}
log.info("recover physics file end, " + mappedFile.getFileName());
break;
}
}
/*
更新MappedFileQueue的刷盘、提交指针
*/
processOffset += mappedFileOffset;
// Set a candidate confirm offset.
// In most cases, this value will be overwritten by confirmLog.init.
// It works if some confirmed messages are lost.
this.setConfirmOffset(lastValidMsgPhyOffset);
this.mappedFileQueue.setFlushedWhere(processOffset);
this.mappedFileQueue.setCommittedWhere(processOffset);
// 移除processOffset之后的所有文件
this.mappedFileQueue.truncateDirtyFiles(processOffset);
// Clear ConsumeQueue redundant data
if (maxPhyOffsetOfConsumeQueue >= processOffset) {
log.warn("maxPhyOffsetOfConsumeQueue({}) >= processOffset({}), truncate dirty logic files", maxPhyOffsetOfConsumeQueue, processOffset);
this.defaultMessageStore.truncateDirtyLogicFiles(processOffset);
}
} else {
// Commitlog case files are deleted
log.warn("The commitlog files are deleted, and delete the consume queue files");
this.mappedFileQueue.setFlushedWhere(0);
this.mappedFileQueue.setCommittedWhere(0);
this.defaultMessageStore.destroyLogics();
}
}3. Broker异常退出的文件恢复
org.apache.rocketmq.store.CommitLog#recoverAbnormally是Broker异常退出（abort文件存在）时的恢复核心方法，代码如下所示。
与Broker正常退出逻辑差不多，区别：恢复是从ConsumeQueue文件集合的最后一个文件往前遍历，找到文件第一条消息存储正常的文件。而判断Commitlog是否正常存储，判定条件：魔数是否正常（第一条消息存储时间戳 = 0时，说明该文件没有任何消息）。
/**
* Broker异常停止，文件恢复（大致与正常恢复一样）
*/
@Deprecated
public void recoverAbnormally(long maxPhyOffsetOfConsumeQueue) {
// recover by the minimum time stamp
boolean checkCRCOnRecover = this.defaultMessageStore.getMessageStoreConfig().isCheckCRCOnRecover();
boolean checkDupInfo = this.defaultMessageStore.getMessageStoreConfig().isDuplicationEnable();
final List<MappedFile> mappedFiles = this.mappedFileQueue.getMappedFiles();
if (!mappedFiles.isEmpty()) {
// 最后一个文件往前走，找到第一个消息存储正常的文件
// Looking beginning to recover from which file
int index = mappedFiles.size() - 1;
MappedFile mappedFile = null;
for (; index >= 0; index--) {
mappedFile = mappedFiles.get(index);
// 判断Commitlog是否正常存储（判定条件：魔数是否正常）；注意第一条消息存储时间戳 = 0时，说明该文件没有任何消息
if (this.isMappedFileMatchedRecover(mappedFile)) {
log.info("recover from this mapped file " + mappedFile.getFileName());
break;
}
}
if (index < 0) { // 没有，则获取第一个
index = 0;
mappedFile = mappedFiles.get(index);
}
ByteBuffer byteBuffer = mappedFile.sliceByteBuffer();
long processOffset = mappedFile.getFileFromOffset();
long mappedFileOffset = 0;
long lastValidMsgPhyOffset = this.getConfirmOffset();
// abnormal recover require dispatching
boolean doDispatch = true;
while (true) {
DispatchRequest dispatchRequest = this.checkMessageAndReturnSize(byteBuffer, checkCRCOnRecover, checkDupInfo);
int size = dispatchRequest.getMsgSize();
if (dispatchRequest.isSuccess()) {
// Normal data
if (size > 0) {
lastValidMsgPhyOffset = processOffset + mappedFileOffset;
mappedFileOffset += size;
if (this.defaultMessageStore.getMessageStoreConfig().isDuplicationEnable()) {
if (dispatchRequest.getCommitLogOffset() < this.defaultMessageStore.getConfirmOffset()) {
this.getMessageStore().onCommitLogDispatch(dispatchRequest, doDispatch, mappedFile, true, false);
}
} else {
// 消息转发到ConsumeQueue文件、IndexFile文件
this.getMessageStore().onCommitLogDispatch(dispatchRequest, doDispatch, mappedFile, true, false);
}
}
// Come the end of the file, switch to the next file
// Since the return 0 representatives met last hole, this can
// not be included in truncate offset
else if (size == 0) {
this.getMessageStore().onCommitLogDispatch(dispatchRequest, doDispatch, mappedFile, true, true);
index++;
if (index >= mappedFiles.size()) {
// The current branch under normal circumstances should
// not happen
log.info("recover physics file over, last mapped file " + mappedFile.getFileName());
break;
} else {
mappedFile = mappedFiles.get(index);
byteBuffer = mappedFile.sliceByteBuffer();
processOffset = mappedFile.getFileFromOffset();
mappedFileOffset = 0;
log.info("recover next physics file, " + mappedFile.getFileName());
}
}
} else {
if (size > 0) {
log.warn("found a half message at {}, it will be truncated.", processOffset + mappedFileOffset);
}
log.info("recover physics file end, " + mappedFile.getFileName() + " pos=" + byteBuffer.position());
break;
}
}
processOffset += mappedFileOffset;
// Set a candidate confirm offset.
// In most cases, this value will be overwritten by confirmLog.init.
// It works if some confirmed messages are lost.
this.setConfirmOffset(lastValidMsgPhyOffset);
this.mappedFileQueue.setFlushedWhere(processOffset);
this.mappedFileQueue.setCommittedWhere(processOffset);
this.mappedFileQueue.truncateDirtyFiles(processOffset);
// Clear ConsumeQueue redundant data
if (maxPhyOffsetOfConsumeQueue >= processOffset) {
log.warn("maxPhyOffsetOfConsumeQueue({}) >= processOffset({}), truncate dirty logic files", maxPhyOffsetOfConsumeQueue, processOffset);
this.defaultMessageStore.truncateDirtyLogicFiles(processOffset);
}
}
// Commitlog case files are deleted
else {
log.warn("The commitlog files are deleted, and delete the consume queue files");
this.mappedFileQueue.setFlushedWhere(0);
this.mappedFileQueue.setCommittedWhere(0);
this.defaultMessageStore.destroyLogics();
}
}三、参考资料
RocketMQ源码分析(十五)之文件恢复_jannals的博客-CSDN博客_rocketmq 文件恢复
RocketMQ5.0.0消息发送_爱我所爱0505的博客-CSDN博客
扫一扫
4371
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
