Introduce many SingleStore wrappers

src/store/mod.rs

@@ -1,10 +1,16 @@
+mod docs_words;
 mod documents_fields;
+mod main;
+mod postings_lists;
 mod synonyms;
-mod words;
+mod updates;
 
+pub use self::docs_words::DocsWords;
 pub use self::documents_fields::{DocumentsFields, DocumentFieldsIter};
+pub use self::main::Main;
+pub use self::postings_lists::PostingsLists;
 pub use self::synonyms::Synonyms;
-pub use self::words::Words;
+pub use self::updates::Updates;
 
 const NUMBER_OF_DOCUMENTS_KEY: &str = "number-of-documents";
 const RANKED_MAP_KEY:          &str = "ranked-map";
@@ -16,31 +22,41 @@ fn aligned_to(bytes: &[u8], align: usize) -> bool {
     (bytes as *const _ as *const () as usize) % align == 0
 }
 
-fn words_indexes_name(name: &str) -> String {
-    format!("{}-words-indexes", name)
-}
-
-fn synonyms_name(name: &str) -> String {
-    format!("{}-synonyms", name)
+fn postings_lists_name(name: &str) -> String {
+    format!("{}-postings-lists", name)
 }
 
 fn documents_fields_name(name: &str) -> String {
     format!("{}-documents-fields", name)
 }
 
-pub fn create(
-    env: &rkv::Rkv,
-    name: &str,
-) -> Result<(Words, Synonyms, DocumentsFields), rkv::StoreError>
-{
+fn synonyms_name(name: &str) -> String {
+    format!("{}-synonyms", name)
+}
+
+fn docs_words_name(name: &str) -> String {
+    format!("{}-docs-words", name)
+}
+
+fn updates_name(name: &str) -> String {
+    format!("{}-updates", name)
+}
+
+#[derive(Copy, Clone)]
+pub struct Index {
+    pub main: Main,
+    pub postings_lists: PostingsLists,
+    pub documents_fields: DocumentsFields,
+    pub synonyms: Synonyms,
+    pub docs_words: DocsWords,
+    pub updates: Updates,
+}
+
+pub fn create(env: &rkv::Rkv, name: &str) -> Result<Index, rkv::StoreError> {
     open_options(env, name, rkv::StoreOptions::create())
 }
 
-pub fn open(
-    env: &rkv::Rkv,
-    name: &str,
-) -> Result<(Words, Synonyms, DocumentsFields), rkv::StoreError>
-{
+pub fn open(env: &rkv::Rkv, name: &str) -> Result<Index, rkv::StoreError> {
     let mut options = rkv::StoreOptions::default();
     options.create = false;
     open_options(env, name, options)
@@ -50,23 +66,30 @@ fn open_options(
     env: &rkv::Rkv,
     name: &str,
     options: rkv::StoreOptions,
-) -> Result<(Words, Synonyms, DocumentsFields), rkv::StoreError>
+) -> Result<Index, rkv::StoreError>
 {
     // create all the database names
     let main_name = name;
-    let words_indexes_name = words_indexes_name(name);
-    let synonyms_name = synonyms_name(name);
+    let postings_lists_name = postings_lists_name(name);
     let documents_fields_name = documents_fields_name(name);
+    let synonyms_name = synonyms_name(name);
+    let docs_words_name = docs_words_name(name);
+    let updates_name = updates_name(name);
 
     // open all the database names
     let main = env.open_single(main_name, options)?;
-    let words_indexes = env.open_single(words_indexes_name.as_str(), options)?;
-    let synonyms = env.open_single(synonyms_name.as_str(), options)?;
+    let postings_lists = env.open_single(postings_lists_name.as_str(), options)?;
     let documents_fields = env.open_single(documents_fields_name.as_str(), options)?;
+    let synonyms = env.open_single(synonyms_name.as_str(), options)?;
+    let docs_words = env.open_single(docs_words_name.as_str(), options)?;
+    let updates = env.open_single(updates_name.as_str(), options)?;
 
-    let words = Words { main, words_indexes };
-    let synonyms = Synonyms { main, synonyms };
-    let documents_fields = DocumentsFields { documents_fields };
-
-    Ok((words, synonyms, documents_fields))
+    Ok(Index {
+        main: Main { main },
+        postings_lists: PostingsLists { postings_lists },
+        documents_fields: DocumentsFields { documents_fields },
+        synonyms: Synonyms { synonyms },
+        docs_words: DocsWords { docs_words },
+        updates: Updates { updates },
+    })
 }