1 files changed, 35 insertions, 2 deletions

diff --git a/bench/Main.hs b/bench/Main.hs
index a7b937ff..d021e03e 100644
--- a/bench/Main.hs
+++ b/bench/Main.hs
@@ -1,14 +1,28 @@
+{-# LANGUAGE OverloadedStrings     #-}
+{-# LANGUAGE MultiParamTypeClasses #-}
 {-# OPTIONS -fno-warn-orphans #-}
 module Main (main) where
 
 import Control.DeepSeq
-import Criterion.Main
 import Network
+import Control.Monad
+import Control.Monad.Logger
+import Control.Monad.Reader
+import Criterion.Main
+import Data.ByteString as BS
+import Network.KRPC
+
 
 import Network.BitTorrent.Exchange.Protocol as BT
 import Data.Torrent.Block as BT
 import Data.Torrent.Bitfield as BT
 
+instance KRPC ByteString ByteString where
+  method = "echo"
+
+instance MonadLogger IO where
+  monadLoggerLog _ _ _ _ = return ()
+
 
 instance NFData PortNumber where
   rnf = rnf . (fromIntegral :: PortNumber -> Int)
@@ -38,5 +52,24 @@ decodeMessages :: ByteString -> Either String [Message]
 decodeMessages = runGet (many get)
 -}
 
+echo :: Handler IO
+echo = handler $ \ _ bs -> return (bs :: ByteString)
+
+addr :: SockAddr
+addr = SockAddrInet 6000 (256 * 256 * 256 + 127)
+
+-- main :: IO ()
+-- main = defaultMain []
 main :: IO ()
-main = defaultMain []
+main = withManager def addr [echo] $ \ m -> (`runReaderT` m) $ do
+    listen
+    liftIO $ defaultMain (benchmarks m)
+  where
+    sizes        = [10, 100, 1000, 10000, 16 * 1024]
+    repetitions  = [1, 10, 100, 1000]
+    benchmarks m = [mkbench m r s | r <- repetitions, s <- sizes]
+      where
+        mkbench action r n =
+          bench (show r ++ "times" ++ "/" ++ show n ++ "bytes") $ nfIO $
+            replicateM r $
+              runReaderT (query addr (BS.replicate n 0)) action