mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-24 07:52:14 +03:00
13fc1c62d1
This is a first step towards clarifying the role of `UnpreparedValue` as part of the IR. It certainly does not belong in the parser framework. PR-URL: https://github.com/hasura/graphql-engine-mono/pull/4588 GitOrigin-RevId: d1582a0b266729b79e00d31057178a4099168e6d
64 lines
2.2 KiB
Haskell
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.Column qualified as RQL
|
|
import Hasura.SQL.Backend
|
|
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
|