add the timestamp unit convert and date64 to date32 convert (apache#10)

rui-mo · Mar 15, 2022 · b811991 · b811991
1 parent a3f3a5e
commit b811991
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/cpp/src/gandiva/function_registry_datetime.cc b/cpp/src/gandiva/function_registry_datetime.cc
@@ -138,6 +138,15 @@ std::vector<NativeFunction> GetDateTimeFunctionRegistry() {
  DATE_TYPES(LAST_DAY_SAFE_NULL_IF_NULL, last_day, {}),
  BASE_NUMERIC_TYPES(TO_TIME_SAFE_NULL_IF_NULL, to_time, {}),
  BASE_NUMERIC_TYPES(TO_TIMESTAMP_SAFE_NULL_IF_NULL, to_timestamp, {})};
+ NativeFunction("convertTimestampUnit", {}, DataTypeVector{timestamp()}, arrow::timestamp(arrow::TimeUnit::MICRO),
+ kResultNullIfNull, "convertTimestampUnit_ms"),
+
+ NativeFunction("convertTimestampUnit", {}, DataTypeVector{arrow::timestamp(arrow::TimeUnit::MICRO)}, timestamp(),
+ kResultNullIfNull, "convertTimestampUnit_us"),
+
+ NativeFunction("castDATE", {}, DataTypeVector{date64()}, date32(),
+ kResultNullIfNull, "castDATE_date64"),
+ DATE_TYPES(LAST_DAY_SAFE_NULL_IF_NULL, last_day, {});
 
  return date_time_fn_registry_;
 }

diff --git a/cpp/src/gandiva/precompiled/time.cc b/cpp/src/gandiva/precompiled/time.cc
@@ -904,6 +904,18 @@ gdv_time32 castTIME_timestamp(gdv_timestamp timestamp_in_millis) {
  return static_cast<int32_t>(millis_since_midnight);
 }
 
+gdv_timestamp convertTimestampUnit_ms(gdv_timestamp timestamp_in_millis) {
+ return timestamp_in_millis * 1000;
+}
+
+gdv_timestamp convertTimestampUnit_us(gdv_timestamp timestamp_in_micro) {
+ return timestamp_in_micro / 1000;
+}
+
+gdv_date32 castDATE_date64(gdv_date64 date_in_millis) {
+ return static_cast<gdv_date32>(date_in_millis / (MILLIS_IN_DAY));
+}
+
 // Gets an arbitrary number and return the number of milliseconds since midnight
 gdv_time32 castTIME_int32(int32_t int_val) {
  if (int_val < 0) {

diff --git a/cpp/src/gandiva/precompiled/types.h b/cpp/src/gandiva/precompiled/types.h
@@ -423,6 +423,9 @@ gdv_timestamp castTIMESTAMP_int64(gdv_int64);
 gdv_date64 castDATE_timestamp(gdv_timestamp);
 gdv_time32 castTIME_utf8(int64_t context, const char* input, int32_t length);
 gdv_time32 castTIME_timestamp(gdv_timestamp timestamp_in_millis);
+gdv_timestamp convertTimestampUnit_ms(gdv_timestamp);
+gdv_timestamp convertTimestampUnit_us(gdv_timestamp);
+gdv_date32 castDATE_date64(gdv_date64 date);
 gdv_time32 castTIME_int32(int32_t int_val);
 const char* castVARCHAR_timestamp_int64(int64_t, gdv_timestamp, gdv_int64, gdv_int32*);
 gdv_date64 last_day_from_timestamp(gdv_date64 millis);