mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-17 04:24:35 +03:00
98d6a635a1
PR-URL: https://github.com/hasura/graphql-engine-mono/pull/5036 GitOrigin-RevId: e6d076edb1a5f6664c941b84d06329ece43751cb
94 lines
2.8 KiB
Haskell
94 lines
2.8 KiB
Haskell
module Hasura.Backends.Postgres.RQLGenerator.GenSelectArgsG
|
|
( genSelectArgsG,
|
|
)
|
|
where
|
|
|
|
import Data.Int (Int64)
|
|
import Hasura.Backends.Postgres.RQLGenerator.GenAssociatedTypes
|
|
( genBooleanOperators,
|
|
genColumn,
|
|
genFunctionArgumentExp,
|
|
genFunctionName,
|
|
genScalarType,
|
|
genTableName,
|
|
genXComputedField,
|
|
)
|
|
import Hasura.Backends.Postgres.SQL.DML (NullsOrder (..), OrderType (..))
|
|
import Hasura.Generator.Common (defaultRange)
|
|
import Hasura.Prelude hiding (maybe, nonEmpty)
|
|
import Hasura.RQL.IR.BoolExp
|
|
import Hasura.RQL.IR.Generator
|
|
( genAnnBoolExp,
|
|
genAnnBoolExpFld,
|
|
genAnnotatedOrderByElement,
|
|
genAnnotatedOrderByItemG,
|
|
)
|
|
import Hasura.RQL.IR.Select (AnnotatedOrderByItemG, SelectArgsG (..))
|
|
import Hasura.RQL.Types.Backend
|
|
import Hasura.SQL.Backend
|
|
import Hedgehog (MonadGen)
|
|
import Hedgehog.Gen qualified as Gen
|
|
|
|
--------------------------------------------------------------------------------
|
|
-- Exported
|
|
|
|
genSelectArgsG :: forall m a. MonadGen m => m a -> m (SelectArgsG ('Postgres 'Vanilla) a)
|
|
genSelectArgsG genA = do
|
|
_saWhere <- where'
|
|
_saOrderBy <- orderBy
|
|
_saLimit <- limit
|
|
_saOffset <- offset
|
|
_saDistinct <- distinct
|
|
pure SelectArgs {..}
|
|
where
|
|
where' :: m (Maybe (AnnBoolExp ('Postgres 'Vanilla) a))
|
|
where' =
|
|
Gen.maybe $
|
|
genAnnBoolExp
|
|
( genAnnBoolExpFld
|
|
genColumn
|
|
genTableName
|
|
genScalarType
|
|
genFunctionName
|
|
genXComputedField
|
|
(genBooleanOperators genA)
|
|
(genFunctionArgumentExp genA)
|
|
genA
|
|
)
|
|
genTableName
|
|
|
|
orderBy :: m (Maybe (NonEmpty (AnnotatedOrderByItemG ('Postgres 'Vanilla) a)))
|
|
orderBy =
|
|
Gen.maybe . Gen.nonEmpty defaultRange $
|
|
genAnnotatedOrderByItemG @_ @('Postgres 'Vanilla)
|
|
genBasicOrderType
|
|
genNullsOrderType
|
|
( genAnnotatedOrderByElement @_ @('Postgres 'Vanilla)
|
|
genColumn
|
|
genTableName
|
|
genScalarType
|
|
genFunctionName
|
|
genXComputedField
|
|
(genBooleanOperators genA)
|
|
(genFunctionArgumentExp genA)
|
|
genA
|
|
)
|
|
|
|
limit :: m (Maybe Int)
|
|
limit = Gen.maybe $ Gen.integral defaultRange
|
|
|
|
offset :: m (Maybe Int64)
|
|
offset = Gen.maybe $ Gen.integral defaultRange
|
|
|
|
distinct :: m (Maybe (NonEmpty (Column ('Postgres 'Vanilla))))
|
|
distinct = Gen.maybe . Gen.nonEmpty defaultRange $ genColumn
|
|
|
|
--------------------------------------------------------------------------------
|
|
-- Unexported Helpers
|
|
|
|
genBasicOrderType :: MonadGen m => m (BasicOrderType ('Postgres 'Vanilla))
|
|
genBasicOrderType = Gen.element [OTAsc, OTDesc]
|
|
|
|
genNullsOrderType :: MonadGen m => m (NullsOrderType ('Postgres 'Vanilla))
|
|
genNullsOrderType = Gen.element [NullsFirst, NullsLast]
|