Don't validate array flatbuffer messages twice (#418)

spiraldb · Jun 26, 2024 · 3d915fb · 3d915fb
1 parent 1d5a68b
commit 3d915fb
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/vortex-ipc/src/message_reader.rs b/vortex-ipc/src/message_reader.rs
@@ -189,14 +189,11 @@ impl<R: VortexRead> MessageReader<R> {
             dtype,
             flatbuffer,
             |flatbuffer| {
-                root::<fb::Message>(flatbuffer)
-                    .map_err(VortexError::from)
-                    .map(|msg| msg.header_as_chunk().unwrap())
-                    .and_then(|chunk| {
-                        chunk
-                            .array()
-                            .ok_or_else(|| vortex_err!("Chunk missing Array"))
-                    })
+                unsafe { root_unchecked::<fb::Message>(flatbuffer) }
+                    .header_as_chunk()
+                    .unwrap()
+                    .array()
+                    .ok_or_else(|| vortex_err!("Chunk missing Array"))
             },
             buffers,
         )?;