8

我可以从多个并行线程安全地分配给 IOVector（来自 Haskell 矢量包）的非重叠索引，还是需要额外的互斥锁？

背景：我想并行运行一组 IO 计算，然后记录所有结果。我首先在列表中返回结果，但这似乎是错误的数据结构。所以我希望使用矢量包中的 IOVector 来存储结果。

我的问题是，我可以写类似的东西：

runPar :: [IO a] -> IO (IOVector a)
runPar tasks = do
  v <- new (length tasks)
  mapM forkIO [task >>= write v i | (i, task) <- zip [0..] tasks]
  -- Wait for all tasks to complete
  return v

这能保证安全吗？或者我是否需要一个类似互斥锁的控件（例如将 v 保存在 an 中MVar）以确保一次只进行一次写入？

4

1 回答 1

2

从代码和评论中的讨论来看，似乎没有理由假设它不是线程安全的。为了支持这一点，这里有一个小测试：

import qualified Data.Vector.Mutable as V
import System.Environment
import Control.Concurrent.Async

thread v i 0 = return ()
thread v i n = do
    x <- V.read v i
    V.write v i $! x-1
    thread v i (n-1)

main = do
    [m,n] <- map read `fmap` getArgs
    v <- V.replicate m n
    tis <- mapM (\i -> async (thread v i n)) [0..m-1]
    mapM_ wait tis
    r <- mapM (V.read v) [0..m-1]
    if all (== 0) r then putStrLn "OK" else putStrLn $ "Not OK: " ++ show r

在这里，即使有 100 个并行产生的线程，从 1000000 开始倒数，使用-RTS -N​​ or -RTS -N10，总是会产生OK。

于 2013-11-24T10:51:47.577 回答