feat: record pooled record memory usage (#846)

Signed-off-by: Robin Han <[email protected]>
AutoMQ · Feb 27, 2024 · 6b7f329 · 6b7f329
1 parent d971f8e
commit 6b7f329
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/core/src/main/scala/kafka/log/streamaspect/ElasticLogFileRecords.java b/core/src/main/scala/kafka/log/streamaspect/ElasticLogFileRecords.java
@@ -56,6 +56,11 @@
 
 public class ElasticLogFileRecords {
     private static final Logger LOGGER = LoggerFactory.getLogger(ElasticLogFileRecords.class);
+    private static final int POOLED_MEMORY_RECORDS = 30;
+    static {
+        DirectByteBufAlloc.registerAllocType(POOLED_MEMORY_RECORDS, "pooled_memory_records");
+    }
+
     protected final AtomicInteger size;
     // only used for recover
     protected final Iterable<RecordBatch> batches;
@@ -320,7 +325,7 @@ private PooledMemoryRecords(long logBaseOffset, List<FetchResult> fetchResults,
             }
             // TODO: create a new ByteBufMemoryRecords data struct to avoid copy
             if (pooled) {
-                this.pack = DirectByteBufAlloc.byteBuffer(size);
+                this.pack = DirectByteBufAlloc.byteBuffer(size, POOLED_MEMORY_RECORDS);
             } else {
                 this.pack = Unpooled.buffer(size);
             }