transaction cost based on requested write-locks

solana-labs · Jan 18, 2024 · fae78b2 · fae78b2
1 parent d79d7a0
commit fae78b2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/cost-model/src/cost_model.rs b/cost-model/src/cost_model.rs
@@ -76,7 +76,7 @@ impl CostModel {
     fn get_write_lock_cost(tx_cost: &mut UsageCostDetails, transaction: &SanitizedTransaction) {
         tx_cost.writable_accounts = Self::get_writable_accounts(transaction);
         tx_cost.write_lock_cost =
-            WRITE_LOCK_UNITS.saturating_mul(tx_cost.writable_accounts.len() as u64);
+            WRITE_LOCK_UNITS.saturating_mul(transaction.message().num_write_locks());
     }
 
     fn get_transaction_cost(

diff --git a/sdk/program/src/message/sanitized.rs b/sdk/program/src/message/sanitized.rs
@@ -360,6 +360,8 @@ impl SanitizedMessage {
         num_signatures
     }
 
+    /// Returns the number of requested write-locks in this message.
+    /// This does not consider if write-locks are demoted.
     pub fn num_write_locks(&self) -> u64 {
         self.account_keys()
             .len()