This is an automated email from the ASF dual-hosted git repository.
richox pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/auron.git
The following commit(s) were added to refs/heads/master by this push:
new 4f5854b3 [AURON #1507] Enable AuronAdaptor#getAuronUDFWrapperContext
for spark (#1508)
4f5854b3 is described below
commit 4f5854b3696af6627b08a6563cd0a7dfd192bbd6
Author: zhangmang <[email protected]>
AuthorDate: Fri Oct 24 17:32:58 2025 +0800
[AURON #1507] Enable AuronAdaptor#getAuronUDFWrapperContext for spark
(#1508)
---
native-engine/datafusion-ext-exprs/src/spark_udf_wrapper.rs | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/native-engine/datafusion-ext-exprs/src/spark_udf_wrapper.rs
b/native-engine/datafusion-ext-exprs/src/spark_udf_wrapper.rs
index 3c1a94fd..e3c43e6e 100644
--- a/native-engine/datafusion-ext-exprs/src/spark_udf_wrapper.rs
+++ b/native-engine/datafusion-ext-exprs/src/spark_udf_wrapper.rs
@@ -27,7 +27,7 @@ use arrow::{
record_batch::{RecordBatch, RecordBatchOptions},
};
use auron_jni_bridge::{
- is_task_running, jni_call, jni_new_direct_byte_buffer, jni_new_global_ref,
jni_new_object,
+ is_task_running, jni_call, jni_call_static, jni_new_direct_byte_buffer,
jni_new_global_ref,
};
use datafusion::{
error::Result,
@@ -94,7 +94,7 @@ impl SparkUDFWrapperExpr {
.get_or_try_init(|| {
let serialized_buf =
jni_new_direct_byte_buffer!(&self.serialized)?;
let jcontext_local =
-
jni_new_object!(SparkAuronUDFWrapperContext(serialized_buf.as_obj()))?;
+
jni_call_static!(JniBridge.getAuronUDFWrapperContext(serialized_buf.as_obj())
-> JObject)?;
jni_new_global_ref!(jcontext_local.as_obj())
})
.cloned()
