Просмотр исходного кода

blockchain: Clippy some iterators.

parazyd 4 лет назад
Родитель
Сommit
624e48775a

+ 2 - 2
src/blockchain2/blockstore.rs

@@ -69,8 +69,8 @@ impl BlockStore {
     /// Be careful as this will try to load everything in memory.
     /// Be careful as this will try to load everything in memory.
     pub fn get_all(&self) -> Result<Vec<Option<(blake3::Hash, Block)>>> {
     pub fn get_all(&self) -> Result<Vec<Option<(blake3::Hash, Block)>>> {
         let mut blocks = vec![];
         let mut blocks = vec![];
-        let mut iterator = self.0.into_iter().enumerate();
-        while let Some((_, r)) = iterator.next() {
+        let iterator = self.0.into_iter().enumerate();
+        for (_, r) in iterator {
             let (k, v) = r.unwrap();
             let (k, v) = r.unwrap();
             let hash_bytes: [u8; 32] = k.as_ref().try_into().unwrap();
             let hash_bytes: [u8; 32] = k.as_ref().try_into().unwrap();
             let block = deserialize(&v)?;
             let block = deserialize(&v)?;

+ 2 - 2
src/blockchain2/metadatastore.rs

@@ -35,8 +35,8 @@ impl StreamletMetadataStore {
     /// Be careful as this will try to lead everything in memory.
     /// Be careful as this will try to lead everything in memory.
     pub fn get_all(&self) -> Result<Vec<Option<(blake3::Hash, StreamletMetadata)>>> {
     pub fn get_all(&self) -> Result<Vec<Option<(blake3::Hash, StreamletMetadata)>>> {
         let mut metadata = vec![];
         let mut metadata = vec![];
-        let mut iterator = self.0.into_iter().enumerate();
-        while let Some((_, r)) = iterator.next() {
+        let iterator = self.0.into_iter().enumerate();
+        for (_, r) in iterator {
             let (k, v) = r.unwrap();
             let (k, v) = r.unwrap();
             let hash_bytes: [u8; 32] = k.as_ref().try_into().unwrap();
             let hash_bytes: [u8; 32] = k.as_ref().try_into().unwrap();
             let m = deserialize(&v)?;
             let m = deserialize(&v)?;

+ 8 - 3
src/blockchain2/nfstore.rs

@@ -28,12 +28,17 @@ impl NullifierStore {
         Ok(())
         Ok(())
     }
     }
 
 
+    /// Check whether given nullifier is in the database
+    pub fn contains(&self, nullifier: &Nullifier) -> Result<bool> {
+        Ok(self.0.contains_key(serialize(nullifier))?)
+    }
+
     /// Retrieve all nullifiers.
     /// Retrieve all nullifiers.
-    /// Be careful as this will try to load everything im memory.
+    /// Be careful as this will try to load everything in memory.
     pub fn get_all(&self) -> Result<Vec<Option<Nullifier>>> {
     pub fn get_all(&self) -> Result<Vec<Option<Nullifier>>> {
         let mut nfs = vec![];
         let mut nfs = vec![];
-        let mut iterator = self.0.into_iter().enumerate();
-        while let Some((_, r)) = iterator.next() {
+        let iterator = self.0.into_iter().enumerate();
+        for (_, r) in iterator {
             let (k, _) = r.unwrap();
             let (k, _) = r.unwrap();
             let nullifier = deserialize(&k)?;
             let nullifier = deserialize(&k)?;
             nfs.push(Some(nullifier))
             nfs.push(Some(nullifier))

+ 7 - 2
src/blockchain2/rootstore.rs

@@ -28,12 +28,17 @@ impl RootStore {
         Ok(())
         Ok(())
     }
     }
 
 
+    /// Check whether given root is in the database
+    pub fn contains(&self, root: &MerkleNode) -> Result<bool> {
+        Ok(self.0.contains_key(serialize(root))?)
+    }
+
     /// Retrieve all merkle roots.
     /// Retrieve all merkle roots.
     /// Be careful as this will try to load everything in memory.
     /// Be careful as this will try to load everything in memory.
     pub fn get_all(&self) -> Result<Vec<Option<MerkleNode>>> {
     pub fn get_all(&self) -> Result<Vec<Option<MerkleNode>>> {
         let mut roots = vec![];
         let mut roots = vec![];
-        let mut iterator = self.0.into_iter().enumerate();
-        while let Some((_, r)) = iterator.next() {
+        let iterator = self.0.into_iter().enumerate();
+        for (_, r) in iterator {
             let (k, _) = r.unwrap();
             let (k, _) = r.unwrap();
             let root = deserialize(&k)?;
             let root = deserialize(&k)?;
             roots.push(Some(root));
             roots.push(Some(root));

+ 2 - 2
src/blockchain2/txstore.rs

@@ -39,8 +39,8 @@ impl TxStore {
     /// Be careful as this will try to load everything in memory.
     /// Be careful as this will try to load everything in memory.
     pub fn get_all(&self) -> Result<Vec<Option<(blake3::Hash, Tx)>>> {
     pub fn get_all(&self) -> Result<Vec<Option<(blake3::Hash, Tx)>>> {
         let mut txs = vec![];
         let mut txs = vec![];
-        let mut iterator = self.0.into_iter().enumerate();
-        while let Some((_, r)) = iterator.next() {
+        let iterator = self.0.into_iter().enumerate();
+        for (_, r) in iterator {
             let (k, v) = r.unwrap();
             let (k, v) = r.unwrap();
             let hash_bytes: [u8; 32] = k.as_ref().try_into().unwrap();
             let hash_bytes: [u8; 32] = k.as_ref().try_into().unwrap();
             let tx = deserialize(&v)?;
             let tx = deserialize(&v)?;