support date32 for date_trunc (apache#12603)

Author: goldmedal

datafusion/functions/src/datetime/date_trunc.rs

@@ -123,7 +123,9 @@ impl ScalarUDFImpl for DateTruncFunc {
 
     fn return_type(&self, arg_types: &[DataType]) -> Result<DataType> {
         match &arg_types[1] {
-            Timestamp(Nanosecond, None) | Utf8 | Null => Ok(Timestamp(Nanosecond, None)),
+            Timestamp(Nanosecond, None) | Utf8 | DataType::Date32 | Null => {
+                Ok(Timestamp(Nanosecond, None))
+            }
             Timestamp(Nanosecond, tz_opt) => Ok(Timestamp(Nanosecond, tz_opt.clone())),
             Timestamp(Microsecond, tz_opt) => Ok(Timestamp(Microsecond, tz_opt.clone())),
             Timestamp(Millisecond, tz_opt) => Ok(Timestamp(Millisecond, tz_opt.clone())),

datafusion/sqllogictest/test_files/dates.slt

@@ -238,5 +238,41 @@ select to_date('2022-01-23', '%Y-%m-%d');
 ----
 2022-01-23
 
+query PPPP
+select
+    date_trunc('YEAR', to_date('2022-02-23', '%Y-%m-%d')),
+    date_trunc('MONTH', to_date('2022-02-23', '%Y-%m-%d')),
+    date_trunc('DAY', to_date('2022-02-23', '%Y-%m-%d')),
+    date_trunc('WEEK', to_date('2022-02-23', '%Y-%m-%d'));
+----
+2022-01-01T00:00:00 2022-02-01T00:00:00 2022-02-23T00:00:00 2022-02-21T00:00:00
+
+query PPPPP
+select
+    date_trunc('HOUR', to_date('2022-02-23', '%Y-%m-%d')),
+    date_trunc('MINUTE', to_date('2022-02-23', '%Y-%m-%d')),
+    date_trunc('SECOND', to_date('2022-02-23', '%Y-%m-%d')),
+    date_trunc('MILLISECOND', to_date('2022-02-23', '%Y-%m-%d')),
+    date_trunc('MICROSECOND', to_date('2022-02-23', '%Y-%m-%d'));
+----
+2022-02-23T00:00:00 2022-02-23T00:00:00 2022-02-23T00:00:00 2022-02-23T00:00:00 2022-02-23T00:00:00
+
+query PPPP
+select
+    date_trunc('YEAR', d2_date),
+    date_trunc('MONTH', d2_date),
+    date_trunc('DAY', d2_date),
+    date_trunc('WEEK', d2_date)
+FROM test
+----
+2022-01-01T00:00:00 2022-12-01T00:00:00 2022-12-12T00:00:00 2022-12-12T00:00:00
+2022-01-01T00:00:00 2022-12-01T00:00:00 2022-12-11T00:00:00 2022-12-05T00:00:00
+2022-01-01T00:00:00 2022-12-01T00:00:00 2022-12-10T00:00:00 2022-12-05T00:00:00
+2022-01-01T00:00:00 2022-12-01T00:00:00 2022-12-09T00:00:00 2022-12-05T00:00:00
+2022-01-01T00:00:00 2022-12-01T00:00:00 2022-12-08T00:00:00 2022-12-05T00:00:00
+2022-01-01T00:00:00 2022-12-01T00:00:00 2022-12-07T00:00:00 2022-12-05T00:00:00
+2022-01-01T00:00:00 2022-12-01T00:00:00 2022-12-06T00:00:00 2022-12-05T00:00:00
+2022-01-01T00:00:00 2022-12-01T00:00:00 2022-12-05T00:00:00 2022-12-05T00:00:00
+
 statement ok
 drop table ts_utf8_data