write to database after successful simulation

worldcoin · Dec 14, 2023 · eb6b1fc · eb6b1fc
1 parent 00ff1c6
commit eb6b1fc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 20 deletions.
diff --git a/src/db.rs b/src/db.rs
@@ -1295,15 +1295,7 @@ mod tests {
         let initial_max_fee_per_gas = U256::from(1);
         let initial_max_priority_fee_per_gas = U256::from(1);
 
-        db.insert_into_tx_hashes(
-            tx_id,
-            tx_hash_1,
-            initial_max_fee_per_gas,
-            initial_max_priority_fee_per_gas,
-        )
-        .await?;
-
-        db.insert_into_sent_transactions(
+        db.insert_tx_broadcast(
             tx_id,
             tx_hash_1,
             initial_max_fee_per_gas,

diff --git a/src/tasks/broadcast.rs b/src/tasks/broadcast.rs
@@ -109,6 +109,8 @@ async fn broadcast_relayer_txs(
             .fill_transaction(&mut typed_transaction, None)
             .await?;
 
+        tracing::debug!(?tx.id, "Simulating tx");
+
         // Simulate the transaction
         match middleware.call(&typed_transaction, None).await {
             Ok(_) => {
@@ -129,7 +131,7 @@ async fn broadcast_relayer_txs(
         let tx_hash = H256::from(ethers::utils::keccak256(&raw_signed_tx));
 
         app.db
-            .insert_into_tx_hashes(
+            .insert_tx_broadcast(
                 &tx.id,
                 tx_hash,
                 max_fee_per_gas,
@@ -155,16 +157,6 @@ async fn broadcast_relayer_txs(
             }
         };
 
-        // Insert the tx into
-        app.db
-            .insert_into_sent_transactions(
-                &tx.id,
-                tx_hash,
-                max_fee_per_gas,
-                max_priority_fee_per_gas,
-            )
-            .await?;
-
         tracing::info!(id = tx.id, hash = ?tx_hash, "Tx broadcast");
     }