shuffle

spiraldb · Oct 17, 2024 · 0b4e10b · 0b4e10b
1 parent 4132d56
commit 0b4e10b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/vortex-serde/src/layouts/read/footer.rs b/vortex-serde/src/layouts/read/footer.rs
@@ -152,7 +152,7 @@ impl FooterReader {
         }
 
         let ps = root::<vortex_flatbuffers::footer::Postscript>(
-            &buf[magic_bytes_loc - 32..magic_bytes_loc],
+            &buf[magic_bytes_loc - (FILE_POSTSCRIPT_SIZE - MAGIC_BYTES.len())..magic_bytes_loc],
         )?;
 
         Ok(Footer {

diff --git a/vortex-serde/src/layouts/read/layouts/flat.rs b/vortex-serde/src/layouts/read/layouts/flat.rs
@@ -9,7 +9,7 @@ use vortex_flatbuffers::footer;
 
 use crate::layouts::read::cache::RelativeLayoutCache;
 use crate::layouts::{
-    LayoutDeserializer, LayoutId, LayoutReader, LayoutSpec, Messages, ReadResult, Scan,
+    LayoutDeserializer, LayoutId, LayoutReader, LayoutSpec, Message, ReadResult, Scan,
 };
 use crate::message_reader::ArrayBufferReader;
 use crate::stream_writer::ByteRange;
@@ -92,8 +92,8 @@ impl FlatLayout {
         }
     }
 
-    fn own_message(&self) -> Messages {
-        vec![(self.cache.absolute_id(&[]), self.range)]
+    fn own_message(&self) -> Message {
+        (self.cache.absolute_id(&[]), self.range)
     }
 
     fn array_from_bytes(&self, mut buf: Bytes) -> VortexResult<Array> {
@@ -128,7 +128,7 @@ impl LayoutReader for FlatLayout {
             self.cached_array = Some(self.array_from_bytes(buf)?);
             self.read_next()
         } else {
-            Ok(Some(ReadResult::ReadMore(self.own_message())))
+            Ok(Some(ReadResult::ReadMore(vec![self.own_message()])))
         }
     }
 }
diff --git a/vortex-serde/src/layouts/read/mod.rs b/vortex-serde/src/layouts/read/mod.rs
@@ -45,11 +45,11 @@ pub struct Scan {
 /// Unique identifier for a message within a layout
 pub type LayoutPartId = u16;
 pub type MessageId = Vec<LayoutPartId>;
-pub type Messages = Vec<(MessageId, ByteRange)>;
+pub type Message = (MessageId, ByteRange);
 
 #[derive(Debug)]
 pub enum ReadResult {
-    ReadMore(Messages),
+    ReadMore(Vec<Message>),
     Batch(Array),
 }