From d217bda12ed385a72ce65b3b73f4a680d15a9665 Mon Sep 17 00:00:00 2001
From: ardocrat <ardocrat@gri.mw>
Date: Sat, 16 May 2026 12:44:21 +0300
Subject: [PATCH] lmdb: store max 10k keys in the iterator

---
 store/src/lmdb.rs | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/store/src/lmdb.rs b/store/src/lmdb.rs
index 1c186a6a..6e3196a4 100644
--- a/store/src/lmdb.rs
+++ b/store/src/lmdb.rs
@@ -759,6 +759,8 @@ where
 	read: Arc<RoTxn<'static, WithoutTls>>,
 	keys: Vec<Vec<u8>>,
 	skip: usize,
+	total_keys: usize,
+	skip_keys: usize,
 	deserialize: F,
 	#[allow(dead_code)]
 	tx_counter: TxCounter,
@@ -777,11 +779,26 @@ where
 				return match (self.deserialize)(k, v) {
 					Ok(v) => {
 						self.skip += 1;
+						self.skip_keys += 1;
 						Some(v)
 					}
 					Err(_) => None,
 				};
 			}
+		} else if self.total_keys > self.skip_keys {
+			let keys = if let Ok(iter) = self.db.iter(&self.read) {
+				iter.move_between_keys()
+					.skip(self.skip_keys)
+					.take(10000)
+					.filter(|kv| kv.is_ok())
+					.map(|kv| kv.unwrap().0.to_vec())
+					.collect::<Vec<Vec<u8>>>()
+			} else {
+				vec![]
+			};
+			self.skip = 0;
+			self.keys = keys;
+			return self.next();
 		}
 		None
 	}
@@ -798,8 +815,10 @@ where
 		read: RoTxn<'static, WithoutTls>,
 		deserialize: F,
 	) -> DatabaseIterator<F, T> {
+		let total_keys = db.iter(&read).unwrap().count();
 		let keys = if let Ok(iter) = db.iter(&read) {
 			iter.move_between_keys()
+				.take(10000)
 				.filter(|kv| kv.is_ok())
 				.map(|kv| kv.unwrap().0.to_vec())
 				.collect::<Vec<Vec<u8>>>()
@@ -810,6 +829,8 @@ where
 			db,
 			read: Arc::new(read),
 			keys,
+			total_keys,
+			skip_keys: 0,
 			skip: 0,
 			deserialize,
 			tx_counter: TxCounter {