Rework search document

Author: allevo

rust-toolchain

@@ -0,0 +1 @@
+1.88.0

src/index.rs

@@ -179,7 +179,25 @@ impl<DocId: DocumentId> ZeboIndex<DocId> {
 
         Ok(())
     }
+
+    pub fn get_all_pages(&self) -> Result<Vec<(u64, PageId)>> {
+        let mut buf = vec![0; (self.offset - 8 - 1) as usize];
+        self.index_file
+            .read_exact_at(&mut buf, 9)
+            .map_err(ZeboError::OperationError)?;
+
+        let expected_page_size = (self.offset - 8 - 1) / (8 + 8);
+
+        let mut pages = Vec::with_capacity(expected_page_size as usize);
+        for chunk in buf.chunks_exact(16) {
+            let page_id = u64::from_be_bytes(chunk[0..8].try_into().unwrap());
+            let starting_doc_id = u64::from_be_bytes(chunk[8..16].try_into().unwrap());
+            pages.push((starting_doc_id, PageId(page_id)));
+        }
+
+        Ok(pages)
+    }
 }
 
-#[derive(Debug)]
+#[derive(Debug, Clone, Copy)]
 pub struct ProbableIndex(pub u64);

src/lib.rs

@@ -313,6 +313,10 @@ impl<const MAX_DOC_PER_PAGE: u32, const PAGE_SIZE: u64, DocId: DocumentId>
         Ok(total_count)
     }
 
+    pub fn get_all_pages(&self) -> Result<Vec<(u64, PageId)>> {
+        self.index.get_all_pages()
+    }
+
     /// Returns information related the this Zebo instance
     pub fn get_info(&self) -> Result<ZeboInfo> {
         let mut page_headers = Vec::with_capacity(self.next_page_id as usize);
@@ -632,6 +636,7 @@ impl DocumentId for u64 {
 
 #[cfg(test)]
 mod tests {
+
     use super::*;
 
     pub fn prepare_test_dir() -> PathBuf {