Return modified value from modify

purescript · May 23, 2018 · c1b9a0d · c1b9a0d
1 parent 4a64b5a
commit c1b9a0d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/Control/Monad/ST/Internal.purs b/src/Control/Monad/ST/Internal.purs
@@ -101,9 +101,9 @@ foreign import read :: forall a r. STRef r a -> ST r a
 foreign import modify' :: forall r a b. (a -> { state :: a, value :: b }) -> STRef r a -> ST r b
 
 -- | Modify the value of a mutable reference by applying a function to the
--- | current value.
-modify :: forall r a. (a -> a) -> STRef r a -> ST r Unit
-modify f = modify' (\s -> { state: f s, value: unit })
+-- | current value. The modified value is returned.
+modify :: forall r a. (a -> a) -> STRef r a -> ST r a
+modify f = modify' \s -> let s' = f s in { state: s', value: s' }
 
 -- | Set the value of a mutable reference.
 foreign import write :: forall a r. a -> STRef r a -> ST r a