spinoff #1251

Author: Devdutt Shenoi

src/event/format/mod.rs

@@ -187,7 +187,8 @@ pub trait EventFormat: Sized {
         rb = replace_columns(
             rb.schema(),
             &rb,
-            &[(0, Arc::new(get_timestamp_array(p_timestamp, rb.num_rows())))],
+            &[0],
+            &[Arc::new(get_timestamp_array(p_timestamp, rb.num_rows()))],
         );
 
         Ok(rb)

src/utils/arrow/mod.rs

@@ -61,19 +61,21 @@ use serde_json::{Map, Value};
 ///
 /// * `schema` - The schema of the record batch.
 /// * `batch` - The record batch to modify.
-/// * `indexed_arrays` - A list of indexes and arrays to replace the columns indexed with.
+/// * `indexes` - The indexes of the columns to replace.
+/// * `arrays` - The new arrays to replace the columns with.
 ///
 /// # Returns
 ///
 /// The modified record batch with the columns replaced.
 pub fn replace_columns(
     schema: Arc<Schema>,
     batch: &RecordBatch,
-    indexed_arrays: &[(usize, Arc<dyn Array + 'static>)],
+    indexes: &[usize],
+    arrays: &[Arc<dyn Array + 'static>],
 ) -> RecordBatch {
     let mut batch_arrays = batch.columns().iter().map(Arc::clone).collect_vec();
-    for (index, arr) in indexed_arrays {
-        batch_arrays[*index] = Arc::clone(arr);
+    for (&index, arr) in indexes.iter().zip(arrays.iter()) {
+        batch_arrays[index] = Arc::clone(arr);
     }
     RecordBatch::try_new(schema, batch_arrays).unwrap()
 }
@@ -176,7 +178,7 @@ mod tests {
 
         let arr: Arc<dyn Array + 'static> = Arc::new(Int32Array::from_value(0, 3));
 
-        let new_rb = replace_columns(schema_ref.clone(), &rb, &[(2, arr)]);
+        let new_rb = replace_columns(schema_ref.clone(), &rb, &[2], &[arr]);
 
         assert_eq!(new_rb.schema(), schema_ref);
         assert_eq!(new_rb.num_columns(), 3);