Implement PrimitiveValueDecoder::decode for ByteRleDecoder

datafusion-contrib · Sep 28, 2024 · ecffdf8 · ecffdf8
1 parent 63c7513
commit ecffdf8
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/encoding/byte.rs b/src/encoding/byte.rs
@@ -244,7 +244,19 @@ impl<R: Read> Iterator for ByteRleDecoder<R> {
     }
 }
 
-impl<R: Read> PrimitiveValueDecoder<i8> for ByteRleDecoder<R> {}
+impl<R: Read> PrimitiveValueDecoder<i8> for ByteRleDecoder<R> {
+    // TODO: can probably implement this better, just copying from iter for now
+    fn decode(&mut self, out: &mut [i8]) -> Result<()> {
+        for x in out.iter_mut() {
+            if self.index == self.leftovers.len() {
+                self.read_values()?;
+            }
+            *x = self.leftovers[self.index] as i8;
+            self.index += 1;
+        }
+        Ok(())
+    }
+}
 
 #[cfg(test)]
 mod tests {