add getUsedMemorySize method for IPointReader

apache · Apr 2, 2024 · 792759c · 792759c
1 parent 54eabc5
commit 792759c
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 3 deletions.
diff --git a/tsfile/src/main/java/org/apache/tsfile/read/common/BatchData.java b/tsfile/src/main/java/org/apache/tsfile/read/common/BatchData.java
@@ -855,6 +855,12 @@ public TimeValuePair currentTimeValuePair() {
       return new TimeValuePair(currentTime(), currentTsPrimitiveType());
     }
 
+    @Override
+    public long getUsedMemorySize() {
+      // not used
+      return 0;
+    }
+
     @Override
     public void close() {
       // do nothing

diff --git a/tsfile/src/main/java/org/apache/tsfile/read/common/block/TsBlock.java b/tsfile/src/main/java/org/apache/tsfile/read/common/block/TsBlock.java
@@ -305,6 +305,11 @@ public TimeValuePair currentTimeValuePair() {
           timeColumn.getLong(rowIndex), valueColumns[columnIndex].getTsPrimitiveType(rowIndex));
     }
 
+    @Override
+    public long getUsedMemorySize() {
+      return getRetainedSizeInBytes();
+    }
+
     @Override
     public void close() {
       // do nothing
@@ -440,6 +445,11 @@ public TimeValuePair currentTimeValuePair() {
           timeColumn.getLong(rowIndex), new TsPrimitiveType.TsVector(currentValue()));
     }
 
+    @Override
+    public long getUsedMemorySize() {
+      return getRetainedSizeInBytes();
+    }
+
     @Override
     public void close() {
       // do nothing

diff --git a/tsfile/src/main/java/org/apache/tsfile/read/reader/IPointReader.java b/tsfile/src/main/java/org/apache/tsfile/read/reader/IPointReader.java
@@ -31,5 +31,7 @@ public interface IPointReader {
 
   TimeValuePair currentTimeValuePair() throws IOException;
 
+  long getUsedMemorySize();
+
   void close() throws IOException;
 }
diff --git a/tsfile/src/main/java/org/apache/tsfile/read/reader/page/LazyLoadAlignedPagePointReader.java b/tsfile/src/main/java/org/apache/tsfile/read/reader/page/LazyLoadAlignedPagePointReader.java
@@ -32,8 +32,8 @@
  */
 public class LazyLoadAlignedPagePointReader implements IPointReader {
 
-  private TimePageReader timeReader;
-  private List<ValuePageReader> valueReaders;
+  private final TimePageReader timeReader;
+  private final List<ValuePageReader> valueReaders;
 
   private boolean hasNextRow = false;
 
@@ -93,5 +93,13 @@ public TimeValuePair currentTimeValuePair() throws IOException {
   }
 
   @Override
-  public void close() throws IOException {}
+  public long getUsedMemorySize() {
+    // not used
+    return 0;
+  }
+
+  @Override
+  public void close() throws IOException {
+    // do nothing
+  }
 }