From 79b88ddeb5d96c1a82397605a5b20ab5d352b848 Mon Sep 17 00:00:00 2001
From: Victor Denisov <vdenisov@mirantis.com>
Date: Sat, 29 Aug 2015 15:49:13 -0700
Subject: [PATCH] Add benchmark for inserts

---
 Benchmark.hs  | 26 ++++++++++++++++++++++++++
 mongoDB.cabal | 24 ++++++++++++++++++++++++
 2 files changed, 50 insertions(+)
 create mode 100644 Benchmark.hs

diff --git a/Benchmark.hs b/Benchmark.hs
new file mode 100644
index 0000000..a299ad7
--- /dev/null
+++ b/Benchmark.hs
@@ -0,0 +1,26 @@
+import Criterion.Main
+
+import Control.Monad (forM_, void)
+import qualified Database.MongoDB as M
+import Data.Bson (Document, Field(..), Label, Val, Value(String, Doc, Bool),
+                  Javascript, at, valueAt, lookup, look, genObjectId, (=:),
+                  (=?))
+
+import Database.MongoDB.Query
+
+import qualified Data.Text as T
+
+main = defaultMain [
+    bgroup "insert" [ bench "100" $ nfIO doInserts ]
+  ]
+
+doInserts = do
+    let docs = (flip map) [0..100] $ \i ->
+            ["name" M.=: (T.pack $ "name " ++ (show i))]
+
+    pipe <- M.connect (M.host "127.0.0.1")
+
+    forM_ docs $ \doc -> do
+      void $ M.access pipe M.master "mongodb-haskell-test" $ M.insert "bigCollection" doc
+
+    M.close pipe
diff --git a/mongoDB.cabal b/mongoDB.cabal
index 1bda803..dd2fcbe 100644
--- a/mongoDB.cabal
+++ b/mongoDB.cabal
@@ -73,3 +73,27 @@ test-suite test
 
   default-language: Haskell2010
   default-extensions: OverloadedStrings
+
+Benchmark bench
+  main-is:            Benchmark.hs
+  type:               exitcode-stdio-1.0
+  Build-depends:      array -any
+                    , base < 5
+                    , binary -any
+                    , bson >= 0.3 && < 0.4
+                    , text
+                    , bytestring -any
+                    , containers -any
+                    , mtl >= 2
+                    , cryptohash -any
+                    , network -any
+                    , parsec -any
+                    , random -any
+                    , random-shuffle -any
+                    , monad-control >= 0.3.1
+                    , lifted-base >= 0.1.0.3
+                    , transformers-base >= 0.4.1
+                    , hashtables >= 1.1.2.0
+                    , criterion
+  default-language: Haskell2010
+  default-extensions: OverloadedStrings