Main file for stm-data-collection-test

Main.hs

import Control.Monad ( forM_ )
import Control.Concurrent.STM
import qualified Data.STM.Bag as Bag
import qualified Data.STM.PriorityQueue as PQ

pqTest :: IO ()
pqTest = do
  putStrLn "Start priority queue test"
  pq <- atomically $ (PQ.new :: STM (PQ.Impl Int Int))
  let kvs = [(2, 1), (5, 3), (1, 2), (4, 5)]
  forM_ kvs $ \(k, v) -> atomically $ PQ.insert pq k v
  x <- atomically $ PQ.deleteMin pq
  putStrLn $ "Must print 2"
  putStrLn $ "x = " ++ show x 

bagTest :: IO ()
bagTest = do
  putStrLn "Start bag test"
  bag <- atomically $ (Bag.new :: STM (Bag.Impl Int))
  atomically $ Bag.add bag 7
  atomically $ Bag.add bag 5
  x <- atomically $ Bag.take bag
  putStrLn $ "Must print 5 or 7"
  putStrLn $ "x = " ++ show x

main :: IO ()
main = do
  bagTest
  pqTest
  putStrLn "Done!"