~ Minor changes.

2 files changed, 10 insertions, 4 deletions

diff --git a/bench/Main.hs b/bench/Main.hs
index 411282a0..87d39f14 100644
--- a/bench/Main.hs
+++ b/bench/Main.hs
@@ -1,6 +1,9 @@
 {-# LANGUAGE OverloadedStrings #-}
 module Main (main) where
 
+import Control.Monad
+import Data.ByteString (ByteString)
+import qualified Data.ByteString as B
 import Criterion.Main
 import Remote.KRPC
 
@@ -8,10 +11,12 @@ import Remote.KRPC
 addr :: RemoteAddr
 addr = (0, 6000)
 
-echo :: Method [Int] [Int]
+echo :: Method ByteString ByteString
 echo = method "echo" ["x"] ["x"]
 
 main :: IO ()
-main = defaultMain $ map mkbench [1, 10, 100, 1000]
+main = defaultMain $ map (mkbench 1)   [1, 10, 100, 1000, 32 * 1024]
+                 ++  map (mkbench 10)  [1, 10, 100, 1000]
   where
-    mkbench n = bench (show n) $ nfIO $ call addr echo [1..n]
\ No newline at end of file
+    mkbench r n = bench (show r ++ "/" ++ show n) $ nfIO $
+                  replicateM r $ call addr echo (B.replicate n 0)
\ No newline at end of file
diff --git a/bench/Server.hs b/bench/Server.hs
index cb5ed316..ece5a7a9 100644
--- a/bench/Server.hs
+++ b/bench/Server.hs
@@ -1,10 +1,11 @@
 {-# LANGUAGE OverloadedStrings #-}
 module Main (main) where
 
+import Data.ByteString (ByteString)
 import Remote.KRPC
 
 
-echo :: Method [Int] [Int]
+echo :: Method ByteString ByteString
 echo = method "echo" ["x"] ["x"]
 
 main :: IO ()