fix closure

apache · Dec 7, 2023 · 47e7cfb · 47e7cfb
1 parent f1ca3be
commit 47e7cfb
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/datafusion/physical-expr/src/string_expressions.rs b/datafusion/physical-expr/src/string_expressions.rs
@@ -323,10 +323,20 @@ fn general_trim<T: OffsetSizeTrait>(
     trim_type: TrimType,
 ) -> Result<ArrayRef> {
     let func = match trim_type {
-        TrimType::Left => str::trim_start_matches::<&str>,
-        TrimType::Right => str::trim_end_matches::<&str>,
-        TrimType::Both => |input, pattern| {
-            str::trim_end_matches(str::trim_start_matches(input, pattern), pattern)
+        TrimType::Left => |input, pattern: &str| {
+            let pattern = pattern.chars().collect::<Vec<char>>();
+            str::trim_start_matches::<&[char]>(input, pattern.as_ref())
+        },
+        TrimType::Right => |input, pattern: &str| {
+            let pattern = pattern.chars().collect::<Vec<char>>();
+            str::trim_end_matches::<&[char]>(input, pattern.as_ref())
+        },
+        TrimType::Both => |input, pattern: &str| {
+            let pattern = pattern.chars().collect::<Vec<char>>();
+            str::trim_end_matches::<&[char]>(
+                str::trim_start_matches::<&[char]>(input, pattern.as_ref()),
+                pattern.as_ref(),
+            )
         },
     };