Simplify implementation of TokenMap.unsafeSubtract.

cardano-foundation · Jun 2, 2023 · 570f8aa · 570f8aa
1 parent 87e05bf
commit 570f8aa
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/lib/primitive/lib/Cardano/Wallet/Primitive/Types/TokenMap.hs b/lib/primitive/lib/Cardano/Wallet/Primitive/Types/TokenMap.hs
@@ -160,6 +160,8 @@ import Numeric.Natural
     ( Natural )
 import Quiet
     ( Quiet (..) )
+import Safe
+    ( fromJustNote )
 
 import qualified Cardano.Wallet.Primitive.Types.TokenQuantity as TokenQuantity
 import qualified Data.Aeson as Aeson
@@ -783,7 +785,5 @@ mapAssetIds f m = fromFlatList $ first f <$> toFlatList m
 -- Throws a run-time exception if the pre-condition is violated.
 --
 unsafeSubtract :: TokenMap -> TokenMap -> TokenMap
-unsafeSubtract a b = F.foldl' acc a $ toFlatList b
-  where
-    acc c (asset, quantity) =
-        adjustQuantity c asset (`TokenQuantity.unsafeSubtract` quantity)
+unsafeSubtract b1 b2 = fromJustNote "TokenMap.unsafeSubtract" $
+    b1 `subtract` b2