use mutate

paritytech · Jan 30, 2025 · 4a53460 · 4a53460
1 parent 137c56b
commit 4a53460
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/polkadot/runtime/parachains/src/on_demand/mod.rs b/polkadot/runtime/parachains/src/on_demand/mod.rs
@@ -495,9 +495,11 @@ where
 				},
 				PaymentType::Credits => {
 					// Charge the sending account the spot price in credits.
-					let credits = Credits::<T>::get(&sender);
-					ensure!(spot_price <= credits, Error::<T>::InsufficientCredits);
-					Credits::<T>::insert(&sender, credits.saturating_sub(spot_price));
+					Credits::<T>::try_mutate(&sender, |credits| -> DispatchResult {
+						ensure!(spot_price <= *credits, Error::<T>::InsufficientCredits);
+						*credits = credits.saturating_sub(spot_price);
+						Ok(())
+					})?;
 				},
 			}