Use "criterion" for benchmarks

Author: awjchen

benchmarks/Counter.hs

@@ -3,16 +3,31 @@
 -- | Perform 100,000 atomic increments using 100 concurrent writers.
 module Main where
 
+import Prelude hiding (read)
+
 import Control.Concurrent
 import Control.Monad
+import Criterion
+import Criterion.Main (defaultMain)
+import Data.Int (Int64)
 import System.Metrics.Counter
 
 main :: IO ()
-main = do
+main = defaultMain
+  [ bench "Increment counter with multiple writers" $
+      whnfIO incrementWithMultipleWriters
+  ]
+
+incrementWithMultipleWriters :: IO Int64
+incrementWithMultipleWriters = do
     counter <- new
     locks <- replicateM n newEmptyMVar
     mapM_ (forkIO . work counter iters) locks
     mapM_ takeMVar locks
+    total <- read counter
+    unless (fromIntegral total == n*iters) $
+      error "Incorrect count!"
+    pure total
   where
     n = 100
     iters = 100000

benchmarks/Distribution.hs

@@ -4,16 +4,31 @@
 -- writers.
 module Main where
 
+import Prelude hiding (read)
+
 import Control.Concurrent
 import Control.Monad
+import Criterion
+import Criterion.Main (defaultMain)
+import Data.Int (Int64)
 import System.Metrics.Distribution
 
 main :: IO ()
-main = do
+main = defaultMain
+  [ bench "Add to distribution with multiple writers" $
+      whnfIO addWithMultipleWriters
+  ]
+
+addWithMultipleWriters :: IO Int64
+addWithMultipleWriters = do
     distrib <- new
     locks <- replicateM n newEmptyMVar
     mapM_ (forkIO . work distrib iters) locks
     mapM_ takeMVar locks
+    total <- count <$> read distrib
+    unless (fromIntegral total == n*iters) $
+      error "Incorrect count!"
+    pure total
   where
     n = 100
     iters = 100000

ekg-core.cabal

@@ -58,7 +58,8 @@ benchmark counter
   type: exitcode-stdio-1.0
   build-depends:
     base,
-    ekg-core
+    ekg-core,
+    criterion ^>= 1.5.9.0
   default-language: Haskell2010
   hs-source-dirs: benchmarks
   ghc-options: -O2 -threaded -Wall
@@ -68,7 +69,8 @@ benchmark distribution
   type: exitcode-stdio-1.0
   build-depends:
     base,
-    ekg-core
+    ekg-core,
+    criterion ^>= 1.5.9.0
   default-language: Haskell2010
   hs-source-dirs: benchmarks
   ghc-options: -O2 -threaded -Wall

sample.hie.yaml

@@ -4,3 +4,7 @@ cradle:
       component: "lib:ekg-core"
     - path: "./test"
       component: "test:ekg-core-test"
+    - path: "./benchmarks"
+      component: "bench:counter"
+    - path: "./benchmarks"
+      component: "bench:distribution"