feat(mempool): add contains func to suspended tx pool (#663)

starkware-libs · Sep 17, 2024 · d60d6e9 · d60d6e9
1 parent b1bca27
commit d60d6e9
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/crates/mempool/src/suspended_transaction_pool.rs b/crates/mempool/src/suspended_transaction_pool.rs
@@ -4,11 +4,15 @@ use starknet_api::core::{ContractAddress, Nonce};
 
 use crate::mempool::TransactionReference;
 
-type AddressNonceToTransaction = HashMap<(ContractAddress, Nonce), TransactionReference>;
+type _AddressNonceToTransaction = HashMap<(ContractAddress, Nonce), TransactionReference>;
 
-#[derive(Debug, Default)]
-pub struct SuspendedTransactionPool {
-    _suspended_tx_pool: AddressNonceToTransaction,
+#[derive(Debug)]
+pub struct _SuspendedTransactionPool {
+    suspended_tx_pool: _AddressNonceToTransaction,
 }
 
-impl SuspendedTransactionPool {}
+impl _SuspendedTransactionPool {
+    pub fn _contains(&self, address: ContractAddress, nonce: Nonce) -> bool {
+        self.suspended_tx_pool.contains_key(&(address, nonce))
+    }
+}