refactor: replace map..concat with flat_map

apache · Jul 9, 2024 · 5fe18d5 · 5fe18d5
1 parent 043c997
commit 5fe18d5
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/datafusion/functions/src/datetime/to_local_time.rs b/datafusion/functions/src/datetime/to_local_time.rs
@@ -56,17 +56,16 @@ impl Default for ToLocalTimeFunc {
 impl ToLocalTimeFunc {
     pub fn new() -> Self {
         let base_sig = |array_type: TimeUnit| {
-            vec![
+            [
                 Exact(vec![Timestamp(array_type, None)]),
                 Exact(vec![Timestamp(array_type, Some(TIMEZONE_WILDCARD.into()))]),
             ]
         };
 
         let full_sig = [Nanosecond, Microsecond, Millisecond, Second]
             .into_iter()
-            .map(base_sig)
-            .collect::<Vec<_>>()
-            .concat();
+            .flat_map(base_sig)
+            .collect::<Vec<_>>();
 
         Self {
             signature: Signature::one_of(full_sig, Volatility::Immutable),