mirror of
https://github.com/nikita-volkov/hasql.git
synced 2024-12-27 12:35:50 +03:00
66 lines
2.5 KiB
Haskell
66 lines
2.5 KiB
Haskell
module Hasql.Private.Session where
|
|
|
|
import qualified Hasql.Private.Connection as Connection
|
|
import qualified Hasql.Private.Decoders.Result as Decoders.Result
|
|
import qualified Hasql.Private.Decoders.Results as Decoders.Results
|
|
import qualified Hasql.Private.Encoders as Encoders
|
|
import qualified Hasql.Private.Encoders.Params as Encoders.Params
|
|
import Hasql.Private.Errors
|
|
import qualified Hasql.Private.IO as IO
|
|
import Hasql.Private.Prelude
|
|
import qualified Hasql.Statement as Statement
|
|
|
|
-- |
|
|
-- A batch of actions to be executed in the context of a database connection.
|
|
newtype Session a
|
|
= Session (ReaderT Connection.Connection (ExceptT QueryError IO) a)
|
|
deriving (Functor, Applicative, Monad, MonadError QueryError, MonadIO, MonadReader Connection.Connection)
|
|
|
|
-- |
|
|
-- Executes a bunch of commands on the provided connection.
|
|
run :: Session a -> Connection.Connection -> IO (Either QueryError a)
|
|
run (Session impl) connection =
|
|
runExceptT
|
|
$ runReaderT impl connection
|
|
|
|
-- |
|
|
-- Possibly a multi-statement query,
|
|
-- which however cannot be parameterized or prepared,
|
|
-- nor can any results of it be collected.
|
|
sql :: ByteString -> Session ()
|
|
sql sql =
|
|
Session
|
|
$ ReaderT
|
|
$ \(Connection.Connection pqConnectionRef integerDatetimes registry) ->
|
|
ExceptT
|
|
$ fmap (mapLeft (QueryError sql []))
|
|
$ withMVar pqConnectionRef
|
|
$ \pqConnection -> do
|
|
r1 <- IO.sendNonparametricStatement pqConnection sql
|
|
r2 <- IO.getResults pqConnection integerDatetimes decoder
|
|
return $ r1 *> r2
|
|
where
|
|
decoder =
|
|
Decoders.Results.single Decoders.Result.noResult
|
|
|
|
-- |
|
|
-- Parameters and a specification of a parametric single-statement query to apply them to.
|
|
statement :: params -> Statement.Statement params result -> Session result
|
|
statement input (Statement.Statement template (Encoders.Params paramsEncoder) decoder preparable) =
|
|
Session
|
|
$ ReaderT
|
|
$ \(Connection.Connection pqConnectionRef integerDatetimes registry) ->
|
|
ExceptT
|
|
$ fmap (mapLeft (QueryError template inputReps))
|
|
$ withMVar pqConnectionRef
|
|
$ \pqConnection -> do
|
|
r1 <- IO.sendParametricStatement pqConnection integerDatetimes registry template paramsEncoder preparable input
|
|
r2 <- IO.getResults pqConnection integerDatetimes (unsafeCoerce decoder)
|
|
return $ r1 *> r2
|
|
where
|
|
inputReps =
|
|
let Encoders.Params.Params (Op encoderOp) = paramsEncoder
|
|
step (_, _, _, rendering) acc =
|
|
rendering : acc
|
|
in foldr step [] (encoderOp input)
|