Skip to content

Commit

Permalink
chore: Expose StripeFactory::read_next_stripe and Stripe::stream_map (#…
Browse files Browse the repository at this point in the history
…10)

Signed-off-by: Xuanwo <[email protected]>
  • Loading branch information
Xuanwo authored Oct 28, 2024
1 parent cf87633 commit dfb1ede
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 4 deletions.
3 changes: 2 additions & 1 deletion src/async_arrow_reader.rs
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,8 @@ impl<R: AsyncChunkReader + 'static> StripeFactory<R> {
.await
}

async fn read_next_stripe(mut self) -> Result<(Self, Option<Stripe>)> {
/// Read the next stripe from the file.
pub async fn read_next_stripe(mut self) -> Result<(Self, Option<Stripe>)> {
let info = self
.inner
.file_metadata
Expand Down
2 changes: 1 addition & 1 deletion src/compression.rs
Original file line number Diff line number Diff line change
Expand Up @@ -284,7 +284,7 @@ impl FallibleStreamingIterator for DecompressorIter {
}

/// A [`Read`]er fulfilling the ORC specification of reading compressed data.
pub(crate) struct Decompressor {
pub struct Decompressor {
decompressor: DecompressorIter,
offset: usize,
is_first: bool,
Expand Down
5 changes: 3 additions & 2 deletions src/stripe.rs
Original file line number Diff line number Diff line change
Expand Up @@ -235,7 +235,8 @@ impl Stripe {
self.number_of_rows
}

pub(crate) fn stream_map(&self) -> &StreamMap {
/// Fetch the stream map
pub fn stream_map(&self) -> &StreamMap {
&self.stream_map
}

Expand All @@ -249,7 +250,7 @@ impl Stripe {
}

#[derive(Debug)]
pub(crate) struct StreamMap {
pub struct StreamMap {
pub inner: HashMap<(u32, Kind), Bytes>,
pub compression: Option<Compression>,
}
Expand Down

0 comments on commit dfb1ede

Please sign in to comment.