在下面的 Haskell 代码中,如何强制主线程等待其所有子线程完成。
我无法使用 forkFinally,如此链接中的“终止程序”部分所示:(http://hackage.haskell.org/package/base-4.7.0.2/docs/Control-Concurrent.html http://hackage.haskell.org/package/base-4.7.0.2/docs/Control-Concurrent.html).
使用 TMVar 时我得到了想要的结果。但我想和 TVar 一起做这件事。
请帮忙。
module Main
where
import Control.Monad
import Control.Concurrent
import Control.Concurrent.STM
type TInt = TVar Int
transTest :: TInt -> Int -> IO ()
transTest n t = do
atomically $ do
t1 <- readTVar n
doSomeJob t
t2 <- readTVar n
writeTVar n t
doSomeJob :: Int -> STM ()
doSomeJob t = do
x <- newTVar 0
let l = 10^6*t
forM_ [1..l] (\i -> do
writeTVar x i )
main :: IO ()
main = do
n <- newTVarIO 0
let v = 5
forkIO (transTest n v)
let v = 3
forkIO (transTest n v)
let v = 7
forkIO (transTest n v)
let v = 1
forkIO (transTest n v)
r <- atomically $ readTVar n
putStrLn("Last updated value = " ++ (show r))
我过去所做的是为每个分叉线程创建一个小MVar,然后使用forkFinally
分叉线程,以便在最后,每个线程都会将一个虚拟值放入 MVar 中(即我使用 MVar 作为同步原语)。然后我可以打电话takeMVar
在那些 MVar 上等待。
我将它包装成一个小辅助函数:
forkThread :: IO () -> IO (MVar ())
forkThread proc = do
handle <- newEmptyMVar
_ <- forkFinally proc (\_ -> putMVar handle ())
return handle
使用这个,您的代码可以更改为类似的内容
-- Fork four threads
threads <- forM [5, 3, 7, 1] (\v -> forkThread (transTest n v))
-- Wait for all of them
mapM_ takeMVar threads
However,那是在我读到 Simon Marlow 的(最优秀的)书《Haskell 中的并行和并发编程》之前,这本书让我意识到async https://hackage.haskell.org/package/async-2.0.2/docs/Control-Concurrent-Async.html包裹。该包提供了一个抽象,它不仅处理所有这些事情,所以你可以只写
-- Runs 'transTest n {5,3,7,1}' in parallel and waits for all threads
_ <- mapConcurrently (transTest n) [5, 3, 7, 1]
...它还处理诸如(异步)异常之类的事情。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)