feat: use AsyncScalarUDFImpl instead of ScalarUDFImpl

Author: Sl1mb0

host/src/lib.rs

@@ -496,7 +496,11 @@ impl AsyncScalarUDFImpl for WasmScalarUdf {
         let columnar_value: ColumnarValue = return_type.try_into()?;
         match columnar_value {
             ColumnarValue::Array(v) => Ok(v),
+<<<<<<< HEAD
             ColumnarValue::Scalar(v) => v.to_array_of_size(args.number_rows as usize),
+=======
+            ColumnarValue::Scalar(v) => v.to_array_of_size(1),
+>>>>>>> c980278 (feat: use AsyncScalarUDFImpl instead of ScalarUDFImpl)
         }
     }
 }

host/tests/integration_tests/python/runtime/http.rs

@@ -17,10 +17,7 @@ use tokio::runtime::Handle;
 use wasmtime_wasi_http::types::DEFAULT_FORBIDDEN_HEADERS;
 use wiremock::{Mock, MockServer, ResponseTemplate, matchers};
 
-use crate::integration_tests::{
-    python::test_utils::{python_component, python_scalar_udf},
-    test_utils::ColumnarValueExt,
-};
+use crate::integration_tests::python::test_utils::{python_component, python_scalar_udf};
 
 #[tokio::test(flavor = "multi_thread")]
 async fn test_requests_simple() {

host/tests/integration_tests/test_utils.rs

@@ -2,6 +2,7 @@ use arrow::array::ArrayRef;
 use datafusion_expr::ColumnarValue;
 
 /// Extension trait for [`ColumnarValue`] for easier testing.
+#[expect(dead_code)]
 pub(crate) trait ColumnarValueExt {
     /// Extracts [`ColumnarValue::Array`] variant.
     ///

query/src/lib.rs

@@ -2,9 +2,11 @@
 #![allow(unused_crate_dependencies)]
 
 use std::collections::HashMap;
+use std::sync::Arc;
 
 use datafusion::execution::TaskContext;
 use datafusion_common::{DataFusionError, Result as DataFusionResult};
+use datafusion_expr::async_udf::AsyncScalarUDFImpl;
 use datafusion_sql::parser::{DFParserBuilder, Statement};
 use sqlparser::ast::{CreateFunctionBody, Expr, Statement as SqlStatement, Value};
 use sqlparser::dialect::dialect_from_str;
@@ -31,7 +33,7 @@ pub struct Lang<'a> {
 #[derive(Debug)]
 pub struct ParsedQuery {
     /// Extracted UDFs from the query
-    pub udfs: Vec<WasmScalarUdf>,
+    pub udfs: Vec<Arc<dyn AsyncScalarUDFImpl>>,
     /// SQL query string with UDF definitions removed
     pub sql: String,
 }

query/tests/integration.rs

@@ -20,6 +20,7 @@ use datafusion_udf_wasm_query::{
 use tokio::runtime::Handle;
 
 mod integration_tests;
+use datafusion_expr::async_udf::AsyncScalarUDF;
 
 use crate::integration_tests::python::test_utils::python_component;