add more

Signed-off-by: Baris Palaska <barispalaska@gmail.com>

Author: palaska

vortex-array/public-api.lock

@@ -22194,7 +22194,7 @@ pub fn vortex_array::Array<vortex_array::arrays::Struct>::with_column(&self, imp
 
 impl vortex_array::Array<vortex_array::arrays::Struct>
 
-pub fn vortex_array::Array<vortex_array::arrays::Struct>::into_record_batch_with_schema(self, impl core::convert::AsRef<arrow_schema::schema::Schema>) -> vortex_error::VortexResult<arrow_array::record_batch::RecordBatch>
+pub fn vortex_array::Array<vortex_array::arrays::Struct>::into_record_batch_with_schema(self, impl core::convert::AsRef<arrow_schema::schema::Schema>, &vortex_session::VortexSession) -> vortex_error::VortexResult<arrow_array::record_batch::RecordBatch>
 
 impl vortex_array::Array<vortex_array::arrays::VarBin>
 

vortex-array/src/arrow/record_batch.rs

@@ -6,8 +6,8 @@ use arrow_array::cast::AsArray;
 use arrow_schema::DataType;
 use arrow_schema::Schema;
 use vortex_error::VortexResult;
+use vortex_session::VortexSession;
 
-use crate::LEGACY_SESSION;
 use crate::VortexSessionExecute;
 use crate::array::IntoArray;
 use crate::arrays::StructArray;
@@ -17,11 +17,12 @@ impl StructArray {
     pub fn into_record_batch_with_schema(
         self,
         schema: impl AsRef<Schema>,
+        session: &VortexSession,
     ) -> VortexResult<RecordBatch> {
         let data_type = DataType::Struct(schema.as_ref().fields.clone());
         let array_ref = self
             .into_array()
-            .execute_arrow(Some(&data_type), &mut LEGACY_SESSION.create_execution_ctx())?;
+            .execute_arrow(Some(&data_type), &mut session.create_execution_ctx())?;
         Ok(RecordBatch::from(array_ref.as_struct()))
     }
 }
@@ -35,6 +36,7 @@ mod tests {
     use arrow_schema::FieldRef;
     use arrow_schema::Schema;
 
+    use crate::LEGACY_SESSION;
     use crate::arrow::record_batch::StructArray;
     use crate::builders::ArrayBuilder;
     use crate::builders::ListBuilder;
@@ -69,7 +71,9 @@ mod tests {
             DataType::LargeListView(FieldRef::new(Field::new_list_field(DataType::Int32, false))),
             true,
         )]));
-        let rb = array.into_record_batch_with_schema(arrow_schema).unwrap();
+        let rb = array
+            .into_record_batch_with_schema(arrow_schema, &LEGACY_SESSION)
+            .unwrap();
 
         let xs = rb.column(0);
         assert_eq!(