27 lines
823 B
Haskell
27 lines
823 B
Haskell
|
module Main where
|
||
|
|
|
||
|
|
import qualified Control.Concurrent.Thread as Thread ( forkIO, result )
|
||
|
|
import Control.Concurrent.MVar
|
||
|
|
import Control.Monad
|
||
|
|
import Data.Time.Clock
|
||
|
|
|
||
|
|
import Banking
|
||
|
|
|
||
|
|
main :: IO ()
|
||
|
|
main = do
|
||
|
|
start <- getCurrentTime
|
||
|
|
bank <- setup
|
||
|
|
bank' <- newMVar bank
|
||
|
|
-- start 10 threads doing 10000 treansactions each
|
||
|
|
threads <- sequence $ Thread.forkIO <$> replicate 10 (foldr1 (>=>) (replicate 10000 testTransaction2) bank')
|
||
|
|
banks <- sequence $ (\(_,w) -> w >>= Thread.result) <$> threads
|
||
|
|
stop <- getCurrentTime
|
||
|
|
printMBanks banks
|
||
|
|
print $ "Time taken: " ++ (show $ diffUTCTime stop start)
|
||
|
|
|
||
|
|
printMBanks :: [MVar Bank] -> IO ()
|
||
|
|
printMBanks (a:as) = do
|
||
|
|
a' <- readMVar a
|
||
|
|
sequence_ $ (readMVar >=> print) <$> a'
|
||
|
|
printMBanks [] = return ()
|