From d4a689ab6f2ed8417c043eb9a5f211f2845bf381 Mon Sep 17 00:00:00 2001 From: Raphael Taylor-Davies Date: Wed, 20 Jul 2022 18:22:31 -0400 Subject: [PATCH] Add inline --- parquet/src/arrow/record_reader/definition_levels.rs | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/parquet/src/arrow/record_reader/definition_levels.rs b/parquet/src/arrow/record_reader/definition_levels.rs index c2fb53cfd9f..5b80886b4f7 100644 --- a/parquet/src/arrow/record_reader/definition_levels.rs +++ b/parquet/src/arrow/record_reader/definition_levels.rs @@ -152,6 +152,7 @@ enum MaybePacked { } impl MaybePacked { + #[inline] fn packed(&mut self) -> &mut PackedDecoder { match self { Self::Packed(d) => d, @@ -159,6 +160,7 @@ impl MaybePacked { } } + #[inline] fn fallback(&mut self) -> &mut ColumnLevelDecoderImpl { match self { Self::Fallback(d) => d, @@ -261,6 +263,11 @@ struct PackedDecoder { } impl PackedDecoder { + #[inline] + fn data(&self) -> &[u8] { + self.data.as_ref().unwrap().data() + } + fn next_rle_block(&mut self) -> Result<()> { let indicator_value = self.decode_header()?; if indicator_value & 1 == 1 { @@ -269,7 +276,7 @@ impl PackedDecoder { self.packed_offset = 0; } else { self.rle_left = (indicator_value >> 1) as usize; - let byte = *self.data.as_ref().get(self.data_offset).ok_or_else(|| { + let byte = *self.data().get(self.data_offset).ok_or_else(|| { ParquetError::EOF( "unexpected end of file whilst decoding definition levels rle value" .into(),