This is an automated email from the ASF dual-hosted git repository.
comphead pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/datafusion.git
The following commit(s) were added to refs/heads/main by this push:
new d5428b21d6 docs: Fix create_udf example (#14405)
d5428b21d6 is described below
commit d5428b21d6486e5b7db525d314ee9119a512c397
Author: nuno-faria <[email protected]>
AuthorDate: Sun Feb 2 18:43:52 2025 +0000
docs: Fix create_udf example (#14405)
---
docs/source/library-user-guide/adding-udfs.md | 27 ++++++++++++++-------------
1 file changed, 14 insertions(+), 13 deletions(-)
diff --git a/docs/source/library-user-guide/adding-udfs.md
b/docs/source/library-user-guide/adding-udfs.md
index 2044cfaa70..a920297697 100644
--- a/docs/source/library-user-guide/adding-udfs.md
+++ b/docs/source/library-user-guide/adding-udfs.md
@@ -123,13 +123,13 @@ There is a an older, more concise, but also more limited
API [`create_udf`] avai
use std::sync::Arc;
use datafusion::arrow::array::{ArrayRef, Int64Array};
-use datafusion::common::Result;
use datafusion::common::cast::as_int64_array;
-use datafusion::physical_plan::functions::columnar_values_to_array;
+use datafusion::common::Result;
+use datafusion::logical_expr::ColumnarValue;
-pub fn add_one(args: &[ColumnarValue]) -> Result<ArrayRef> {
+pub fn add_one(args: &[ColumnarValue]) -> Result<ColumnarValue> {
// Error handling omitted for brevity
- let args = columnar_values_to_array(args)?;
+ let args = ColumnarValue::values_to_arrays(args)?;
let i64s = as_int64_array(&args[0])?;
let new_array = i64s
@@ -137,7 +137,7 @@ pub fn add_one(args: &[ColumnarValue]) -> Result<ArrayRef> {
.map(|array_elem| array_elem.map(|value| value + 1))
.collect::<Int64Array>();
- Ok(Arc::new(new_array))
+ Ok(ColumnarValue::from(Arc::new(new_array) as ArrayRef))
}
```
@@ -146,10 +146,11 @@ This "works" in isolation, i.e. if you have a slice of
`ArrayRef`s, you can call
```rust
let input = vec![Some(1), None, Some(3)];
-let input = Arc::new(Int64Array::from(input)) as ArrayRef;
+let input = ColumnarValue::from(Arc::new(Int64Array::from(input)) as ArrayRef);
-let result = add_one( & [input]).unwrap();
-let result = result.as_any().downcast_ref::<Int64Array>().unwrap();
+let result = add_one(&[input]).unwrap();
+let binding = result.into_array(1).unwrap();
+let result = binding.as_any().downcast_ref::<Int64Array>().unwrap();
assert_eq!(result, &Int64Array::from(vec![Some(2), None, Some(4)]));
```
@@ -169,11 +170,11 @@ use datafusion::arrow::datatypes::DataType;
use std::sync::Arc;
let udf = create_udf(
-"add_one",
-vec![DataType::Int64],
-Arc::new(DataType::Int64),
-Volatility::Immutable,
-Arc::new(add_one),
+ "add_one",
+ vec![DataType::Int64],
+ DataType::Int64,
+ Volatility::Immutable,
+ Arc::new(add_one),
);
```
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
