xxhash-ffi-0.3.1: bench/Bench.hs

{-# OPTIONS_GHC -Wno-deprecations #-}

module Main (main) where

import qualified Data.ByteString as B
import qualified Data.Digest.Adler32 as Adler32
import qualified Data.Digest.CRC32 as CRC32
import qualified Data.Digest.XXHash.FFI as FFI
import qualified Data.Hashable as Hashable

import Test.Tasty.Bench

main :: IO ()
main = do
  -- apt-get install wamerican
  med <- B.readFile "/usr/share/dict/words"

  defaultMain
    [ bgroup
        "10B"
        [ bench "Data.Digest.XXHash.FFI.xxh32" $ nf (FFI.xxh32 bs10b) 0
        , bench "Data.Digest.XXHash.FFI.xxh64" $ nf (FFI.xxh64 bs10b) 0
        , bench "Data.Digest.Adler32" $ nf Adler32.adler32 bs10b
        , bench "Data.Digest.CRC32" $ nf CRC32.crc32 bs10b
        , bench "Data.Digest.XXHash.FFI.XXH3" $ nf Hashable.hash (FFI.XXH3 bs10b)
        , bench "Data.Hashable" $ nf Hashable.hash bs10b
        ]
    , bgroup
        "1kB"
        [ bench "Data.Digest.XXHash.FFI.xxh32" $ nf (FFI.xxh32 bs1k) 0
        , bench "Data.Digest.XXHash.FFI.xxh64" $ nf (FFI.xxh64 bs1k) 0
        , bench "Data.Digest.Adler32" $ nf Adler32.adler32 bs1k
        , bench "Data.Digest.CRC32" $ nf CRC32.crc32 bs1k
        , bench "Data.Digest.XXHash.FFI.XXH3" $ nf Hashable.hash (FFI.XXH3 bs1k)
        , bench "Data.Hashable" $ nf Hashable.hash bs1k
        ]
    , bgroup
        "4kB"
        [ bench "Data.Digest.XXHash.FFI.xxh32" $ nf (FFI.xxh32 bs4k) 0
        , bench "Data.Digest.XXHash.FFI.xxh64" $ nf (FFI.xxh64 bs4k) 0
        , bench "Data.Digest.Adler32" $ nf Adler32.adler32 bs4k
        , bench "Data.Digest.CRC32" $ nf CRC32.crc32 bs4k
        , bench "Data.Digest.XXHash.FFI.XXH3" $ nf Hashable.hash (FFI.XXH3 bs4k)
        , bench "Data.Hashable" $ nf Hashable.hash bs4k
        ]
    , bgroup
        "1MB"
        [ bench "Data.Digest.XXHash.FFI.xxh32" $ nf (FFI.xxh32 med) 0
        , bench "Data.Digest.XXHash.FFI.xxh64" $ nf (FFI.xxh64 med) 0
        , bench "Data.Digest.Adler32" $ nf Adler32.adler32 med
        , bench "Data.Digest.CRC32" $ nf CRC32.crc32 med
        , bench "Data.Digest.XXHash.FFI.XXH3" $ nf Hashable.hash (FFI.XXH3 med)
        , bench "Data.Hashable" $ nf Hashable.hash med
        ]
    ]

bs10b :: B.ByteString
bs10b = fst $ B.unfoldrN 10 (\x -> Just (x, x)) 0

bs1k :: B.ByteString
bs1k = fst $ B.unfoldrN 1024 (\x -> Just (x, x)) 0

bs4k :: B.ByteString
bs4k = fst $ B.unfoldrN 4096 (\x -> Just (x, x)) 0