Backport #238 onto v1.14 (#264)

* Preallocate Bundle Cost (#238)
jito-foundation · Mar 21, 2023 · d19bd9e · d19bd9e
1 parent 6725ab0
commit d19bd9e
Show file tree

Hide file tree

Showing 10 changed files with 234 additions and 35 deletions.
diff --git a/core/src/banking_stage.rs b/core/src/banking_stage.rs
@@ -1656,8 +1656,12 @@ impl BankingStage {
 
         let transaction_costs = qos_service.compute_transaction_costs(txs.iter());
 
-        let (transactions_qos_results, num_included) =
-            qos_service.select_transactions_per_cost(txs.iter(), transaction_costs.iter(), bank);
+        let (transactions_qos_results, num_included) = qos_service.select_transactions_per_cost(
+            txs.iter(),
+            transaction_costs.iter(),
+            bank.slot(),
+            &mut bank.write_cost_tracker().unwrap(),
+        );
 
         let cost_model_throttled_transactions_count = txs.len().saturating_sub(num_included);
 

diff --git a/core/src/bundle_sanitizer.rs b/core/src/bundle_sanitizer.rs
@@ -120,7 +120,10 @@ pub fn get_sanitized_bundle(
         return Err(BundleSanitizerError::FailedCheckTransactions);
     }
 
-    Ok(SanitizedBundle { transactions })
+    Ok(SanitizedBundle {
+        transactions,
+        uuid: packet_bundle.uuid,
+    })
 }
 
 // This function deserializes packets into transactions, computes the blake3 hash of transaction