diff --git a/tieredstore/src/main/java/org/apache/rocketmq/tieredstore/core/MessageStoreFetcherImpl.java b/tieredstore/src/main/java/org/apache/rocketmq/tieredstore/core/MessageStoreFetcherImpl.java index c5e5f31ee22..e94185626a7 100644 --- a/tieredstore/src/main/java/org/apache/rocketmq/tieredstore/core/MessageStoreFetcherImpl.java +++ b/tieredstore/src/main/java/org/apache/rocketmq/tieredstore/core/MessageStoreFetcherImpl.java @@ -56,6 +56,7 @@ public class MessageStoreFetcherImpl implements MessageStoreFetcher { private final String brokerName; private final MetadataStore metadataStore; private final MessageStoreConfig storeConfig; + private final org.apache.rocketmq.store.config.MessageStoreConfig messageStoreConfig; private final TieredMessageStore messageStore; private final IndexService indexService; private final FlatFileStore flatFileStore; @@ -71,6 +72,7 @@ public MessageStoreFetcherImpl(TieredMessageStore messageStore, MessageStoreConf FlatFileStore flatFileStore, IndexService indexService) { this.storeConfig = storeConfig; + this.messageStoreConfig = messageStore.getMessageStoreConfig(); this.brokerName = storeConfig.getBrokerName(); this.flatFileStore = flatFileStore; this.messageStore = messageStore; @@ -148,7 +150,7 @@ protected GetMessageResultExt getMessageFromCache( if (result.getMessageCount() == maxCount) { break; } - if (result.getBufferTotalSize() >= storeConfig.getReadAheadMessageSizeThreshold()) { + if (result.getBufferTotalSize() >= messageStoreConfig.getMaxTransferBytesOnMessageInMemory()) { break; } }