use try_get_one to parse compute_unit_price and compute_unit_limit

solana-labs · Mar 29, 2024 · 86a5aef · 86a5aef
1 parent 619e6e4
commit 86a5aef
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/token/cli/src/config.rs b/token/cli/src/config.rs
@@ -296,9 +296,18 @@ impl<'a> Config<'a> {
                 (default_program_id, false)
             };
 
-        let nonce_blockhash = value_of(matches, BLOCKHASH_ARG.name);
-        let compute_unit_price = value_of(matches, COMPUTE_UNIT_PRICE_ARG.name);
-        let compute_unit_limit = value_of(matches, COMPUTE_UNIT_LIMIT_ARG.name);
+        let compute_unit_price = matches
+            .try_get_one::<u64>(COMPUTE_UNIT_PRICE_ARG.name)
+            .ok()
+            .flatten()
+            .copied();
+
+        let compute_unit_limit = matches
+            .try_get_one::<u32>(COMPUTE_UNIT_LIMIT_ARG.name)
+            .ok()
+            .flatten()
+            .copied();
+
         let nonce_blockhash = matches
             .try_get_one::<Hash>(BLOCKHASH_ARG.name)
             .ok()