+ Add benchmarks for bitfields.

1 files changed, 15 insertions, 2 deletions

diff --git a/bench/serialization.hs b/bench/Main.hs
index c6d14328..4ce7c818 100644
--- a/bench/serialization.hs
+++ b/bench/Main.hs
@@ -6,7 +6,7 @@ import Control.DeepSeq
 import Criterion.Main
 import Data.ByteString (ByteString)
 import Data.Serialize
-import Network.BitTorrent
+import Network.BitTorrent as BT
 
 
 instance NFData BlockIx where
@@ -33,12 +33,25 @@ encodeMessages xs = runPut (mapM_ put xs)
 decodeMessages :: ByteString -> Either String [Message]
 decodeMessages = runGet (many get)
 
+bitfieldDiff :: Int -> Bitfield
+bitfieldDiff n = BT.empty n `difference` BT.empty n
+
+bitfieldMin :: Int -> Maybe Int
+bitfieldMin n = findMin (BT.empty n)
+
+bitfieldMax :: Int -> Maybe Int
+bitfieldMax n = findMax (BT.empty n)
+
 main :: IO ()
 main = do
-  let datas = replicate 100000 (Request (BlockIx 0 0 0))
+  let datas = replicate 10000 (Request (BlockIx 0 0 0))
 
   defaultMain
     [ datas `deepseq` bench "message/encode"   $ nf encodeMessages datas
     , let binary = encodeMessages datas in
       binary `deepseq` bench "message/decode"  $ nf decodeMessages binary
+
+    , bench "bitfield/difference"  $ nf bitfieldDiff 1000000
+    , bench "bitfield/min"         $ nf bitfieldMin  10000000
+    , bench "bitfield/max"         $ nf bitfieldMax  10000000
     ]
\ No newline at end of file