feat: add AccountTransactionIndex insert (#316)

Until we support fee escalation, this should always return None. Subsequent PRs will use this DS, and panic if `Some` is returned. commit-id:b0eee880 Co-authored-by: Gilad Chase <[email protected]>
starkware-libs · Jun 30, 2024 · bd27a39 · bd27a39
1 parent f077bb8
commit bd27a39
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/crates/mempool/src/transaction_pool.rs b/crates/mempool/src/transaction_pool.rs
@@ -80,3 +80,10 @@ impl TransactionPool {
 pub struct AccountTransactionIndex(
     pub HashMap<ContractAddress, BTreeMap<Nonce, TransactionReference>>,
 );
+
+impl AccountTransactionIndex {
+    /// If the transaction already exists in the mapping, the old value is returned.
+    pub fn insert(&mut self, tx: TransactionReference) -> Option<TransactionReference> {
+        self.0.entry(tx.sender_address).or_default().insert(tx.nonce, tx)
+    }
+}