[NSE-81] add missing setNulls methods in ArrowWritableColumnVector (#79)

* add missing setNulls methods in ArrowWritableColumnVector * miss a @OverRide
oap-project · Feb 4, 2021 · 7d8f210 · 7d8f210
1 parent faeebb2
commit 7d8f210
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/core/src/main/java/com/intel/oap/vectorized/ArrowWritableColumnVector.java b/core/src/main/java/com/intel/oap/vectorized/ArrowWritableColumnVector.java
@@ -1694,6 +1694,13 @@ final void setNull(int rowId) {
       writer.setNull(rowId);
     }
 
+    @Override
+    final void setNulls(int rowId, int count) {
+      for (int i = 0; i < count; i++) {
+        writer.setNull(rowId + i);
+      }
+    }
+
     @Override
     final void setBytes(int rowId, int count, byte[] src, int srcIndex) {
       writer.setSafe(rowId, src, srcIndex, count);
@@ -1718,6 +1725,13 @@ private static class BinaryWriter extends ArrowVectorWriter {
     final void setNull(int rowId) {
       writer.setNull(rowId);
     }
+
+    @Override
+    final void setNulls(int rowId, int count) {
+      for (int i = 0; i < count; i++) {
+        writer.setNull(rowId + i);
+      }
+    }
   }
 
   private static class DateWriter extends ArrowVectorWriter {
@@ -1744,6 +1758,13 @@ void setInts(int rowId, int count, int value) {
     final void setNull(int rowId) {
       writer.setNull(rowId);
     }
+
+    @Override
+    final void setNulls(int rowId, int count) {
+      for (int i = 0; i < count; i++) {
+        writer.setNull(rowId + i);
+      }
+    }
   }
 
   private static class TimestampWriter extends ArrowVectorWriter {
@@ -1770,6 +1791,13 @@ void setLong(int rowId, long value) {
     final void setNull(int rowId) {
       writer.setNull(rowId);
     }
+
+    @Override
+    final void setNulls(int rowId, int count) {
+      for (int i = 0; i < count; i++) {
+        writer.setNull(rowId + i);
+      }
+    }
   }
 
   private static class ArrayWriter extends ArrowVectorWriter {