summaryrefslogtreecommitdiff
path: root/libraries/base/tests/listThreads.hs
blob: 5037ea2e32582aa3d9ac419212d07d0135498c62 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
import Data.Maybe
import Control.Monad
import qualified Data.Set as S
import Control.Concurrent
import Data.List (sort)
import GHC.Conc.Sync

dummyThread :: MVar () -> Int -> IO ()
dummyThread mvar n = do
  tid <- myThreadId
  labelThread tid ("thread-"++show n)
  readMVar mvar

main :: IO ()
main = do
  mvar <- newEmptyMVar
  let mkThread n = do
        tid <- forkIO $ readMVar mvar
        let lbl = "thread-"++show n
        labelThread tid lbl
        return lbl

  expectedLabels <- S.fromList <$> mapM mkThread [0..100]
  threads <- listThreads
  labels <- S.fromList . catMaybes <$> mapM threadLabel threads
  unless (S.null $ expectedLabels `S.difference` labels) $
      putStrLn $ unlines [ "thread labels don't match", show expectedLabels, show labels ]
  putMVar mvar ()