entwine-0.0.3: test/bench.hs

{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE NoImplicitPrelude #-}
import           Control.Concurrent
import           Control.Monad.Trans.Either

import           Criterion.Main
import           Criterion.Types (Config (..))

import           Entwine.Data
import           Entwine.Parallel
import           Entwine.P

import           System.IO

import           Test.QuickCheck.Instances ()

run_ :: EitherT (RunError ()) IO a -> IO a
run_ =
  eitherT (const $ fail "fail") pure

run :: IO (Either (RunError ()) a) -> IO a
run =
   (=<<) (either (const $ fail "fail") pure)

main :: IO ()
main = do
  let pro n = \q -> forM_ [1..n] (writeQueue q)

      work_ :: Int -> EitherT () IO ()
      work_ =
        EitherT . work

      work :: Int -> IO (Either () ())
      work _ = liftIO $ do
        threadDelay 10000 {-- 10 ms --}
        pure $ Right ()

  let cfg =
        defaultConfig {
            reportFile = Just "dist/build/twine-bench.html"
          }


  defaultMainWith cfg [
      bgroup "sem-100" [
          bench "work-1000" (nfIO . run_ $ consume_ (pro 1000) 100 work_)
        , bench "work-1000" (nfIO . run $ consume (pro 1000) 100 work)
        ]
    , bgroup "sem-1000" [
          bench "work-1000" (nfIO . run_ $ consume_ (pro 1000) 1000 work_)
        , bench "work-1000" (nfIO . run $ consume (pro 1000) 1000 work)
        , bench "work-10000" (nfIO . run_ $ consume_ (pro 10000) 1000 work_)
        , bench "work-10000" (nfIO . run $ consume (pro 10000) 1000 work)
        , bench "work-100000" (nfIO . run_ $ consume_ (pro 100000) 1000 work_)
        , bench "work-100000" (nfIO . run $ consume (pro 100000) 1000 work)
        ]
    ]