[AURON #1942] Implement columnar aggregate buffers

native-engine/auron-jni-bridge/src/jni_bridge.rs

@@ -1254,10 +1254,10 @@ pub struct SparkUDAFWrapperContext<'a> {
     pub method_merge_ret: ReturnType,
     pub method_eval: JMethodID,
     pub method_eval_ret: ReturnType,
-    pub method_serializeRows: JMethodID,
-    pub method_serializeRows_ret: ReturnType,
-    pub method_deserializeRows: JMethodID,
-    pub method_deserializeRows_ret: ReturnType,
+    pub method_exportRows: JMethodID,
+    pub method_exportRows_ret: ReturnType,
+    pub method_importRows: JMethodID,
+    pub method_importRows_ret: ReturnType,
     pub method_spill: JMethodID,
     pub method_spill_ret: ReturnType,
     pub method_unspill: JMethodID,
@@ -1307,18 +1307,18 @@ impl<'a> SparkUDAFWrapperContext<'a> {
                 "(Lorg/apache/spark/sql/auron/BufferRowsColumn;[IJ)V",
             )?,
             method_eval_ret: ReturnType::Primitive(Primitive::Void),
-            method_serializeRows: env.get_method_id(
+            method_exportRows: env.get_method_id(
                 class,
-                "serializeRows",
-                "(Lorg/apache/spark/sql/auron/BufferRowsColumn;[I)[B",
+                "exportRows",
+                "(Lorg/apache/spark/sql/auron/BufferRowsColumn;[IJ)V",
             )?,
-            method_serializeRows_ret: ReturnType::Array,
-            method_deserializeRows: env.get_method_id(
+            method_exportRows_ret: ReturnType::Primitive(Primitive::Void),
+            method_importRows: env.get_method_id(
                 class,
-                "deserializeRows",
-                "(Ljava/nio/ByteBuffer;)Lorg/apache/spark/sql/auron/BufferRowsColumn;",
+                "importRows",
+                "(J)Lorg/apache/spark/sql/auron/BufferRowsColumn;",
             )?,
-            method_deserializeRows_ret: ReturnType::Object,
+            method_importRows_ret: ReturnType::Object,
             method_spill: env.get_method_id(
                 class,
                 "spill",

native-engine/datafusion-ext-exprs/src/spark_partition_id.rs

@@ -146,7 +146,7 @@ mod tests {
                     assert_eq!(int_arr.value(i), 5);
                 }
             }
-            _ => panic!("Expected Array result"),
+            _ => unreachable!("Expected Array result"),
         }
     }
 
@@ -172,7 +172,7 @@ mod tests {
                         assert_eq!(int_arr.value(i), partition_id as i32);
                     }
                 }
-                _ => panic!("Expected Array result"),
+                _ => unreachable!("Expected Array result"),
             }
         }
     }