Initial commit for benchmarks

- Extracted out code for creating a connection to the test DB
- Added ‘benchmark’ section/target to cabal file

Author: saurabhnanda

Test/Test.hs

@@ -23,19 +23,16 @@ import qualified Data.Time   as Time
 import qualified Data.Aeson as Json
 import qualified Data.Text as T
 
-import           System.Environment (lookupEnv)
-
-import           Control.Applicative ((<$>), (<*>), (<|>))
+import           Control.Applicative ((<$>), (<*>))
 import qualified Control.Applicative as A
 import qualified Control.Arrow as Arr
 import           Control.Arrow ((&&&), (***), (<<<), (>>>))
 
 import           GHC.Int (Int64)
+import TestConnection
 
 import Test.Hspec
 
-import qualified Configuration.Dotenv as Dotenv
-
 {-
 
 Status
@@ -914,24 +911,7 @@ jsonbTests = [testJsonGetFieldValue  table9Q,testJsonGetFieldText  table9Q,
 
 main :: IO ()
 main = do
-  let envVarName = "POSTGRES_CONNSTRING"
-
-  connectStringEnvVar <- lookupEnv envVarName
-
-  connectStringDotEnv <- do vars <- Dotenv.parseFile ".env"
-                            return (lookup envVarName vars)
-                         `Dotenv.onMissingFile`
-                         return Nothing
-
-  let connectString = connectStringEnvVar <|> connectStringDotEnv
-
-  conn <- maybe
-    (fail ("Set " ++ envVarName ++ " environment variable\n"
-           ++ "For example " ++ envVarName ++ "='user=tom dbname=opaleye_test "
-           ++ "host=localhost port=25433 password=tom'"))
-    (PGS.connectPostgreSQL . String.fromString)
-    connectString
-
+  conn <- getTestDbConnection
   dropAndCreateDB conn
 
   let insert (writeable, columndata) =

Test/TestConnection.hs

@@ -0,0 +1,24 @@
+module TestConnection where
+
+import qualified Database.PostgreSQL.Simple as PGS
+import System.Environment
+import qualified Configuration.Dotenv as Dotenv
+import qualified Data.String as String
+import Control.Applicative ((<|>))
+
+getTestDbConnection :: IO PGS.Connection
+getTestDbConnection = do
+  let envVarName = "POSTGRES_CONNSTRING"
+  connectStringEnvVar <- lookupEnv envVarName
+  connectStringDotEnv <- do vars <- Dotenv.parseFile ".env"
+                            return (lookup envVarName vars)
+                         `Dotenv.onMissingFile`
+                         return Nothing
+  let connectString = connectStringEnvVar <|> connectStringDotEnv
+  conn <- maybe
+          (fail ("Set " ++ envVarName ++ " environment variable\n"
+                 ++ "For example " ++ envVarName ++ "='user=tom dbname=opaleye_test "
+                 ++ "host=localhost port=25433 password=tom'"))
+          (PGS.connectPostgreSQL . String.fromString)
+          connectString
+  pure conn

benchmarks/Main.hs

@@ -0,0 +1,10 @@
+module Main where
+
+import Opaleye.RunQuery
+import Criterion.Main
+import TestConnection
+
+main :: IO ()
+main = do
+  conn <- getTestDbConnection
+  pure ()

opaleye.cabal

@@ -98,6 +98,7 @@ test-suite test
   type: exitcode-stdio-1.0
   main-is: Test.hs
   other-modules: QuickCheck
+               , TestConnection
   hs-source-dirs: Test
   build-depends:
     aeson >= 0.6 && < 1.3,
@@ -138,3 +139,20 @@ test-suite tutorial
     time,
     opaleye
   ghc-options: -Wall
+
+test-suite benchmark
+  default-language: Haskell2010
+  type: exitcode-stdio-1.0
+  main-is: Main.hs
+  other-modules: TestConnection
+  hs-source-dirs: benchmarks
+                , Test
+  build-depends: base >= 4 && < 5
+               , postgresql-simple
+               , profunctors
+               , product-profunctors >= 0.6
+               , time
+               , opaleye
+               , criterion
+               , dotenv
+  ghc-options: -Wall