fix get message from tiered store from cache too large

apache · Dec 25, 2024 · eea8264 · eea8264
1 parent 1c35adb
commit eea8264
Showing 1 changed file with 3 additions and 0 deletions.
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
@@ -148,6 +148,9 @@ protected GetMessageResultExt getMessageFromCache(
             if (result.getMessageCount() == maxCount) {
                 break;
             }
+            if (result.getBufferTotalSize() >= storeConfig.getReadAheadMessageSizeThreshold()) {
+                break;
+            }
         }
         result.setStatus(result.getMessageCount() > 0 ?
             GetMessageStatus.FOUND : GetMessageStatus.NO_MATCHED_MESSAGE);