modified into_scalar_udf method

Author: goldmedal

datafusion-examples/examples/async_udf.rs

@@ -24,7 +24,6 @@ use datafusion::common::internal_err;
 use datafusion::common::types::{logical_int64, logical_string};
 use datafusion::common::utils::take_function_args;
 use datafusion::config::ConfigOptions;
-use datafusion::execution::{FunctionRegistry, SessionStateBuilder};
 use datafusion::logical_expr::async_udf::{
     AsyncScalarFunctionArgs, AsyncScalarUDF, AsyncScalarUDFImpl,
 };
@@ -40,15 +39,14 @@ use std::sync::Arc;
 
 #[tokio::main]
 async fn main() -> Result<()> {
-    let mut state = SessionStateBuilder::new().build();
+    let ctx: SessionContext = SessionContext::new();
 
     let async_upper = AsyncUpper::new();
     let udf = AsyncScalarUDF::new(Arc::new(async_upper));
-    state.register_udf(udf.into_scalar_udf())?;
+    ctx.register_udf(udf.into_scalar_udf());
     let async_equal = AsyncEqual::new();
     let udf = AsyncScalarUDF::new(Arc::new(async_equal));
-    state.register_udf(udf.into_scalar_udf())?;
-    let ctx = SessionContext::new_with_state(state);
+    ctx.register_udf(udf.into_scalar_udf());
     ctx.register_batch("animal", animal()?)?;
 
     // use Async UDF in the projection

datafusion/expr/src/async_udf.rs

@@ -100,8 +100,8 @@ impl AsyncScalarUDF {
 
     /// Turn this AsyncUDF into a ScalarUDF, suitable for
     /// registering in the context
-    pub fn into_scalar_udf(self) -> Arc<ScalarUDF> {
-        Arc::new(ScalarUDF::new_from_impl(self))
+    pub fn into_scalar_udf(self) -> ScalarUDF {
+        ScalarUDF::new_from_impl(self)
     }
 
     /// Invoke the function asynchronously with the async arguments

datafusion/physical-plan/src/async_func.rs

@@ -116,11 +116,11 @@ impl DisplayAs for AsyncFuncExec {
         let exprs = expr.join(", ");
         match t {
             DisplayFormatType::Default | DisplayFormatType::Verbose => {
-                write!(f, "AsyncFuncExec: async_expr=[{}]", exprs)
+                write!(f, "AsyncFuncExec: async_expr=[{exprs}]")
             }
             DisplayFormatType::TreeRender => {
                 writeln!(f, "format=async_expr")?;
-                writeln!(f, "async_expr={}", exprs)?;
+                writeln!(f, "async_expr={exprs}")?;
                 Ok(())
             }
         }