module Control.Concurrent.PooledIO.Independent (
run,
runLimited,
runUnlimited,
) where
import Control.Concurrent.PooledIO.Monad (forkFinally, withNumCapabilities)
import Control.Concurrent.MVar (MVar, newEmptyMVar, takeMVar)
import Control.Exception (evaluate)
import Control.Monad (replicateM_)
run :: [IO ()] -> IO ()
run = withNumCapabilities runLimited
runLimited :: Int -> [IO ()] -> IO ()
runLimited numCaps acts = do
let (start, queue) = splitAt numCaps acts
n <- evaluate $ length start
mvar <- newEmptyMVar
mapM_ (forkFinally mvar) start
mapM_ (\act -> takeMVar mvar >> forkFinally mvar act) queue
replicateM_ n $ takeMVar mvar
runUnlimited :: [IO ()] -> IO ()
runUnlimited acts =
mapM_ takeMVar =<< mapM fork acts
fork :: IO () -> IO (MVar ())
fork act = do
mvar <- newEmptyMVar
forkFinally mvar act
return mvar