From cdcf0eed9a84703003d600baf2c44e51c7050bff Mon Sep 17 00:00:00 2001 From: irenjj Date: Mon, 17 Feb 2025 20:26:13 +0800 Subject: [PATCH] chore: Migrate Encoding functions to invoke_with_args --- datafusion/functions/src/encoding/inner.rs | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/datafusion/functions/src/encoding/inner.rs b/datafusion/functions/src/encoding/inner.rs index 68a6d1006052..51e8c6968866 100644 --- a/datafusion/functions/src/encoding/inner.rs +++ b/datafusion/functions/src/encoding/inner.rs @@ -103,12 +103,11 @@ impl ScalarUDFImpl for EncodeFunc { }) } - fn invoke_batch( + fn invoke_with_args( &self, - args: &[ColumnarValue], - _number_rows: usize, + args: datafusion_expr::ScalarFunctionArgs, ) -> Result { - encode(args) + encode(&args.args) } fn coerce_types(&self, arg_types: &[DataType]) -> Result> { @@ -183,12 +182,11 @@ impl ScalarUDFImpl for DecodeFunc { Ok(arg_types[0].to_owned()) } - fn invoke_batch( + fn invoke_with_args( &self, - args: &[ColumnarValue], - _number_rows: usize, + args: datafusion_expr::ScalarFunctionArgs, ) -> Result { - decode(args) + decode(&args.args) } fn coerce_types(&self, arg_types: &[DataType]) -> Result> {