gaspersic/solana
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Reduce budget request instruction length (#20636) (#20644)

Browse Source 
(cherry picked from commit c231cfe235)

Co-authored-by: Jack May <jack@solana.com>
This commit is contained in:
mergify[bot]
2021-10-20 12:17:29 -07:00
committed by GitHub
parent 53f4bde471
commit d5fc81e12a
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
sdk/src/compute_budget.rs
View File

@ -12,7 +12,7 @@ use solana_sdk::{
crate::declare_id!("ComputeBudget111111111111111111111111111111"); crate::declare_id!("ComputeBudget111111111111111111111111111111");
const MAX_UNITS: u64 = 1_000_000; const MAX_UNITS: u32 = 1_000_000;
/// Compute Budget Instructions /// Compute Budget Instructions
#[derive( #[derive(
@ -30,11 +30,11 @@ const MAX_UNITS: u64 = 1_000_000;
pub enum ComputeBudgetInstruction { pub enum ComputeBudgetInstruction {
/// Request a specific maximum number of compute units the transaction is /// Request a specific maximum number of compute units the transaction is
/// allowed to consume. /// allowed to consume.
RequestUnits(u64), RequestUnits(u32),
} }
/// Create a `ComputeBudgetInstruction::RequestUnits` `Instruction` /// Create a `ComputeBudgetInstruction::RequestUnits` `Instruction`
pub fn request_units(units: u64) -> Instruction { pub fn request_units(units: u32) -> Instruction {
Instruction::new_with_borsh(id(), &ComputeBudgetInstruction::RequestUnits(units), vec![]) Instruction::new_with_borsh(id(), &ComputeBudgetInstruction::RequestUnits(units), vec![])
} }
@ -52,7 +52,7 @@ pub fn process_request(
if units > MAX_UNITS { if units > MAX_UNITS {
return Err(error); return Err(error);
} }
compute_budget.max_units = units; compute_budget.max_units = units as u64;
} }
} }
Ok(()) Ok(())
@ -134,7 +134,7 @@ mod tests {
assert_eq!( assert_eq!(
compute_budget, compute_budget,
BpfComputeBudget { BpfComputeBudget {
max_units: MAX_UNITS, max_units: MAX_UNITS as u64,
..BpfComputeBudget::default() ..BpfComputeBudget::default()
} }
); );