+ Add bench for intersection & union

1 files changed, 15 insertions, 6 deletions

diff --git a/bench/Main.hs b/bench/Main.hs
index 4ce7c818..59cb9ea7 100644
--- a/bench/Main.hs
+++ b/bench/Main.hs
@@ -33,15 +33,22 @@ 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)
 
+bitfieldDiff :: Int -> Bitfield
+bitfieldDiff n = BT.empty n `difference` BT.empty n
+
+bitfieldInter :: Int -> Bitfield
+bitfieldInter n = BT.empty n `intersection` BT.empty n
+
+bitfieldUnion :: Int -> Bitfield
+bitfieldUnion n = BT.empty n `union` BT.empty n
+
+
 main :: IO ()
 main = do
   let datas = replicate 10000 (Request (BlockIx 0 0 0))
@@ -51,7 +58,9 @@ main = do
     , 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
+    , bench "bitfield/min"          $ nf bitfieldMin   10000000
+    , bench "bitfield/max"          $ nf bitfieldMax   10000000
+    , bench "bitfield/difference"   $ nf bitfieldDiff  1000000
+    , bench "bitfield/intersection" $ nf bitfieldInter 1000000
+    , bench "bitfield/union"        $ nf bitfieldUnion 1000000
     ]
\ No newline at end of file