feat: add possibility to not simplify UDTF args

Author: askalt

datafusion/common/src/config.rs

@@ -257,6 +257,9 @@ config_namespace! {
         /// parallel file scanning. Setting this to `true` ensures that newlines in values are
         /// parsed successfully, which may reduce performance.
         pub newlines_in_values: bool, default = false
+
+        /// Specifies whether UDTF arguments expressions must be simplified.
+        pub simplify_udtf_args: bool, default = true
     }
 }
 

datafusion/core/src/execution/session_state.rs

@@ -1844,14 +1844,17 @@ impl ContextProvider for SessionContextProvider<'_> {
             );
         let simplifier = ExprSimplifier::new(simplify_context);
         let schema = DFSchema::empty();
-        let args = args
-            .into_iter()
-            .map(|arg| {
-                simplifier
-                    .coerce(arg, &schema)
-                    .and_then(|e| simplifier.simplify(e))
-            })
-            .collect::<datafusion_common::Result<Vec<_>>>()?;
+        let args = if !self.state.config.options().catalog.simplify_udtf_args {
+            args
+        } else {
+            args.into_iter()
+                .map(|arg| {
+                    simplifier
+                        .coerce(arg, &schema)
+                        .and_then(|e| simplifier.simplify(e))
+                })
+                .collect::<datafusion_common::Result<Vec<_>>>()?
+        };
         let provider = tbl_func.create_table_provider_with_args(TableFunctionArgs {
             args: &args,
             session: self.state,