(tsfile) branch develop updated: add getUsedMemorySize method for IPointReader

[jackietien]

This is an automated email from the ASF dual-hosted git repository.

jackietien pushed a commit to branch develop
in repository https://gitbox.apache.org/repos/asf/tsfile.git


The following commit(s) were added to refs/heads/develop by this push:
     new 4dd3742d add getUsedMemorySize method for IPointReader
4dd3742d is described below

commit 4dd3742ddcb0dda375c72adb08358114ad0801c9
Author: Beyyes <cgf1...@foxmail.com>
AuthorDate: Tue Apr 2 21:54:55 2024 +0800

    add getUsedMemorySize method for IPointReader
---
 .../main/java/org/apache/tsfile/read/common/BatchData.java |  6 ++++++
 .../java/org/apache/tsfile/read/common/block/TsBlock.java  | 10 ++++++++++
 .../java/org/apache/tsfile/read/reader/IPointReader.java   |  2 ++
 .../read/reader/page/LazyLoadAlignedPagePointReader.java   | 14 +++++++++++---
 4 files changed, 29 insertions(+), 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
index bc072b0e..bd89bcc7 100644
--- 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 class BatchData {
       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
index 1460e4f1..d6b7ca3d 100644
--- 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 class TsBlock {
           timeColumn.getLong(rowIndex), 
valueColumns[columnIndex].getTsPrimitiveType(rowIndex));
     }
 
+    @Override
+    public long getUsedMemorySize() {
+      return getRetainedSizeInBytes();
+    }
+
     @Override
     public void close() {
       // do nothing
@@ -440,6 +445,11 @@ public class TsBlock {
           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
index a01d615f..1ec4600e 100644
--- 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
index 07cde5f5..b28200ef 100644
--- 
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 @@ import java.util.List;
  */
 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 class LazyLoadAlignedPagePointReader implements 
IPointReader {
   }
 
   @Override
-  public void close() throws IOException {}
+  public long getUsedMemorySize() {
+    // not used
+    return 0;
+  }
+
+  @Override
+  public void close() throws IOException {
+    // do nothing
+  }
 }