mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-17 20:41:49 +03:00
8a77386fcf
### Description This PR adds the required IR for DB to DB joins, based on @paf31 and @0x777 's `feature/db-to-db` branch. To do so, it also refactors the IR to introduce a new type parameter, `r`, which is used to recursively constructs the `v` parameter of remote QueryDBs. When collecting remote joins, we replace `r` with `Const Void`, indicating at the type level that there cannot be any leftover remote join. Furthermore, this PR refactors IR.Select for readability, moves some code from IR.Root to IR.Select to avoid having to deal with circular dependencies, and makes it compile by adding `error` in all new cases in the execution pipeline. The diff doesn't make it clear, but most of Select.hs is actually unchanged. Declarations have just been reordered by topic, in the following order: - type declarations - instance declarations - type aliases - constructor functions - traverse functions https://github.com/hasura/graphql-engine-mono/pull/1580 Co-authored-by: Phil Freeman <630306+paf31@users.noreply.github.com> GitOrigin-RevId: bbdcb4119cec8bb3fc32f1294f91b8dea0728721
120 lines
3.8 KiB
Haskell
120 lines
3.8 KiB
Haskell
module Hasura.RQL.IR.Insert where
|
|
|
|
import Hasura.Prelude
|
|
|
|
import Data.Kind (Type)
|
|
|
|
import Hasura.RQL.IR.BoolExp
|
|
import Hasura.RQL.IR.Returning
|
|
import Hasura.RQL.Types.Backend
|
|
import Hasura.RQL.Types.Column
|
|
import Hasura.RQL.Types.Relationship
|
|
import Hasura.SQL.Backend
|
|
|
|
|
|
data AnnInsert (b :: BackendType) (r :: BackendType -> Type) v
|
|
= AnnInsert
|
|
{ _aiFieldName :: !Text
|
|
, _aiIsSingle :: !Bool
|
|
, _aiData :: !(MultiObjIns b v)
|
|
, _aiOutput :: !(MutationOutputG b r v)
|
|
}
|
|
|
|
data AnnIns (b :: BackendType) a v
|
|
= AnnIns
|
|
{ _aiInsObj :: !a
|
|
, _aiTableName :: !(TableName b)
|
|
, _aiConflictClause :: !(Maybe (ConflictClauseP1 b v))
|
|
, _aiCheckCond :: !(AnnBoolExp b v, Maybe (AnnBoolExp b v))
|
|
, _aiTableCols :: ![ColumnInfo b]
|
|
, _aiDefVals :: !(PreSetColsG b v)
|
|
}
|
|
|
|
type SingleObjIns b v = AnnIns b (AnnInsObj b v) v
|
|
type MultiObjIns b v = AnnIns b [AnnInsObj b v] v
|
|
|
|
data RelIns (b :: BackendType) a
|
|
= RelIns
|
|
{ _riAnnIns :: !a
|
|
, _riRelInfo :: !(RelInfo b)
|
|
} deriving (Show, Eq)
|
|
|
|
type ObjRelIns b v = RelIns b (SingleObjIns b v)
|
|
type ArrRelIns b v = RelIns b (MultiObjIns b v)
|
|
|
|
data AnnInsObj (b :: BackendType) v
|
|
= AnnInsObj
|
|
{ _aioColumns :: ![(Column b, v)]
|
|
, _aioObjRels :: ![ObjRelIns b v]
|
|
, _aioArrRels :: ![ArrRelIns b v]
|
|
}
|
|
|
|
|
|
instance Semigroup (AnnInsObj backend v) where
|
|
(AnnInsObj col1 obj1 rel1) <> (AnnInsObj col2 obj2 rel2) =
|
|
AnnInsObj (col1 <> col2) (obj1 <> obj2) (rel1 <> rel2)
|
|
|
|
instance Monoid (AnnInsObj backend v) where
|
|
mempty = AnnInsObj [] [] []
|
|
|
|
|
|
data ConflictTarget (b :: BackendType)
|
|
= CTColumn ![Column b]
|
|
| CTConstraint !(ConstraintName b)
|
|
deriving instance Backend b => Show (ConflictTarget b)
|
|
deriving instance Backend b => Eq (ConflictTarget b)
|
|
|
|
data ConflictClauseP1 (b :: BackendType) v
|
|
= CP1DoNothing !(Maybe (ConflictTarget b))
|
|
| CP1Update !(ConflictTarget b) ![Column b] !(PreSetColsG b v) (AnnBoolExp b v)
|
|
deriving (Functor, Foldable, Traversable)
|
|
|
|
|
|
data InsertQueryP1 (b :: BackendType)
|
|
= InsertQueryP1
|
|
{ iqp1Table :: !(TableName b)
|
|
, iqp1Cols :: ![Column b]
|
|
, iqp1Tuples :: ![[SQLExpression b]]
|
|
, iqp1Conflict :: !(Maybe (ConflictClauseP1 b (SQLExpression b)))
|
|
, iqp1CheckCond :: !(AnnBoolExpSQL b, Maybe (AnnBoolExpSQL b))
|
|
, iqp1Output :: !(MutationOutput b)
|
|
, iqp1AllCols :: ![ColumnInfo b]
|
|
}
|
|
|
|
|
|
traverseAnnInsert
|
|
:: (Applicative f, Backend backend)
|
|
=> (a -> f b)
|
|
-> AnnInsert backend r a
|
|
-> f (AnnInsert backend r b)
|
|
traverseAnnInsert f (AnnInsert fieldName isSingle annIns mutationOutput) =
|
|
AnnInsert fieldName isSingle
|
|
<$> traverseMulti annIns
|
|
<*> traverseMutationOutput f mutationOutput
|
|
where
|
|
traverseMulti (AnnIns objs tableName conflictClause checkCond columns defaultValues) = AnnIns
|
|
<$> traverse traverseObject objs
|
|
<*> pure tableName
|
|
<*> traverse (traverse f) conflictClause
|
|
<*> ( (,)
|
|
<$> traverseAnnBoolExp f (fst checkCond)
|
|
<*> traverse (traverseAnnBoolExp f) (snd checkCond)
|
|
)
|
|
<*> pure columns
|
|
<*> traverse f defaultValues
|
|
traverseSingle (AnnIns obj tableName conflictClause checkCond columns defaultValues) = AnnIns
|
|
<$> traverseObject obj
|
|
<*> pure tableName
|
|
<*> traverse (traverse f) conflictClause
|
|
<*> ( (,)
|
|
<$> traverseAnnBoolExp f (fst checkCond)
|
|
<*> traverse (traverseAnnBoolExp f) (snd checkCond)
|
|
)
|
|
<*> pure columns
|
|
<*> traverse f defaultValues
|
|
traverseObject (AnnInsObj columns objRels arrRels) = AnnInsObj
|
|
<$> traverse (traverse f) columns
|
|
<*> traverse (traverseRel traverseSingle) objRels
|
|
<*> traverse (traverseRel traverseMulti) arrRels
|
|
traverseRel z (RelIns object relInfo) = RelIns <$> z object <*> pure relInfo
|