mirror of
https://github.com/nikita-volkov/hasql.git
synced 2024-11-22 10:05:27 +03:00
108 lines
3.1 KiB
Haskell
108 lines
3.1 KiB
Haskell
module Main where
|
|
|
|
import Prelude
|
|
import Bug
|
|
import Criterion
|
|
import Criterion.Main
|
|
import qualified Hasql.Connection as A
|
|
import qualified Hasql.Session as B
|
|
import qualified Hasql.Statement as C
|
|
import qualified Hasql.Decoders as D
|
|
import qualified Hasql.Encoders as E
|
|
import qualified Data.Vector as F
|
|
|
|
|
|
main =
|
|
do
|
|
Right connection <- acquireConnection
|
|
useConnection connection
|
|
where
|
|
acquireConnection =
|
|
A.acquire ""
|
|
useConnection connection =
|
|
defaultMain
|
|
[
|
|
sessionBench "largeResultInVector" sessionWithSingleLargeResultInVector
|
|
,
|
|
sessionBench "largeResultInList" sessionWithSingleLargeResultInList
|
|
,
|
|
sessionBench "manyLargeResults" sessionWithManyLargeResults
|
|
,
|
|
sessionBench "manySmallResults" sessionWithManySmallResults
|
|
]
|
|
where
|
|
sessionBench :: NFData a => String -> B.Session a -> Benchmark
|
|
sessionBench name session =
|
|
bench name (nfIO (fmap (either ($bug "") id) (B.run session connection)))
|
|
|
|
|
|
-- * Sessions
|
|
-------------------------
|
|
|
|
sessionWithManySmallParameters :: Vector (Int64, Int64) -> B.Session ()
|
|
sessionWithManySmallParameters =
|
|
$(todo "sessionWithManySmallParameters")
|
|
|
|
sessionWithSingleLargeResultInVector :: B.Session (Vector (Int64, Int64))
|
|
sessionWithSingleLargeResultInVector =
|
|
B.statement () statementWithManyRowsInVector
|
|
|
|
sessionWithManyLargeResults :: B.Session [Vector (Int64, Int64)]
|
|
sessionWithManyLargeResults =
|
|
replicateM 1000 (B.statement () statementWithManyRowsInVector)
|
|
|
|
sessionWithSingleLargeResultInList :: B.Session (List (Int64, Int64))
|
|
sessionWithSingleLargeResultInList =
|
|
B.statement () statementWithManyRowsInList
|
|
|
|
sessionWithManySmallResults :: B.Session [(Int64, Int64)]
|
|
sessionWithManySmallResults =
|
|
replicateM 1000 (B.statement () statementWithSingleRow)
|
|
|
|
|
|
-- * Statements
|
|
-------------------------
|
|
|
|
statementWithManyParameters :: C.Statement (Vector (Int64, Int64)) ()
|
|
statementWithManyParameters =
|
|
$(todo "statementWithManyParameters")
|
|
|
|
statementWithSingleRow :: C.Statement () (Int64, Int64)
|
|
statementWithSingleRow =
|
|
C.Statement template encoder decoder True
|
|
where
|
|
template =
|
|
"SELECT 1, 2"
|
|
encoder =
|
|
conquer
|
|
decoder =
|
|
D.singleRow row
|
|
where
|
|
row =
|
|
tuple <$> (D.column . D.nonNullable) D.int8 <*> (D.column . D.nonNullable) D.int8
|
|
where
|
|
tuple !a !b =
|
|
(a, b)
|
|
|
|
statementWithManyRows :: (D.Row (Int64, Int64) -> D.Result result) -> C.Statement () result
|
|
statementWithManyRows decoder =
|
|
C.Statement template encoder (decoder rowDecoder) True
|
|
where
|
|
template =
|
|
"SELECT generate_series(0,1000) as a, generate_series(1000,2000) as b"
|
|
encoder =
|
|
conquer
|
|
rowDecoder =
|
|
tuple <$> (D.column . D.nonNullable) D.int8 <*> (D.column . D.nonNullable) D.int8
|
|
where
|
|
tuple !a !b =
|
|
(a, b)
|
|
|
|
statementWithManyRowsInVector :: C.Statement () (Vector (Int64, Int64))
|
|
statementWithManyRowsInVector =
|
|
statementWithManyRows D.rowVector
|
|
|
|
statementWithManyRowsInList :: C.Statement () (List (Int64, Int64))
|
|
statementWithManyRowsInList =
|
|
statementWithManyRows D.rowList
|