Come posso assicurarmi che il filo principale termini dopo che tutti gli altri fili sono finiti?

Question

import Control.Concurrent 
main = do 
    forkIO $ putStrLn "123" 
    forkIO $ putStrLn "456" 

Ho scritto il codice sopra. Ma quando l'ho eseguito, ho sempre ricevuto solo 123. 456 non viene stampato. Suppongo che sia perché il thread principale è finito prima dell'altro thread, quindi l'intero programma è appena finito.

Come posso evitare questo? Qualsiasi API può assicurarsi che il thread principale sia terminato dopo che tutti i thread sono terminati?

OS: OS X 10.8.3

compilatore: GHC 7.4.2

Answer 1

utilizzare la libreria async:

import Control.Concurrent.Async 

main = do 
    a1 <- async $ putStrLn "123" 
    a2 <- async $ putStrLn "456" 
    mapM_ wait [a1, a2] 

Ciò equivale a una soluzione di Daniel, se non con due piccoli vantaggi:

• assicura che tutte le eccezioni sollevate nelle filettature biforcute ottenere re-sollevati nella discussione genitore senza causare un deadlock
• È più conveniente

Answer 2

import Control.Concurrent 
main = do 
    done <- newEmptyMVar 
    forkIO $ putStrLn "123" >> putMVar done() 
    forkIO $ putStrLn "456" >> putMVar done() 
    takeMVar done 
    takeMVar done 
    -- OR: replicateM_ 2 (takeMVar done)