如何将parMap与monadic函数一起使用?

时间:2010-02-09 23:59:43

标签: multithreading haskell concurrency parallel-processing monads

我有一个monadic函数getRate:

getRate :: String -> IO Double

我想将此函数映射到String的列表中。通常情况下,我会这样做:

mapM getRate ["foo", "bar"]

但是由于每次调用getRate进行网络调用,我都想并行化地图,以便在一个单独的线程中获取每个速率(或者至少在队列中分散)。我在考虑像

这样的东西
parMapM getRate ["foo", "bar"]

但没有parMapM函数,parMap不适用于monadic函数。

我该怎么办?

2 个答案:

答案 0 :(得分:7)

你应该使用Control.Concurrent并在Control.Concurrent.MVar周围同步;类似的东西:

fork1 :: (a -> IO b) -> a -> IO (MVar b)
fork1 f x =
  do
    cell <- newEmptyMVar
    forkIO (do { result <- f x; putMVar cell result })
    return cell

fork :: (a -> IO b) -> [a] -> IO [MVar b]
fork f = mapM (fork1 f)

join :: [MVar b] -> IO [b]
join = mapM takeMVar

forkJoin :: (a -> IO b) -> [a] -> IO [b]
forkJoin f xs = (fork f xs) >>= join

此部分(fork,join)看起来顺序。在实践中发生的事情是线程在fork中顺序触发,并且集合点遍历等待每个线程。但IO同时发生。

请注意,如果您需要调用外部函数,则应使用forkOS而不是forkIO。

答案 1 :(得分:6)

还有一个monad-parallel包提供mapM :: MonadParallel m => (a -> m b) -> [a] -> m [b]。查看MonadParallel的IO实例,它的工作方式与Dominic的答案相同。