use IndexMap for multiproof

dojoengine · Dec 19, 2024 · d2741d4 · d2741d4
1 parent 9108a12
commit d2741d4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -46,6 +46,7 @@ starknet-types-core = { version = "0.1.7", default-features = false, features =
 rocksdb = { optional = true, version = "0.22", features = [
   "multi-threaded-cf",
 ] }
+indexmap = "2.2.6"
 
 [dev-dependencies]
 env_logger = "0.11.3"

diff --git a/src/trie/proof.rs b/src/trie/proof.rs
@@ -11,10 +11,11 @@ use crate::{
         merkle_node::{Node, NodeHandle},
         tree::NodeKey,
     },
-    BitSlice, BitVec, BonsaiDatabase, BonsaiStorageError, HashMap, HashSet,
+    BitSlice, BitVec, BonsaiDatabase, BonsaiStorageError, HashSet,
 };
 use core::{marker::PhantomData, mem};
 use hashbrown::hash_set;
+use indexmap::IndexMap;
 use starknet_types_core::{felt::Felt, hash::StarkHash};
 
 #[derive(Debug, thiserror::Error)]
@@ -58,7 +59,7 @@ impl ProofNode {
 }
 
 #[derive(Debug, Clone)]
-pub struct MultiProof(pub HashMap<Felt, ProofNode>);
+pub struct MultiProof(pub IndexMap<Felt, ProofNode>);
 impl MultiProof {
     /// If the proof proves more than just the provided `key_values`, this function will not fail.
     /// Not the most optimized way of doing it, but we don't actually need to verify proofs in madara.