ARROW-5975: [C++][Gandiva] support castTIMESTAMP(date)

Closes apache#4902 from pprudhvi/casttimestamp-date and squashes the following commits: 49de552 <Prudhvi Porandla> add castTIMESTAMP_date64 method and its unittest Authored-by: Prudhvi Porandla <[email protected]> Signed-off-by: Pindikura Ravindra <[email protected]>
tvamsikalyan · Jul 31, 2019 · 8bdade4 · 8bdade4
1 parent ca2758e
commit 8bdade4
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 1 deletion.
diff --git a/cpp/src/gandiva/function_registry_datetime.cc b/cpp/src/gandiva/function_registry_datetime.cc
@@ -61,7 +61,9 @@ std::vector<NativeFunction> GetDateTimeFunctionRegistry() {
                      kResultNullInternal, "gdv_fn_to_date_utf8_utf8_int32",
                      NativeFunction::kNeedsContext |
                          NativeFunction::kNeedsFunctionHolder |
-                         NativeFunction::kCanReturnErrors)};
+                         NativeFunction::kCanReturnErrors),
+      NativeFunction("castTIMESTAMP", {}, DataTypeVector{date64()}, timestamp(),
+                     kResultNullIfNull, "castTIMESTAMP_date64")};
 
   return date_time_fn_registry_;
 }

diff --git a/cpp/src/gandiva/precompiled/time.cc b/cpp/src/gandiva/precompiled/time.cc
@@ -686,4 +686,7 @@ timestamp castTIMESTAMP_utf8(int64_t context, const char* input, int32 length) {
   }
   return std::chrono::time_point_cast<milliseconds>(date_time).time_since_epoch().count();
 }
+
+timestamp castTIMESTAMP_date64(date64 date_in_millis) { return date_in_millis; }
+
 }  // extern "C"
diff --git a/cpp/src/gandiva/precompiled/time_test.cc b/cpp/src/gandiva/precompiled/time_test.cc
@@ -89,6 +89,9 @@ TEST(TestTime, TestCastTimestamp) {
             -1187308799080);
   EXPECT_EQ(castTIMESTAMP_utf8(context_ptr, "1857-02-11 20:31:40.920 -05:30", 30),
             -3562264699080);
+  EXPECT_EQ(castTIMESTAMP_date64(
+                castDATE_utf8(context_ptr, "2000-09-23 9:45:30.920 +08:00", 29)),
+            castTIMESTAMP_utf8(context_ptr, "2000-09-23 0:00:00.000 +00:00", 29));
 }
 
 #ifndef _WIN32

diff --git a/cpp/src/gandiva/precompiled/types.h b/cpp/src/gandiva/precompiled/types.h
@@ -163,6 +163,7 @@ int32 utf8_length(int64 context, const char* data, int32 data_len);
 date64 castDATE_utf8(int64_t execution_context, const char* input, int32 length);
 
 timestamp castTIMESTAMP_utf8(int64_t execution_context, const char* input, int32 length);
+timestamp castTIMESTAMP_date64(date64);
 
 }  // extern "C"