mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-19 13:31:43 +03:00
df10ccf310
PR-URL: https://github.com/hasura/graphql-engine-mono/pull/9374 GitOrigin-RevId: 073e51480e2be97e32938fa61a3547374e04ead8
83 lines
3.3 KiB
Haskell
83 lines
3.3 KiB
Haskell
-- | This module defines the top-level translation functions pertaining to
|
|
-- queries that are not aggregation queries, i.e. so-called "simple" selects
|
|
-- into Postgres AST.
|
|
module Hasura.Backends.Postgres.Translate.Select.Simple
|
|
( mkSQLSelect,
|
|
selectQuerySQL,
|
|
)
|
|
where
|
|
|
|
import Control.Monad.Writer.Strict (runWriter)
|
|
import Database.PG.Query (Query)
|
|
import Hasura.Backends.Postgres.SQL.DML qualified as S
|
|
import Hasura.Backends.Postgres.SQL.Types (IsIdentifier (toIdentifier))
|
|
import Hasura.Backends.Postgres.Translate.Select.AnnotatedFieldJSON
|
|
import Hasura.Backends.Postgres.Translate.Select.Internal.Extractor (asJsonAggExtr)
|
|
import Hasura.Backends.Postgres.Translate.Select.Internal.GenerateSelect (generateSQLSelectFromArrayNode)
|
|
import Hasura.Backends.Postgres.Translate.Select.Internal.Helpers (selectToSelectWith, toQuery)
|
|
import Hasura.Backends.Postgres.Translate.Select.Internal.Process (processAnnSimpleSelect)
|
|
import Hasura.Backends.Postgres.Translate.Types
|
|
import Hasura.Prelude
|
|
import Hasura.RQL.IR.Select
|
|
( AnnSelectG (_asnStrfyNum),
|
|
AnnSimpleSelect,
|
|
)
|
|
import Hasura.RQL.Types.Backend (Backend)
|
|
import Hasura.RQL.Types.BackendType (BackendType (Postgres))
|
|
import Hasura.RQL.Types.Common
|
|
( FieldName (FieldName),
|
|
JsonAggSelect,
|
|
)
|
|
|
|
-- | Translates IR to Postgres queries for simple SELECTs (select queries that
|
|
-- are not aggregations, including subscriptions).
|
|
--
|
|
-- See 'mkSQLSelect' for the Postgres AST.
|
|
selectQuerySQL ::
|
|
forall pgKind.
|
|
(Backend ('Postgres pgKind), PostgresAnnotatedFieldJSON pgKind) =>
|
|
JsonAggSelect ->
|
|
AnnSimpleSelect ('Postgres pgKind) ->
|
|
Query
|
|
selectQuerySQL jsonAggSelect =
|
|
toQuery
|
|
. selectToSelectWith
|
|
. mkSQLSelect jsonAggSelect
|
|
|
|
mkSQLSelect ::
|
|
forall pgKind m.
|
|
( Backend ('Postgres pgKind),
|
|
PostgresAnnotatedFieldJSON pgKind,
|
|
MonadWriter CustomSQLCTEs m
|
|
) =>
|
|
JsonAggSelect ->
|
|
AnnSimpleSelect ('Postgres pgKind) ->
|
|
m S.Select
|
|
mkSQLSelect jsonAggSelect annSel = do
|
|
let permLimitSubQuery = PLSQNotRequired
|
|
-- run an intermediary step in order to obtain:
|
|
-- SelectSource: the primary source, along with its where clause and sorting/slicing information
|
|
-- NodeExtractors: a map from aliases which need to be selected to the corresponding required SQLExp to select the value
|
|
-- : the join tree required for relationships (built via @MonadWriter@)
|
|
-- : any top-level Common Table Expressions needed for Native Queries
|
|
((selectSource, nodeExtractors), SelectWriter {_swJoinTree = joinTree, _swCustomSQLCTEs = customSQLCTEs}) =
|
|
runWriter
|
|
$ flip runReaderT strfyNum
|
|
$ flip evalStateT initialNativeQueryFreshIdStore
|
|
$ processAnnSimpleSelect sourcePrefixes rootFldName permLimitSubQuery annSel
|
|
|
|
selectNode = SelectNode nodeExtractors joinTree
|
|
topExtractor =
|
|
asJsonAggExtr jsonAggSelect rootFldAls permLimitSubQuery
|
|
$ orderByForJsonAgg selectSource
|
|
arrayNode = MultiRowSelectNode [topExtractor] selectNode
|
|
tell customSQLCTEs
|
|
|
|
pure $ generateSQLSelectFromArrayNode selectSource arrayNode $ S.BELit True
|
|
where
|
|
strfyNum = _asnStrfyNum annSel
|
|
rootFldIdentifier = toIdentifier rootFldName
|
|
sourcePrefixes = SourcePrefixes rootFldIdentifier rootFldIdentifier
|
|
rootFldName = FieldName "root"
|
|
rootFldAls = S.toColumnAlias $ toIdentifier rootFldName
|