Jak mogę się upewnić, że główny wątek kończy się po zakończeniu wszystkich pozostałych wątków?

Question

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

Napisałem powyższy kod. Ale kiedy ją wykonałem, zawsze miałem tylko 123. 456 nie jest drukowany. Chyba dlatego, że główny wątek zakończył się przed drugim wątkiem, więc cały program właśnie się skończył.

Jak mogę temu zapobiec? Każdy api może upewnić się, że główny wątek zakończył się po zakończeniu wszystkich wątków?

OS: OS X 10.8.3

kompilator: ghc 7.4.2

Answer 1

Użyj async biblioteka:

import Control.Concurrent.Async 

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

Jest to równoważne rozwiązania Daniela, z wyjątkiem dwóch niewielkie zalety:

• To gwarantuje, że wszelkie wyjątki podniesione w rozdwojonych wątków dostać przebił w wątku nadrzędnego bez powodowania impasu
• To wygodniejsze

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)