feat: adding flush to dustdata struct

Author: peeeuzin

src/dustdata.rs

@@ -205,9 +205,19 @@ impl DustData {
         self.lsm.contains(key)
     }
 
+    /// List all keys.
+    /// # Returns
+    /// - `Result<Vec<String>>` a vector of all keys.
     pub fn list_keys(&self) -> Result<Vec<String>> {
         Ok(self.lsm.list_keys())
     }
+
+    /// Flush all data to disk.
+    /// # Returns
+    /// - `Result<()>` if successful returns nothing.
+    pub fn flush(&mut self) -> Result<()> {
+        self.lsm.flush()
+    }
 }
 
 #[cfg(test)]

src/lib.rs

@@ -107,7 +107,7 @@ mod dustdata_tests {
         .unwrap();
 
         // flush the sstable
-        dd.lsm.flush();
+        dd.flush().unwrap();
 
         let ls = dd.list_keys().unwrap();
 

src/storage/lsm/mod.rs

@@ -128,7 +128,7 @@ impl Lsm {
         self.bloom_filter.lock().unwrap().insert(key);
 
         if self.memtable_size >= self.lsm_config.flush_threshold {
-            self.flush();
+            self.flush().unwrap();
         }
 
         Ok(())
@@ -204,7 +204,7 @@ impl Lsm {
         Ok(())
     }
 
-    pub fn flush(&mut self) {
+    pub fn flush(&mut self) -> Result<()> {
         if self.lsm_config.verbose {
             logs!("Flushing memtable to disk...");
         }
@@ -237,6 +237,8 @@ impl Lsm {
 
         self.memtable.lock().unwrap().clear();
         self.memtable_size = 0;
+
+        Ok(())
     }
 
     pub fn get_memtable(&self) -> BTreeMap<String, bson::Bson> {