shithub: MicroHs

ref: f7223e89e63a69b557ab4287128d1ec6ef5190c9
dir: /tests/ST.hs/

module ST(main) where
import Prelude
import Control.Monad.ST
import Data.STRef
import Debug.Trace

facST :: forall s . Int -> ST s Int
facST n = do
  ri <- newSTRef 1
  rr <- newSTRef 1
  let loop = do
        i <- readSTRef ri
        if i > n then
          return ()
         else do
          writeSTRef ri (i + 1)
          modifySTRef rr (i *)
          loop
  loop
  readSTRef rr
  
fac :: Int -> Int
fac n = runST (facST n)

main :: IO ()
main = do
  print (fac 10)