Add column builder compatibility (#367)

apache · Jan 9, 2025 · 6ced6e1 · 6ced6e1
1 parent afd6ec4
commit 6ced6e1
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 0 deletions.
diff --git a/.../tsfile/src/main/java/org/apache/tsfile/read/common/block/column/DoubleColumnBuilder.java b/.../tsfile/src/main/java/org/apache/tsfile/read/common/block/column/DoubleColumnBuilder.java
@@ -66,6 +66,21 @@ public int getPositionCount() {
     return positionCount;
   }
 
+  @Override
+  public ColumnBuilder writeLong(long value) {
+    return writeDouble(value);
+  }
+
+  @Override
+  public ColumnBuilder writeFloat(float value) {
+    return writeDouble(value);
+  }
+
+  @Override
+  public ColumnBuilder writeInt(int value) {
+    return writeDouble(value);
+  }
+
   @Override
   public ColumnBuilder writeDouble(double value) {
     if (values.length <= positionCount) {

diff --git a/java/tsfile/src/main/java/org/apache/tsfile/read/common/block/column/FloatColumnBuilder.java b/java/tsfile/src/main/java/org/apache/tsfile/read/common/block/column/FloatColumnBuilder.java
@@ -66,6 +66,11 @@ public int getPositionCount() {
     return positionCount;
   }
 
+  @Override
+  public ColumnBuilder writeInt(int value) {
+    return writeFloat(value);
+  }
+
   @Override
   public ColumnBuilder writeFloat(float value) {
     if (values.length <= positionCount) {

diff --git a/java/tsfile/src/main/java/org/apache/tsfile/read/common/block/column/LongColumnBuilder.java b/java/tsfile/src/main/java/org/apache/tsfile/read/common/block/column/LongColumnBuilder.java
@@ -66,6 +66,11 @@ public int getPositionCount() {
     return positionCount;
   }
 
+  @Override
+  public ColumnBuilder writeInt(int value) {
+    return writeLong(value);
+  }
+
   @Override
   public ColumnBuilder writeLong(long value) {
     if (values.length <= positionCount) {

diff --git a/java/tsfile/src/main/java/org/apache/tsfile/read/common/block/column/TimeColumnBuilder.java b/java/tsfile/src/main/java/org/apache/tsfile/read/common/block/column/TimeColumnBuilder.java
@@ -54,6 +54,11 @@ public TimeColumnBuilder(ColumnBuilderStatus columnBuilderStatus, int expectedEn
     updateDataSize();
   }
 
+  @Override
+  public ColumnBuilder writeInt(int value) {
+    return writeLong(value);
+  }
+
   @Override
   public ColumnBuilder writeLong(long value) {
     if (values.length <= positionCount) {