mirror of
https://github.com/nikita-volkov/hasql.git
synced 2024-12-27 12:35:50 +03:00
99 lines
2.5 KiB
Haskell
99 lines
2.5 KiB
Haskell
module Main where
|
|
|
|
import Prelude
|
|
import Bug
|
|
import qualified Hasql.Connection as A
|
|
import qualified Hasql.Session as B
|
|
import qualified Hasql.Query 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
|
|
traceEventIO "START Session"
|
|
Right result <- B.run sessionWithManySmallResults connection
|
|
traceEventIO "STOP Session"
|
|
return ()
|
|
where
|
|
acquireConnection =
|
|
A.acquire settings
|
|
where
|
|
settings =
|
|
A.settings host port user password database
|
|
where
|
|
host = "localhost"
|
|
port = 5432
|
|
user = "postgres"
|
|
password = ""
|
|
database = "postgres"
|
|
|
|
|
|
-- * Sessions
|
|
-------------------------
|
|
|
|
sessionWithManySmallParameters :: Vector (Int64, Int64) -> B.Session ()
|
|
sessionWithManySmallParameters =
|
|
$(todo "sessionWithManySmallParameters")
|
|
|
|
sessionWithSingleLargeResultInVector :: B.Session (Vector (Int64, Int64))
|
|
sessionWithSingleLargeResultInVector =
|
|
B.query () queryWithManyRowsInVector
|
|
|
|
sessionWithSingleLargeResultInList :: B.Session (List (Int64, Int64))
|
|
sessionWithSingleLargeResultInList =
|
|
B.query () queryWithManyRowsInList
|
|
|
|
sessionWithManySmallResults :: B.Session (Vector (Int64, Int64))
|
|
sessionWithManySmallResults =
|
|
F.replicateM 1000 (B.query () queryWithSingleRow)
|
|
|
|
|
|
-- * Statements
|
|
-------------------------
|
|
|
|
queryWithManyParameters :: C.Query (Vector (Int64, Int64)) ()
|
|
queryWithManyParameters =
|
|
$(todo "statementWithManyParameters")
|
|
|
|
queryWithSingleRow :: C.Query () (Int64, Int64)
|
|
queryWithSingleRow =
|
|
C.Query template encoder decoder True
|
|
where
|
|
template =
|
|
"SELECT 1, 2"
|
|
encoder =
|
|
conquer
|
|
decoder =
|
|
D.singleRow row
|
|
where
|
|
row =
|
|
tuple <$> D.column D.int8 <*> D.column D.int8
|
|
where
|
|
tuple !a !b =
|
|
(a, b)
|
|
|
|
queryWithManyRows :: (D.Row (Int64, Int64) -> D.Result result) -> C.Query () result
|
|
queryWithManyRows decoder =
|
|
C.Query 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.int8 <*> D.column D.int8
|
|
where
|
|
tuple !a !b =
|
|
(a, b)
|
|
|
|
queryWithManyRowsInVector :: C.Query () (Vector (Int64, Int64))
|
|
queryWithManyRowsInVector =
|
|
queryWithManyRows D.rowVector
|
|
|
|
queryWithManyRowsInList :: C.Query () (List (Int64, Int64))
|
|
queryWithManyRowsInList =
|
|
queryWithManyRows D.rowList
|