benchmark-blake2: implement benchmark

Author: vaibhavsagar

benchmark-blake2/Main.hs

@@ -1,8 +1,11 @@
 {-# LANGUAGE OverloadedStrings #-}
-{-# LANGUAGE PackageImports    #-}
+{-# LANGUAGE PackageImports #-}
 
 module Main where
 
+import Data.ByteString.Char8 (pack)
+import Data.List (foldl')
+
 import           "criterion"         Criterion.Main
 import qualified "memory"            Data.ByteArray             as B
 import qualified "base16-bytestring" Data.ByteString.Base16     as B16
@@ -25,16 +28,20 @@ patchedHash = Patched.hash 64 mempty
 
 main :: IO ()
 main = defaultMain
-  [ bgroup "blake2"
-    [
-
+  [ bgroup "hashing"
+    [ bench "blake2"         $ nfIO (bencher originalHash)
+    , bench "blake2-patched" $ nfIO (bencher patchedHash)
+    , bench "cryptonite"     $ nfIO (bencher cryptoniteHash)
     ]
-  , bgroup "blake2-patched"
-    [
+  ]
 
-    ]
-  , bgroup "cryptonite"
-    [
+hashes :: IO [BS.ByteString]
+hashes = let
+  ns = map (cryptoniteHash . pack . show) [1..100000]
+  in length ns `seq` pure ns
 
-    ]
-  ]
+bencher :: (BS.ByteString -> BS.ByteString) -> IO Bool
+bencher hash = do
+  hs <- hashes
+  let every = foldl' (\(i, b) h -> if (hash . pack . show $ (i+1)) == h then (i+1, b) else (i+1, False)) (0, True) hs
+  pure $ snd every