diff --git a/omnioperator/omniop-hive-extension/src/main/java/com/huawei/boostkit/hive/reader/OmniOrcRecordReader.java b/omnioperator/omniop-hive-extension/src/main/java/com/huawei/boostkit/hive/reader/OmniOrcRecordReader.java
index 1456cd56564aa59161a2941a6d241b6a5195fd7d..e61c142d046c4bee25bf82bafc04b0a86725a2fb 100644
--- a/omnioperator/omniop-hive-extension/src/main/java/com/huawei/boostkit/hive/reader/OmniOrcRecordReader.java
+++ b/omnioperator/omniop-hive-extension/src/main/java/com/huawei/boostkit/hive/reader/OmniOrcRecordReader.java
@@ -19,9 +19,19 @@
 package com.huawei.boostkit.hive.reader;
 
 import static com.huawei.boostkit.hive.cache.VectorCache.BATCH;
+import static com.huawei.boostkit.hive.expression.TypeUtils.DEFAULT_VARCHAR_LENGTH;
 import static org.apache.hadoop.hive.ql.io.orc.OrcInputFormat.getDesiredRowTypeDescr;
 import static org.apache.hadoop.hive.serde2.ColumnProjectionUtils.READ_COLUMN_IDS_CONF_STR;
 
+import nova.hetu.omniruntime.type.BooleanDataType;
+import nova.hetu.omniruntime.type.DataType;
+import nova.hetu.omniruntime.type.Decimal128DataType;
+import nova.hetu.omniruntime.type.DoubleDataType;
+import nova.hetu.omniruntime.type.IntDataType;
+import nova.hetu.omniruntime.type.LongDataType;
+import nova.hetu.omniruntime.type.ShortDataType;
+import nova.hetu.omniruntime.type.VarcharDataType;
+import nova.hetu.omniruntime.vector.Decimal128Vec;
 import nova.hetu.omniruntime.vector.Vec;
 import nova.hetu.omniruntime.vector.VecBatch;
 
@@ -35,6 +45,7 @@ import org.apache.hadoop.hive.ql.io.sarg.SearchArgument;
 import org.apache.hadoop.hive.ql.io.sarg.SearchArgumentImpl;
 import org.apache.hadoop.hive.ql.plan.api.OperatorType;
 import org.apache.hadoop.hive.serde2.SerDeStats;
+import org.apache.hadoop.hive.serde2.objectinspector.PrimitiveObjectInspector;
 import org.apache.hadoop.io.NullWritable;
 import org.apache.hadoop.mapred.FileSplit;
 import org.apache.hadoop.mapred.RecordReader;
@@ -45,12 +56,31 @@ import org.apache.orc.TypeDescription;
 
 import java.io.IOException;
 import java.util.Arrays;
+import java.util.HashMap;
 import java.util.HashSet;
 import java.util.List;
+import java.util.Map;
 import java.util.Set;
 import java.util.stream.Collectors;
 
 public class OmniOrcRecordReader implements RecordReader<NullWritable, VecBatchWrapper>, StatsProvidingRecordReader {
+    private static final Map<TypeDescription.Category, DataType> CATEGORY_TO_OMNI_TYPE = new HashMap<TypeDescription.Category, DataType>() {
+        {
+            put(TypeDescription.Category.SHORT, ShortDataType.SHORT);
+            put(TypeDescription.Category.INT, IntDataType.INTEGER);
+            put(TypeDescription.Category.LONG, LongDataType.LONG);
+            put(TypeDescription.Category.BOOLEAN, BooleanDataType.BOOLEAN);
+            put(TypeDescription.Category.DOUBLE, DoubleDataType.DOUBLE);
+            put(TypeDescription.Category.STRING, new VarcharDataType(DEFAULT_VARCHAR_LENGTH));
+            put(TypeDescription.Category.TIMESTAMP, LongDataType.LONG);
+            put(TypeDescription.Category.DATE, IntDataType.INTEGER);
+            put(TypeDescription.Category.BYTE, ShortDataType.SHORT);
+            put(TypeDescription.Category.FLOAT, DoubleDataType.DOUBLE);
+            put(TypeDescription.Category.DECIMAL, Decimal128DataType.DECIMAL128);
+            put(TypeDescription.Category.CHAR, VarcharDataType.VARCHAR);
+            put(TypeDescription.Category.VARCHAR, VarcharDataType.VARCHAR);
+        }
+    };
     protected OrcColumnarBatchScanReader recordReader;
     protected Vec[] vecs;
     protected final long offset;
@@ -59,6 +89,7 @@ public class OmniOrcRecordReader implements RecordReader<NullWritable, VecBatchW
     protected final SerDeStats stats;
     protected List<Integer> included;
     protected Operator tableScanOp;
+    protected int[] typeIds;
 
     OmniOrcRecordReader(Configuration conf, FileSplit split) throws IOException {
         TypeDescription schema = getDesiredRowTypeDescr(conf, false, Integer.MAX_VALUE);
@@ -70,6 +101,10 @@ public class OmniOrcRecordReader implements RecordReader<NullWritable, VecBatchW
         TypeDescription requiredSchema = getRequiredSchema(schema);
         options.schema(requiredSchema);
         options.include(null);
+        typeIds = new int[included.size()];
+        for (int i=0; i< requiredSchema.getChildren().size(); i++) {
+            typeIds[i] = CATEGORY_TO_OMNI_TYPE.get(requiredSchema.getChildren().get(i).getCategory()).getId().toValue();
+        }
         recordReader = new OrcColumnarBatchScanReader();
         recordReader.initializeReaderJava(split.getPath().toUri(), readerOptions);
         recordReader.initializeRecordReaderJava(options);
@@ -127,7 +162,7 @@ public class OmniOrcRecordReader implements RecordReader<NullWritable, VecBatchW
         if (included.size() == 0) {
             batchSize = (int) recordReader.getNumberOfRowsJava();
         } else {
-            batchSize = recordReader.next(vecs);
+            batchSize = recordReader.next(vecs, typeIds);
         }
         if (batchSize == 0) {
             return false;
diff --git a/omnioperator/omniop-hive-extension/src/main/java/com/huawei/boostkit/hive/reader/OmniVectorizedOrcRecordReader.java b/omnioperator/omniop-hive-extension/src/main/java/com/huawei/boostkit/hive/reader/OmniVectorizedOrcRecordReader.java
index 0fdb7c2e1eb9403044f9d234e65df8dbbecca801..9340e031e6a953948431bab508578102b27876b5 100644
--- a/omnioperator/omniop-hive-extension/src/main/java/com/huawei/boostkit/hive/reader/OmniVectorizedOrcRecordReader.java
+++ b/omnioperator/omniop-hive-extension/src/main/java/com/huawei/boostkit/hive/reader/OmniVectorizedOrcRecordReader.java
@@ -77,7 +77,7 @@ public class OmniVectorizedOrcRecordReader extends OmniOrcRecordReader {
         if (included.size() == 0) {
             batchSize = (int) recordReader.getNumberOfRowsJava();
         } else {
-            batchSize = recordReader.next(vecs);
+            batchSize = recordReader.next(vecs, typeIds);
         }
         if (batchSize == 0) {
             return false;
diff --git a/omnioperator/omniop-hive-extension/src/main/java/com/huawei/boostkit/hive/reader/OrcColumnarBatchScanReader.java b/omnioperator/omniop-hive-extension/src/main/java/com/huawei/boostkit/hive/reader/OrcColumnarBatchScanReader.java
index 24f9de934fe4a02e39abd36504bc052d54d8468a..41e293fb96300adf72d3b670f147e543acc73081 100644
--- a/omnioperator/omniop-hive-extension/src/main/java/com/huawei/boostkit/hive/reader/OrcColumnarBatchScanReader.java
+++ b/omnioperator/omniop-hive-extension/src/main/java/com/huawei/boostkit/hive/reader/OrcColumnarBatchScanReader.java
@@ -246,17 +246,19 @@ public class OrcColumnarBatchScanReader {
         jniReader.recordReaderSeekToRow(recordReader, rowNumber);
     }
 
-    public int next(Vec[] vecList) {
+    public int next(Vec[] vecList, int[] typeIds) {
         int vectorCnt = vecList.length;
-        int[] typeIds = new int[realColsCnt];
-        long[] vecNativeIds = new long[realColsCnt];
+        long[] vecNativeIds = new long[typeIds.length];
         long rtn = jniReader.recordReaderNext(recordReader, batchReader, typeIds, vecNativeIds);
         if (rtn == 0) {
             return 0;
         }
         int nativeGetId = 0;
+        int rowCount = 0;
         for (int i = 0; i < vectorCnt; i++) {
             if (colsToGet[i] != 0) {
+                vecList[i] = getNullVec(DataType.DataTypeId.values()[typeIds[nativeGetId]], rowCount);
+                nativeGetId++;
                 continue;
             }
             switch (DataType.DataTypeId.values()[typeIds[nativeGetId]]) {
@@ -296,7 +298,51 @@ public class OrcColumnarBatchScanReader {
                 }
             }
             nativeGetId++;
+            rowCount = vecList[i].getSize();
         }
         return (int) rtn;
     }
+
+    private Vec getNullVec(DataType.DataTypeId value, int rowCount) {
+        Vec vec;
+        switch (value) {
+            case OMNI_BOOLEAN: {
+                vec = new BooleanVec(rowCount);
+                break;
+            }
+            case OMNI_SHORT: {
+                vec = new ShortVec(rowCount);
+                break;
+            }
+            case OMNI_DATE32:
+            case OMNI_INT: {
+                vec = new IntVec(rowCount);
+                break;
+            }
+            case OMNI_LONG:
+            case OMNI_DECIMAL64: {
+                vec = new LongVec(rowCount);
+                break;
+            }
+            case OMNI_DOUBLE: {
+                vec = new DoubleVec(rowCount);
+                break;
+            }
+            case OMNI_VARCHAR: {
+                vec = new VarcharVec(rowCount);
+                break;
+            }
+            case OMNI_DECIMAL128: {
+                vec = new Decimal128Vec(rowCount);
+                break;
+            }
+            default: {
+                throw new RuntimeException("UnSupport type for ColumnarFileScan:" + value);
+            }
+        }
+        byte[] nullsBuf = new byte[rowCount];
+        Arrays.fill(nullsBuf, (byte) 1);
+        vec.setNullsBuf(nullsBuf);
+        return vec;
+    }
 }
diff --git a/omnioperator/omniop-native-reader/cpp/src/jni/OrcColumnarBatchJniReader.cpp b/omnioperator/omniop-native-reader/cpp/src/jni/OrcColumnarBatchJniReader.cpp
index 6cd256b8632c3f25d2c030a1c55b0f988002540b..99956458a90124e168d708fc49358f97549fa622 100644
--- a/omnioperator/omniop-native-reader/cpp/src/jni/OrcColumnarBatchJniReader.cpp
+++ b/omnioperator/omniop-native-reader/cpp/src/jni/OrcColumnarBatchJniReader.cpp
@@ -511,53 +511,95 @@ uint64_t CopyToOmniDecimal128VecFrom64(orc::ColumnVectorBatch *field)
     return (uint64_t)newVector;
 }
 
-int CopyToOmniVec(const orc::Type *type, int &omniTypeId, uint64_t &omniVecId, orc::ColumnVectorBatch *field,
+uint64_t dealLongVectorBatch(DataTypeId id, orc::ColumnVectorBatch *field) {
+    switch (id) {
+        case omniruntime::type::OMNI_BOOLEAN:
+            return CopyFixedWidth<OMNI_BOOLEAN, orc::LongVectorBatch>(field);
+        case omniruntime::type::OMNI_SHORT:
+            return CopyFixedWidth<OMNI_SHORT, orc::LongVectorBatch>(field);
+        case omniruntime::type::OMNI_INT:
+            return CopyFixedWidth<OMNI_INT, orc::LongVectorBatch>(field);
+        case omniruntime::type::OMNI_LONG:
+            return CopyOptimizedForInt64<OMNI_LONG, orc::LongVectorBatch>(field);
+        case omniruntime::type::OMNI_DATE32:
+            return CopyFixedWidth<OMNI_DATE32, orc::LongVectorBatch>(field);
+        case omniruntime::type::OMNI_DATE64:
+            return CopyOptimizedForInt64<OMNI_DATE64, orc::LongVectorBatch>(field);
+        default: {
+            throw std::runtime_error("dealLongVectorBatch not support for type: " + id);
+        }
+    }
+    return -1;
+}
+
+uint64_t dealDoubleVectorBatch(DataTypeId id, orc::ColumnVectorBatch *field) {
+    switch (id) {
+        case omniruntime::type::OMNI_DOUBLE:
+            return CopyOptimizedForInt64<OMNI_DOUBLE, orc::DoubleVectorBatch>(field);
+        default: {
+            throw std::runtime_error("dealDoubleVectorBatch not support for type: " + id);
+        }
+    }
+    return -1;
+}
+
+uint64_t dealDecimal64VectorBatch(DataTypeId id, orc::ColumnVectorBatch *field) {
+    switch (id) {
+        case omniruntime::type::OMNI_DECIMAL64:
+            return CopyToOmniDecimal64Vec(field);
+        case omniruntime::type::OMNI_DECIMAL128:
+            return CopyToOmniDecimal128VecFrom64(field);
+        default: {
+            throw std::runtime_error("dealDecimal64VectorBatch not support for type: " + id);
+        }
+    }
+    return -1;
+}
+
+uint64_t dealDecimal128VectorBatch(DataTypeId id, orc::ColumnVectorBatch *field) {
+    switch (id) {
+        case omniruntime::type::OMNI_DECIMAL128:
+            return CopyToOmniDecimal128Vec(field);
+        default: {
+            throw std::runtime_error("dealDecimal128VectorBatch not support for type: " + id);
+        }
+    }
+    return -1;
+}
+
+int CopyToOmniVec(const orc::Type *type, int omniTypeId, uint64_t &omniVecId, orc::ColumnVectorBatch *field,
     bool isDecimal64Transfor128)
 {
+    DataTypeId dataTypeId = static_cast<DataTypeId>(omniTypeId);
     switch (type->getKind()) {
         case orc::TypeKind::BOOLEAN:
-            omniTypeId = static_cast<jint>(OMNI_BOOLEAN);
-            omniVecId = CopyFixedWidth<OMNI_BOOLEAN, orc::LongVectorBatch>(field);
-            break;
         case orc::TypeKind::SHORT:
-            omniTypeId = static_cast<jint>(OMNI_SHORT);
-            omniVecId = CopyFixedWidth<OMNI_SHORT, orc::LongVectorBatch>(field);
-            break;
         case orc::TypeKind::DATE:
-            omniTypeId = static_cast<jint>(OMNI_DATE32);
-            omniVecId = CopyFixedWidth<OMNI_INT, orc::LongVectorBatch>(field);
-            break;
         case orc::TypeKind::INT:
-            omniTypeId = static_cast<jint>(OMNI_INT);
-            omniVecId = CopyFixedWidth<OMNI_INT, orc::LongVectorBatch>(field);
-            break;
         case orc::TypeKind::LONG:
-            omniTypeId = static_cast<int>(OMNI_LONG);
-            omniVecId = CopyOptimizedForInt64<OMNI_LONG, orc::LongVectorBatch>(field);
+            omniVecId = dealLongVectorBatch(dataTypeId, field);
             break;
         case orc::TypeKind::DOUBLE:
-            omniTypeId = static_cast<int>(OMNI_DOUBLE);
-            omniVecId = CopyOptimizedForInt64<OMNI_DOUBLE, orc::DoubleVectorBatch>(field);
+            omniVecId = dealDoubleVectorBatch(dataTypeId, field);
             break;
         case orc::TypeKind::CHAR:
-            omniTypeId = static_cast<int>(OMNI_VARCHAR);
+            if (dataTypeId != OMNI_VARCHAR) {
+                throw std::runtime_error("Cannot transfer to other OMNI_TYPE but VARCHAR for orc char");
+            }
             omniVecId = CopyCharType(field);
             break;
         case orc::TypeKind::STRING:
         case orc::TypeKind::VARCHAR:
-            omniTypeId = static_cast<int>(OMNI_VARCHAR);
+            if (dataTypeId != OMNI_VARCHAR) {
+                throw std::runtime_error("Cannot transfer to other OMNI_TYPE but VARCHAR for orc string/varchar");
+            }
             omniVecId = CopyVarWidth(field);
             break;
         case orc::TypeKind::DECIMAL:
             if (type->getPrecision() > MAX_DECIMAL64_DIGITS) {
-                omniTypeId = static_cast<int>(OMNI_DECIMAL128);
-                omniVecId = CopyToOmniDecimal128Vec(field);
-            } else if (isDecimal64Transfor128) {
-                omniTypeId = static_cast<int>(OMNI_DECIMAL128);
-                omniVecId = CopyToOmniDecimal128VecFrom64(field);
+                omniVecId = dealDecimal128VectorBatch(dataTypeId, field);
             } else {
-                omniTypeId = static_cast<int>(OMNI_DECIMAL64);
-                omniVecId = CopyToOmniDecimal64Vec(field);
+                omniVecId = dealDecimal64VectorBatch(dataTypeId, field);
             }
             break;
         default: {
@@ -576,16 +618,16 @@ JNIEXPORT jlong JNICALL Java_com_huawei_boostkit_scan_jni_OrcColumnarBatchJniRea
     const orc::Type &baseTp = rowReaderPtr->getSelectedType();
     int vecCnt = 0;
     long batchRowSize = 0;
+    auto ptr = env->GetIntArrayElements(typeId, JNI_FALSE);
     if (rowReaderPtr->next(*columnVectorBatch)) {
         orc::StructVectorBatch *root = dynamic_cast<orc::StructVectorBatch *>(columnVectorBatch);
         vecCnt = root->fields.size();
         batchRowSize = root->fields[0]->numElements;
         for (int id = 0; id < vecCnt; id++) {
             auto type = baseTp.getSubtype(id);
-            int omniTypeId = 0;
+            int omniTypeId = ptr[id];
             uint64_t omniVecId = 0;
             CopyToOmniVec(type, omniTypeId, omniVecId, root->fields[id], isDecimal64Transfor128);
-            env->SetIntArrayRegion(typeId, id, 1, &omniTypeId);
             jlong omniVec = static_cast<jlong>(omniVecId);
             env->SetLongArrayRegion(vecNativeId, id, 1, &omniVec);
         }
diff --git a/omnioperator/omniop-native-reader/cpp/src/jni/OrcColumnarBatchJniReader.h b/omnioperator/omniop-native-reader/cpp/src/jni/OrcColumnarBatchJniReader.h
index 829f5c0744d3d563601ec6506ebfc82b5a020e93..e0c33b26c7f191771fc2c2bf7aaa1a278db8f424 100644
--- a/omnioperator/omniop-native-reader/cpp/src/jni/OrcColumnarBatchJniReader.h
+++ b/omnioperator/omniop-native-reader/cpp/src/jni/OrcColumnarBatchJniReader.h
@@ -141,7 +141,7 @@ int BuildLeaves(PredicateOperatorType leafOp, std::vector<orc::Literal> &litList
 
 bool StringToBool(const std::string &boolStr);
 
-int CopyToOmniVec(const orc::Type *type, int &omniTypeId, uint64_t &omniVecId, orc::ColumnVectorBatch *field,
+int CopyToOmniVec(const orc::Type *type, int omniTypeId, uint64_t &omniVecId, orc::ColumnVectorBatch *field,
     bool isDecimal64Transfor128);
 
 #ifdef __cplusplus
diff --git a/omnioperator/omniop-spark-extension/java/src/main/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchScanReader.java b/omnioperator/omniop-spark-extension/java/src/main/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchScanReader.java
index 8edbdf4622f0b5888cd4fd680fee6fc1eb3c4880..611a10826042a7916ff7d98f804c7ee5eba319b1 100644
--- a/omnioperator/omniop-spark-extension/java/src/main/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchScanReader.java
+++ b/omnioperator/omniop-spark-extension/java/src/main/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchScanReader.java
@@ -34,7 +34,6 @@ import org.slf4j.LoggerFactory;
 import java.net.URI;
 import java.sql.Date;
 import java.util.ArrayList;
-import java.util.Arrays;
 import java.util.List;
 
 public class OrcColumnarBatchScanReader {
@@ -253,8 +252,7 @@ public class OrcColumnarBatchScanReader {
         }
     }
 
-    public int next(Vec[] vecList) {
-        int[] typeIds = new int[realColsCnt];
+    public int next(Vec[] vecList, int[] typeIds) {
         long[] vecNativeIds = new long[realColsCnt];
         long rtn = jniReader.recordReaderNext(recordReader, batchReader, typeIds, vecNativeIds);
         if (rtn == 0) {
diff --git a/omnioperator/omniop-spark-extension/java/src/main/java/org/apache/spark/sql/execution/datasources/orc/OmniOrcColumnarBatchReader.java b/omnioperator/omniop-spark-extension/java/src/main/java/org/apache/spark/sql/execution/datasources/orc/OmniOrcColumnarBatchReader.java
index 24a93ede4cf4e17ccd0050157e04a848ab38ba0d..e8e7db3af876ce726d9b22e8ead47cf9f15a9fbe 100644
--- a/omnioperator/omniop-spark-extension/java/src/main/java/org/apache/spark/sql/execution/datasources/orc/OmniOrcColumnarBatchReader.java
+++ b/omnioperator/omniop-spark-extension/java/src/main/java/org/apache/spark/sql/execution/datasources/orc/OmniOrcColumnarBatchReader.java
@@ -19,6 +19,7 @@
 package org.apache.spark.sql.execution.datasources.orc;
 
 import com.google.common.annotations.VisibleForTesting;
+import com.huawei.boostkit.spark.expression.OmniExpressionAdaptor;
 import com.huawei.boostkit.spark.jni.OrcColumnarBatchScanReader;
 import nova.hetu.omniruntime.vector.Vec;
 import org.apache.hadoop.conf.Configuration;
@@ -79,6 +80,8 @@ public class OmniOrcColumnarBatchReader extends RecordReader<Void, ColumnarBatch
 
   private Vec[] vecs;
 
+  private int[] vecTypeIds;
+
   private Reader.Options options;
 
   public OmniOrcColumnarBatchReader(int capacity) {
@@ -242,6 +245,11 @@ public class OmniOrcColumnarBatchReader extends RecordReader<Void, ColumnarBatch
     recordReader.initBatchJava(capacity);
     vecs = new Vec[orcVectorWrappers.length];
     columnarBatch = new ColumnarBatch(orcVectorWrappers);
+
+    vecTypeIds = new int[orcVectorWrappers.length];
+    for (int i = 0; i < vecTypeIds.length; i++) {
+      vecTypeIds[i] = OmniExpressionAdaptor.sparkTypeToOmniType(resultSchema.fields()[i].dataType());
+    }
   }
 
   /**
@@ -253,7 +261,7 @@ public class OmniOrcColumnarBatchReader extends RecordReader<Void, ColumnarBatch
     if ((resultFields.length == 1 && requestedDataColIds[0] == -1) || resultFields.length == 0 || requiredFields.length == 0) {
       batchSize = (int) recordReader.getNumberOfRowsJava();
     } else {
-      batchSize = recordReader.next(vecs);
+      batchSize = recordReader.next(vecs, vecTypeIds);
     }
     if (batchSize == 0) {
       return false;
diff --git a/omnioperator/omniop-spark-extension/java/src/main/scala/com/huawei/boostkit/spark/expression/OmniExpressionAdaptor.scala b/omnioperator/omniop-spark-extension/java/src/main/scala/com/huawei/boostkit/spark/expression/OmniExpressionAdaptor.scala
index a0e9484408c867e24ce052fd01553d33fda5cbfa..3776fdad1768c91f59e8fc2c1b23b83768faebdb 100644
--- a/omnioperator/omniop-spark-extension/java/src/main/scala/com/huawei/boostkit/spark/expression/OmniExpressionAdaptor.scala
+++ b/omnioperator/omniop-spark-extension/java/src/main/scala/com/huawei/boostkit/spark/expression/OmniExpressionAdaptor.scala
@@ -663,6 +663,10 @@ object OmniExpressionAdaptor extends Logging {
     }
   }
 
+  def sparkTypeToOmniType(dataType: DataType): Int = {
+    sparkTypeToOmniType(dataType, Metadata.empty).getId.ordinal()
+  }
+
   def sparkTypeToOmniType(dataType: DataType, metadata: Metadata = Metadata.empty):
   nova.hetu.omniruntime.`type`.DataType = {
     dataType match {
diff --git a/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderDataTypeTest.java b/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderDataTypeTest.java
index 0c65d0bcc16354910ba8808995caf9fff4779aba..2001ae7d034709fa2a74435cd103e65d53cd2ce6 100644
--- a/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderDataTypeTest.java
+++ b/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderDataTypeTest.java
@@ -40,6 +40,10 @@ import java.util.ArrayList;
 import java.net.URI;
 import java.net.URISyntaxException;
 
+import static nova.hetu.omniruntime.type.DataType.DataTypeId.OMNI_LONG;
+import static nova.hetu.omniruntime.type.DataType.DataTypeId.OMNI_VARCHAR;
+import static nova.hetu.omniruntime.type.DataType.DataTypeId.OMNI_INT;
+
 @FixMethodOrder(value = MethodSorters.NAME_ASCENDING )
 public class OrcColumnarBatchJniReaderDataTypeTest extends TestCase {
     public OrcColumnarBatchScanReader orcColumnarBatchScanReader;
@@ -101,7 +105,7 @@ public class OrcColumnarBatchJniReaderDataTypeTest extends TestCase {
 
     @Test
     public void testNext() {
-        int[] typeId = new int[4];
+        int[] typeId = new int[] {OMNI_LONG.ordinal(), OMNI_VARCHAR.ordinal(), OMNI_VARCHAR.ordinal(), OMNI_INT.ordinal()};
         long[] vecNativeId = new long[4];
         long rtn = orcColumnarBatchScanReader.jniReader.recordReaderNext(orcColumnarBatchScanReader.recordReader, orcColumnarBatchScanReader.batchReader, typeId, vecNativeId);
         assertTrue(rtn == 4096);
diff --git a/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderNotPushDownTest.java b/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderNotPushDownTest.java
index 0a06c7016a1f945e433f140e4e1759569b5824f2..358eb6c62b627d5ce9927e256bad2294f4448a47 100644
--- a/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderNotPushDownTest.java
+++ b/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderNotPushDownTest.java
@@ -37,6 +37,9 @@ import java.util.ArrayList;
 import java.net.URI;
 import java.net.URISyntaxException;
 
+import static nova.hetu.omniruntime.type.DataType.DataTypeId.OMNI_LONG;
+import static nova.hetu.omniruntime.type.DataType.DataTypeId.OMNI_VARCHAR;
+
 @FixMethodOrder(value = MethodSorters.NAME_ASCENDING )
 public class OrcColumnarBatchJniReaderNotPushDownTest extends TestCase {
     public OrcColumnarBatchScanReader orcColumnarBatchScanReader;
@@ -92,7 +95,7 @@ public class OrcColumnarBatchJniReaderNotPushDownTest extends TestCase {
 
     @Test
     public void testNext() {
-        int[] typeId = new int[2];
+        int[] typeId = new int[] {OMNI_LONG.ordinal(), OMNI_VARCHAR.ordinal()};
         long[] vecNativeId = new long[2];
         long rtn = orcColumnarBatchScanReader.jniReader.recordReaderNext(orcColumnarBatchScanReader.recordReader, orcColumnarBatchScanReader.batchReader, typeId, vecNativeId);
         assertTrue(rtn == 4096);
diff --git a/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderPushDownTest.java b/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderPushDownTest.java
index 28598ebc9a2322c86892e779c73014a2dbbf6aaf..029516c1c33dc66f7a3caab4b39a25af52d754f7 100644
--- a/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderPushDownTest.java
+++ b/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderPushDownTest.java
@@ -45,6 +45,9 @@ import java.net.URISyntaxException;
 import java.lang.reflect.Array;
 import java.util.ArrayList;
 
+import static nova.hetu.omniruntime.type.DataType.DataTypeId.OMNI_LONG;
+import static nova.hetu.omniruntime.type.DataType.DataTypeId.OMNI_VARCHAR;
+
 @FixMethodOrder(value = MethodSorters.NAME_ASCENDING )
 public class OrcColumnarBatchJniReaderPushDownTest extends TestCase {
     public OrcColumnarBatchScanReader orcColumnarBatchScanReader;
@@ -146,7 +149,7 @@ public class OrcColumnarBatchJniReaderPushDownTest extends TestCase {
 
     @Test
     public void testNext() {
-        int[] typeId = new int[2];
+        int[] typeId = new int[] {OMNI_LONG.ordinal(), OMNI_VARCHAR.ordinal()};
         long[] vecNativeId = new long[2];
         long rtn = orcColumnarBatchScanReader.jniReader.recordReaderNext(orcColumnarBatchScanReader.recordReader, orcColumnarBatchScanReader.batchReader, typeId, vecNativeId);
         assertTrue(rtn == 4096);
diff --git a/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderSparkORCNotPushDownTest.java b/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderSparkORCNotPushDownTest.java
index 4b9625930a13a7ed570bf1d741a3b919780b50ce..0c83e19102b1a03d4133cacdc246ba1a9f639489 100644
--- a/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderSparkORCNotPushDownTest.java
+++ b/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderSparkORCNotPushDownTest.java
@@ -40,6 +40,10 @@ import java.util.ArrayList;
 import java.net.URI;
 import java.net.URISyntaxException;
 
+import static nova.hetu.omniruntime.type.DataType.DataTypeId.OMNI_LONG;
+import static nova.hetu.omniruntime.type.DataType.DataTypeId.OMNI_VARCHAR;
+import static nova.hetu.omniruntime.type.DataType.DataTypeId.OMNI_INT;
+
 @FixMethodOrder(value = MethodSorters.NAME_ASCENDING )
 public class OrcColumnarBatchJniReaderSparkORCNotPushDownTest extends TestCase {
     public OrcColumnarBatchScanReader orcColumnarBatchScanReader;
@@ -101,7 +105,7 @@ public class OrcColumnarBatchJniReaderSparkORCNotPushDownTest extends TestCase {
 
     @Test
     public void testNext() {
-        int[] typeId = new int[4];
+        int[] typeId = new int[] {OMNI_LONG.ordinal(), OMNI_VARCHAR.ordinal(), OMNI_VARCHAR.ordinal(), OMNI_INT.ordinal()};
         long[] vecNativeId = new long[4];
         long rtn = orcColumnarBatchScanReader.jniReader.recordReaderNext(orcColumnarBatchScanReader.recordReader, orcColumnarBatchScanReader.batchReader, typeId, vecNativeId);
         assertTrue(rtn == 4096);
diff --git a/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderSparkORCPushDownTest.java b/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderSparkORCPushDownTest.java
index 196d1ff350efa81be59282b090a9375edfd01a10..8246e5606e626fee237ea246d1d5454ee37232c0 100644
--- a/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderSparkORCPushDownTest.java
+++ b/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderSparkORCPushDownTest.java
@@ -40,6 +40,10 @@ import java.util.ArrayList;
 import java.net.URI;
 import java.net.URISyntaxException;
 
+import static nova.hetu.omniruntime.type.DataType.DataTypeId.OMNI_LONG;
+import static nova.hetu.omniruntime.type.DataType.DataTypeId.OMNI_VARCHAR;
+import static nova.hetu.omniruntime.type.DataType.DataTypeId.OMNI_INT;
+
 @FixMethodOrder(value = MethodSorters.NAME_ASCENDING )
 public class OrcColumnarBatchJniReaderSparkORCPushDownTest extends TestCase {
     public OrcColumnarBatchScanReader orcColumnarBatchScanReader;
@@ -147,7 +151,7 @@ public class OrcColumnarBatchJniReaderSparkORCPushDownTest extends TestCase {
 
     @Test
     public void testNext() {
-        int[] typeId = new int[4];
+        int[] typeId = new int[] {OMNI_LONG.ordinal(), OMNI_VARCHAR.ordinal(), OMNI_VARCHAR.ordinal(), OMNI_INT.ordinal()};
         long[] vecNativeId = new long[4];
         long rtn = orcColumnarBatchScanReader.jniReader.recordReaderNext(orcColumnarBatchScanReader.recordReader, orcColumnarBatchScanReader.batchReader, typeId, vecNativeId);
         assertTrue(rtn == 4096);
diff --git a/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderTest.java b/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderTest.java
index c8581f35ebc605845896b5f35731b0908779f326..eab15fef660250780e0beb311b489e3ceeb8ff5b 100644
--- a/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderTest.java
+++ b/omnioperator/omniop-spark-extension/java/src/test/java/com/huawei/boostkit/spark/jni/OrcColumnarBatchJniReaderTest.java
@@ -21,20 +21,15 @@ package com.huawei.boostkit.spark.jni;
 import com.esotericsoftware.kryo.Kryo;
 import com.esotericsoftware.kryo.io.Input;
 import junit.framework.TestCase;
-import nova.hetu.omniruntime.type.DataType;
-import nova.hetu.omniruntime.vector.IntVec;
 import nova.hetu.omniruntime.vector.LongVec;
 import nova.hetu.omniruntime.vector.VarcharVec;
 import nova.hetu.omniruntime.vector.Vec;
 import org.apache.commons.codec.binary.Base64;
 import org.apache.hadoop.hive.ql.io.sarg.SearchArgument;
 import org.apache.hadoop.hive.ql.io.sarg.SearchArgumentImpl;
-import org.apache.orc.OrcConf;
 import org.apache.orc.OrcFile;
-import org.apache.orc.Reader;
 import org.apache.orc.TypeDescription;
 import org.apache.orc.mapred.OrcInputFormat;
-import org.json.JSONObject;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.FixMethodOrder;
@@ -51,6 +46,9 @@ import org.apache.orc.Reader.Options;
 
 import static org.junit.Assert.*;
 
+import static nova.hetu.omniruntime.type.DataType.DataTypeId.OMNI_LONG;
+import static nova.hetu.omniruntime.type.DataType.DataTypeId.OMNI_VARCHAR;
+
 @FixMethodOrder(value = MethodSorters.NAME_ASCENDING )
 public class OrcColumnarBatchJniReaderTest extends TestCase {
     public Configuration conf = new Configuration();
@@ -152,7 +150,8 @@ public class OrcColumnarBatchJniReaderTest extends TestCase {
     @Test
     public void testNext() {
         Vec[] vecs = new Vec[2];
-        long rtn = orcColumnarBatchScanReader.next(vecs);
+        int[] typeId = new int[] {OMNI_LONG.ordinal(), OMNI_VARCHAR.ordinal()};
+        long rtn = orcColumnarBatchScanReader.next(vecs, typeId);
         assertTrue(rtn == 4096);
         assertTrue(((LongVec) vecs[0]).get(0) == 1);
         String str = new String(((VarcharVec) vecs[1]).get(0));