fix timestamp in native side (#373)

oap-project · Jun 25, 2021 · 343ea32 · 343ea32
1 parent 8fe1253
commit 343ea32
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/native-sql-engine/cpp/src/codegen/arrow_compute/ext/whole_stage_codegen_kernel.cc b/native-sql-engine/cpp/src/codegen/arrow_compute/ext/whole_stage_codegen_kernel.cc
@@ -615,7 +615,8 @@ extern "C" void MakeCodeGen(arrow::compute::ExecContext *ctx,
     std::stringstream codes_ss;
     for (int i = 0; i < output_field_list.size(); i++) {
       auto data_type = output_field_list[i]->type();
-      if (data_type->id() == arrow::Type::DECIMAL) {
+      if (data_type->id() == arrow::Type::DECIMAL ||
+          data_type->id() == arrow::Type::TIMESTAMP) {
         codes_ss << "builder_" << i << "_ = std::make_shared<"
                  << GetTypeString(data_type, "Builder")
                  << ">(arrow::" << GetArrowTypeDefString(data_type)

diff --git a/native-sql-engine/cpp/src/precompile/unsafe_array.cc b/native-sql-engine/cpp/src/precompile/unsafe_array.cc
@@ -41,7 +41,8 @@ namespace precompile {
   PROCESS(arrow::Date32Type)             \
   PROCESS(arrow::Date64Type)             \
   PROCESS(arrow::StringType)             \
-  PROCESS(arrow::Decimal128Type)
+  PROCESS(arrow::Decimal128Type)         \
+  PROCESS(arrow::TimestampType)
 
 arrow::Status MakeUnsafeArray(std::shared_ptr<arrow::DataType> type, int idx,
                               const std::shared_ptr<arrow::Array>& in,