Add inline

apache · Jul 20, 2022 · d4a689a · d4a689a
1 parent 82c9694
commit d4a689a
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/parquet/src/arrow/record_reader/definition_levels.rs b/parquet/src/arrow/record_reader/definition_levels.rs
@@ -152,13 +152,15 @@ enum MaybePacked {
 }
 
 impl MaybePacked {
+    #[inline]
     fn packed(&mut self) -> &mut PackedDecoder {
         match self {
             Self::Packed(d) => d,
             _ => panic!("expected packed"),
         }
     }
 
+    #[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(),