Refactors gas fns arguments

anoma · Mar 22, 2023 · c1b77bb · c1b77bb
1 parent b38c8cb
commit c1b77bb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/core/src/ledger/gas.rs b/core/src/ledger/gas.rs
@@ -200,8 +200,8 @@ impl VpGasMeter {
 }
 
 impl VpsGas {
-    /// Set the gas cost from a single VP run.
-    pub fn set(&mut self, vp_gas_meter: &VpGasMeter) -> Result<()> {
+    /// Set the gas cost from a single VP run. It consumes the [`VpGasMeter`] instance which shouldn't be accessed passed this point.
+    pub fn set(&mut self, vp_gas_meter: VpGasMeter) -> Result<()> {
         debug_assert_eq!(self.max, None);
         debug_assert!(self.rest.is_empty());
         self.max = Some(vp_gas_meter.current_gas);

diff --git a/shared/src/ledger/protocol/mod.rs b/shared/src/ledger/protocol/mod.rs
@@ -434,7 +434,7 @@ where
             // Returning error from here will short-circuit the VP parallel
             // execution. It's important that we only short-circuit gas
             // errors to get deterministic gas costs
-            result.gas_used.set(&gas_meter).map_err(Error::GasError)?;
+            result.gas_used.set(gas_meter).map_err(Error::GasError)?;
             match accept {
                 Ok(accepted) => {
                     if !accepted {