feat: compute_hash_on_elements takes ExactSizeIterator

xJonathanLEI · Jun 3, 2024 · c4e700a · c4e700a
1 parent 1ba71e3
commit c4e700a
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/starknet-core/src/crypto.rs b/starknet-core/src/crypto.rs
@@ -46,14 +46,19 @@ mod errors {
 }
 pub use errors::{EcdsaSignError, EcdsaVerifyError};
 
-pub fn compute_hash_on_elements(data: &[FieldElement]) -> FieldElement {
+pub fn compute_hash_on_elements<'a, ESI, II>(data: II) -> FieldElement
+where
+    ESI: ExactSizeIterator<Item = &'a FieldElement>,
+    II: IntoIterator<IntoIter = ESI>,
+{
     let mut current_hash = FieldElement::ZERO;
+    let iter = data.into_iter();
+    let data_len = FieldElement::from(iter.len());
 
-    for item in data.iter() {
+    for item in iter {
         current_hash = pedersen_hash(&current_hash, item);
     }
 
-    let data_len = FieldElement::from(data.len());
     pedersen_hash(&current_hash, &data_len)
 }