fix: Read arguably malformed Parquet files from go

This ports a change made upstream in apache/arrow-rs#7052
pola-rs · Feb 6, 2025 · e023d14 · e023d14
1 parent 185ee3b
commit e023d14
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/thrift/protocol/compact.rs b/src/thrift/protocol/compact.rs
@@ -199,7 +199,7 @@ where
                 let b = self.read_byte()?;
                 match b {
                     0x01 => Ok(true),
-                    0x02 => Ok(false),
+                    0x00 | 0x02 => Ok(false),
                     unkn => Err(Error::Protocol(ProtocolError {
                         kind: ProtocolErrorKind::InvalidData,
                         message: format!("cannot convert {} into bool", unkn),

diff --git a/src/thrift/protocol/compact_stream.rs b/src/thrift/protocol/compact_stream.rs
@@ -193,7 +193,7 @@ impl<R: VarIntAsyncReader + AsyncRead + Unpin + Send> TInputStreamProtocol
                 let b = self.read_byte().await?;
                 match b {
                     0x01 => Ok(true),
-                    0x02 => Ok(false),
+                    0x00 | 0x02 => Ok(false),
                     unkn => Err(Error::Protocol(ProtocolError {
                         kind: ProtocolErrorKind::InvalidData,
                         message: format!("cannot convert {} into bool", unkn),