Add range_by, refactor reading from file

1 files changed, 8 insertions, 1 deletions

diff --git a/log_db/src/memtable_primary.rs b/log_db/src/memtable_primary.rs
index 85648d4..e2921a6 100644
--- a/log_db/src/memtable_primary.rs
+++ b/log_db/src/memtable_primary.rs
@@ -1,4 +1,4 @@
-use super::common::*;
+use super::*;
 use std::collections::BTreeMap;
 
 pub struct PrimaryMemtable {
@@ -30,4 +30,11 @@ impl PrimaryMemtable {
     pub fn remove(&mut self, key: &IndexableValue) -> Option<LogKey> {
         self.records.remove(key)
     }
+
+    pub fn range<B: RangeBounds<IndexableValue>>(&self, range: B) -> Vec<LogKey> {
+        self.records
+            .range(range)
+            .map(|(_, log_key)| log_key.clone())
+            .collect()
+    }
 }