Add function registry for to_timestamp method

jpedroantunes · jpedroantunes · commit b41c3c2bf5fe · 2021-05-17T08:21:21.000-03:00
diff --git a/cpp/src/gandiva/function_registry_common.h b/cpp/src/gandiva/function_registry_common.h
@@ -174,6 +174,15 @@ typedef std::unordered_map<const FunctionSignature*, const NativeFunction*, KeyH
   NativeFunction(#NAME, std::vector<std::string> ALIASES, DataTypeVector{TYPE()}, \
                  date64(), kResultNullIfNull, ARROW_STRINGIFY(NAME##_from_##TYPE))
 
+// To timestamp functions (used with data/time types) that :
+// - NULL handling is of type NULL_IF_NULL
+//
+// The pre-compiled fn name includes the base name & input type name. eg:
+// - to_timestamp_date64
+#define TO_TIMESTAMP_SAFE_NULL_IF_NULL(NAME, ALIASES, TYPE)                       \
+  NativeFunction(#NAME, std::vector<std::string> ALIASES, DataTypeVector{TYPE()}, \
+                 timestamp(), kResultNullIfNull, ARROW_STRINGIFY(NAME##_##TYPE))
+
 // Hash32 functions that :
 // - NULL handling is of type NULL_NEVER
 //
diff --git a/cpp/src/gandiva/function_registry_datetime.cc b/cpp/src/gandiva/function_registry_datetime.cc
@@ -86,7 +86,9 @@ std::vector<NativeFunction> GetDateTimeFunctionRegistry() {
       NativeFunction("extractDay", {}, DataTypeVector{day_time_interval()}, int64(),
                      kResultNullIfNull, "extractDay_daytimeinterval"),
 
-      DATE_TYPES(LAST_DAY_SAFE_NULL_IF_NULL, last_day, {})};
+      DATE_TYPES(LAST_DAY_SAFE_NULL_IF_NULL, last_day, {}),
+
+      DATE_TYPES(TO_TIMESTAMP_SAFE_NULL_IF_NULL, to_timestamp, {})};
 
   return date_time_fn_registry_;
 }

Original file line number	Diff line number	Diff line change
`@@ -86,7 +86,9 @@ std::vector<NativeFunction> GetDateTimeFunctionRegistry() {`
`86`	`86`	`NativeFunction("extractDay", {}, DataTypeVector{day_time_interval()}, int64(),`
`87`	`87`	`kResultNullIfNull, "extractDay_daytimeinterval"),`
`88`	`88`
`89`		`- DATE_TYPES(LAST_DAY_SAFE_NULL_IF_NULL, last_day, {})};`
	`89`	`+ DATE_TYPES(LAST_DAY_SAFE_NULL_IF_NULL, last_day, {}),`
	`90`	`+`
	`91`	`+ DATE_TYPES(TO_TIMESTAMP_SAFE_NULL_IF_NULL, to_timestamp, {})};`
`90`	`92`
`91`	`93`	`return date_time_fn_registry_;`
`92`	`94`	`}`