diff --git a/src/async_arrow_reader.rs b/src/async_arrow_reader.rs index 94a0565..851a228 100644 --- a/src/async_arrow_reader.rs +++ b/src/async_arrow_reader.rs @@ -68,7 +68,7 @@ impl From> for StripeFactory { } } -struct StripeFactory { +pub struct StripeFactory { inner: Cursor, is_end: bool, } @@ -132,6 +132,11 @@ impl ArrowStreamReader { } } + /// Extracts the inner `StripeFactory` and `SchemaRef` from the `ArrowStreamReader`. + pub fn into_parts(self) -> (Option>>, SchemaRef) { + (self.factory, self.schema_ref) + } + pub fn schema(&self) -> SchemaRef { self.schema_ref.clone() }