From 60c9d1037f012e76cf8f8c11d82b065b1f97269f Mon Sep 17 00:00:00 2001
From: Jan Tuomi <jan@jantuomi.fi>
Date: Sat, 11 Jan 2025 23:16:06 +0200
Subject: Add range_by, refactor reading from file

---
 log_db/src/memtable_primary.rs | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

(limited to 'log_db/src/memtable_primary.rs')

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()
+    }
 }
-- 
cgit v1.3