RocketMQ主從讀寫分離機制的原理是什么

這篇文章將為大家詳細講解有關RocketMQ主從讀寫分離機制的原理是什么，文章內容質量較高，因此小編分享給大家做個參考，希望大家閱讀完這篇文章后對相關知識有一定的了解。

RocketMQ 的讀寫分離機制又跟上述描寫的不太一致，RocketMQ 有屬于自己的一套讀寫分離邏輯，它會判斷主服務器的消息堆積量來決定消費者是否向從服務器拉取消息消費。

決定消費者是否向從服務器拉取消息消費的值存在 GetMessageResult 類中：

org.apache.rocketmq.store.GetMessageResult：

private boolean suggestPullingFromSlave = false;

其默認值為 false，即默認消費者不會消費從服務器，以下邏輯可以改變該值：

org.apache.rocketmq.store.DefaultMessageStore#getMessage：

long diff = maxOffsetPy - maxPhyOffsetPulling;
long memory = (long) (StoreUtil.TOTAL_PHYSICAL_MEMORY_SIZE
    * (this.messageStoreConfig.getAccessMessageInMemoryMaxRatio() / 100.0));
getResult.setSuggestPullingFromSlave(diff > memory);

其中 maxOffsetPy 為當前最大物理偏移量，maxPhyOffsetPulling 為本次消息拉取最大物理偏移量，他們的差即可表示消息堆積量，TOTAL_PHYSICAL_MEMORY_SIZE 表示當前系統物理內存，accessMessageInMemoryMaxRatio 的默認值為 40，以上邏輯即可算出當前消息堆積量是否大于物理內存的 40 %，如果大于則將 suggestPullingFromSlave 設置為 true。

接下來該參數值會在消息拉取邏輯里面產生作用：

org.apache.rocketmq.broker.processor.PullMessageProcessor#processRequest:

if (getMessageResult.isSuggestPullingFromSlave()) {
  responseHeader.setSuggestWhichBrokerId(subscriptionGroupConfig.getWhichBrokerWhenConsumeSlowly());
} else {
  responseHeader.setSuggestWhichBrokerId(MixAll.MASTER_ID);
}

switch (this.brokerController.getMessageStoreConfig().getBrokerRole()) {
  case ASYNC_MASTER:
  case SYNC_MASTER:
    break;
  case SLAVE:
    if (!this.brokerController.getBrokerConfig().isSlaveReadEnable()) {
      response.setCode(ResponseCode.PULL_RETRY_IMMEDIATELY);
      responseHeader.setSuggestWhichBrokerId(MixAll.MASTER_ID);
    }
    break;
}

if (this.brokerController.getBrokerConfig().isSlaveReadEnable()) {
  // consume too slow ,redirect to another machine
  if (getMessageResult.isSuggestPullingFromSlave()) {
    responseHeader.setSuggestWhichBrokerId(subscriptionGroupConfig.getWhichBrokerWhenConsumeSlowly());
  }
  // consume ok
  else {
    responseHeader.setSuggestWhichBrokerId(subscriptionGroupConfig.getBrokerId());
  }
} else {
  responseHeader.setSuggestWhichBrokerId(MixAll.MASTER_ID);
}

如果發現主服務器的消息堆積超過了物理內存的 40%，則會設置 suggestWhichBrokerId 為從服務器 broker ID。

這里還會有個 slaveReadEnable 值來決定是否可以從從服務器拉取消息：

1. 如果 slaveReadEnable=true，并且堆積量已經超過物理內存 40%時，則建議從從服務器拉取消息，否則還是從主服務器拉取消息；

2. 如果 slaveReadEnable=false，則消息者只能從主服務器中拉取消息。

org.apache.rocketmq.client.impl.consumer.PullAPIWrapper#updatePullFromWhichNode：

public void updatePullFromWhichNode(final MessageQueue mq, final long brokerId) {
    AtomicLong suggest = this.pullFromWhichNodeTable.get(mq);
    if (null == suggest) {
        this.pullFromWhichNodeTable.put(mq, new AtomicLong(brokerId));
    } else {
        suggest.set(brokerId);
    }
}

當消費者收到拉取響應回來的數據后，會將下次建議拉取的 brokerID 緩存起來。下次拉取消息就會從 pullFromWhichNodeTable 中取出拉取 brokerId。

關于RocketMQ主從讀寫分離機制的原理是什么就分享到這里了，希望以上內容可以對大家有一定的幫助，可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。