1
0
mirror of https://github.com/hasura/graphql-engine.git synced 2024-12-23 15:32:27 +03:00
graphql-engine/server/src-lib/Hasura/Backends/MySQL/Plan.hs
2023-04-24 18:37:33 +00:00

64 lines
2.2 KiB
Haskell

-- | Planning MySQL queries and subscriptions.
module Hasura.Backends.MySQL.Plan
( planQuery,
queryToActionForest,
)
where
import Control.Monad.Validate
import Data.Aeson qualified as J
import Data.ByteString.Lazy (toStrict)
import Data.Text.Extended
import Data.Tree
import Hasura.Backends.MySQL.DataLoader.Plan qualified as DataLoader
import Hasura.Backends.MySQL.FromIr
import Hasura.Backends.MySQL.Types
import Hasura.Base.Error
import Hasura.Prelude hiding (first)
import Hasura.RQL.IR
import Hasura.RQL.Types.BackendType
import Hasura.RQL.Types.Column qualified as RQL
import Hasura.Session
-- | Plan the query and then produce a forest of actions for the executor.
queryToActionForest ::
MonadError QErr m =>
UserInfo ->
QueryDB 'MySQL Void (UnpreparedValue 'MySQL) ->
m (DataLoader.HeadAndTail, Forest DataLoader.PlannedAction)
queryToActionForest userInfo qrf = do
select <- planQuery (_uiSession userInfo) qrf
let (!headAndTail, !plannedActionsList) =
DataLoader.runPlan
(DataLoader.planSelectHeadAndTail Nothing Nothing select)
!actionsForest = DataLoader.actionsForest id plannedActionsList
pure (headAndTail, actionsForest)
planQuery ::
MonadError QErr m =>
SessionVariables ->
QueryDB 'MySQL Void (UnpreparedValue 'MySQL) ->
m Select
planQuery sessionVariables queryDB = do
rootField <- traverse (prepareValueQuery sessionVariables) queryDB
runValidate (runFromIr (fromRootField rootField))
`onLeft` (throw400 NotSupported . tshow)
-- | Prepare a value without any query planning; we just execute the
-- query with the values embedded.
prepareValueQuery ::
MonadError QErr m =>
SessionVariables ->
UnpreparedValue 'MySQL ->
m Expression
prepareValueQuery sessionVariables =
\case
UVLiteral x -> pure x
UVSession -> pure $ ValueExpression $ BinaryValue $ toStrict $ J.encode sessionVariables
UVParameter _ RQL.ColumnValue {..} -> pure $ ValueExpression cvValue
UVSessionVar _typ sessionVariable -> do
value <-
getSessionVariableValue sessionVariable sessionVariables
`onNothing` throw400 NotFound ("missing session variable: " <>> sessionVariable)
pure $ ValueExpression $ TextValue value