Improve splitUTxO

cardano-scaling · Jun 12, 2024 · 4a273e2 · 4a273e2
1 parent 0beaeb8
commit 4a273e2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/hydra-node/src/Hydra/Chain/Direct/State.hs b/hydra-node/src/Hydra/Chain/Direct/State.hs
@@ -1040,9 +1040,9 @@ genDecrementTx numParties = do
 splitUTxO :: UTxO -> (UTxO, UTxO)
 splitUTxO utxo =
   let pairs = UTxO.pairs utxo
-      toDecommit = List.head $ List.filter ((> 0) . selectLovelace . txOutValue . snd) pairs
-      restOfUTxO = List.filter ((fst toDecommit /=) . fst) pairs
-   in (UTxO.fromPairs restOfUTxO, UTxO.singleton toDecommit)
+   in case uncons $ List.filter ((> 0) . selectLovelace . txOutValue . snd) pairs of
+        Nothing -> (mempty, mempty)
+        Just (toDecommit, rest) -> (UTxO.fromPairs rest, UTxO.singleton toDecommit)
 
 genCloseTx :: Int -> Gen (ChainContext, OpenState, Tx, ConfirmedSnapshot Tx)
 genCloseTx numParties = do