Add keys() shortcut for simple keys to MultiIndex / UniqueIndex

CosmWasm · Jul 22, 2021 · 5325f00 · 5325f00
1 parent acd2c93
commit 5325f00
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/packages/storage-plus/src/indexes.rs b/packages/storage-plus/src/indexes.rs
@@ -185,6 +185,17 @@ where
         self.sub_prefix(K::SubPrefix::new())
             .range(store, min, max, order)
     }
+
+    pub fn keys<'c>(
+        &'c self,
+        store: &'c dyn Storage,
+        min: Option<Bound>,
+        max: Option<Bound>,
+        order: Order,
+    ) -> Box<dyn Iterator<Item = Vec<u8>> + 'c> {
+        self.sub_prefix(K::SubPrefix::new())
+            .keys(store, min, max, order)
+    }
 }
 
 #[derive(Deserialize, Serialize)]
@@ -300,4 +311,15 @@ where
         self.sub_prefix(K::SubPrefix::new())
             .range(store, min, max, order)
     }
+
+    pub fn keys<'c>(
+        &self,
+        store: &'c dyn Storage,
+        min: Option<Bound>,
+        max: Option<Bound>,
+        order: Order,
+    ) -> Box<dyn Iterator<Item = StdResult<Vec<u8>>> + 'c> {
+        self.sub_prefix(K::SubPrefix::new())
+            .keys(store, min, max, order)
+    }
 }