feat: Implement multi-get for more backends

FuelLabs · Oct 29, 2024 · 833ade0 · 833ade0
1 parent 35b25d5
commit 833ade0
Show file tree

Hide file tree

Showing 5 changed files with 41 additions and 0 deletions.
diff --git a/crates/fuel-core/src/state/data_source.rs b/crates/fuel-core/src/state/data_source.rs
@@ -64,6 +64,14 @@ where
         self.data.get(key, column)
     }
 
+    fn get_batch<'a>(
+        &'a self,
+        keys: BoxedIter<'a, Vec<u8>>,
+        column: Self::Column,
+    ) -> BoxedIter<'a, StorageResult<Option<Value>>> {
+        self.data.get_batch(keys, column)
+    }
+
     fn read(
         &self,
         key: &[u8],

diff --git a/crates/fuel-core/src/state/generic_database.rs b/crates/fuel-core/src/state/generic_database.rs
@@ -134,6 +134,14 @@ where
         KeyValueInspect::get(&self.storage, key, column)
     }
 
+    fn get_batch<'a>(
+        &'a self,
+        keys: BoxedIter<'a, Vec<u8>>,
+        column: Self::Column,
+    ) -> BoxedIter<'a, StorageResult<Option<Value>>> {
+        KeyValueInspect::get_batch(&self.storage, keys, column)
+    }
+
     fn read(
         &self,
         key: &[u8],

diff --git a/crates/fuel-core/src/state/iterable_key_value_view.rs b/crates/fuel-core/src/state/iterable_key_value_view.rs
@@ -63,6 +63,14 @@ where
         self.0.get(key, column)
     }
 
+    fn get_batch<'a>(
+        &'a self,
+        keys: BoxedIter<'a, Vec<u8>>,
+        column: Self::Column,
+    ) -> BoxedIter<'a, StorageResult<Option<Value>>> {
+        self.0.get_batch(keys, column)
+    }
+
     fn read(
         &self,
         key: &[u8],

diff --git a/crates/fuel-core/src/state/key_value_view.rs b/crates/fuel-core/src/state/key_value_view.rs
@@ -1,4 +1,5 @@
 use fuel_core_storage::{
+    iter::BoxedIter,
     kv_store::{
         KeyValueInspect,
         StorageColumn,
@@ -50,6 +51,14 @@ where
         self.0.get(key, column)
     }
 
+    fn get_batch<'a>(
+        &'a self,
+        keys: BoxedIter<'a, Vec<u8>>,
+        column: Self::Column,
+    ) -> BoxedIter<'a, StorageResult<Option<Value>>> {
+        self.0.get_batch(keys, column)
+    }
+
     fn read(
         &self,
         key: &[u8],

diff --git a/crates/storage/src/structured_storage.rs b/crates/storage/src/structured_storage.rs
@@ -146,6 +146,14 @@ where
         self.inner.get(key, column)
     }
 
+    fn get_batch<'a>(
+        &'a self,
+        keys: BoxedIter<'a, Vec<u8>>,
+        column: Self::Column,
+    ) -> BoxedIter<'a, StorageResult<Option<Value>>> {
+        self.inner.get_batch(keys, column)
+    }
+
     fn read(
         &self,
         key: &[u8],