mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-17 20:41:49 +03:00
11a454c2d6
This commit applies ormolu to the whole Haskell code base by running `make format`. For in-flight branches, simply merging changes from `main` will result in merge conflicts. To avoid this, update your branch using the following instructions. Replace `<format-commit>` by the hash of *this* commit. $ git checkout my-feature-branch $ git merge <format-commit>^ # and resolve conflicts normally $ make format $ git commit -a -m "reformat with ormolu" $ git merge -s ours post-ormolu https://github.com/hasura/graphql-engine-mono/pull/2404 GitOrigin-RevId: 75049f5c12f430c615eafb4c6b8e83e371e01c8e
62 lines
2.1 KiB
Haskell
62 lines
2.1 KiB
Haskell
module Hasura.Backends.Postgres.Translate.Mutation
|
|
( mkSelectExpFromColumnValues,
|
|
)
|
|
where
|
|
|
|
import Data.HashMap.Strict qualified as Map
|
|
import Data.Text.Extended
|
|
import Hasura.Backends.Postgres.SQL.DML qualified as S
|
|
import Hasura.Backends.Postgres.SQL.Types
|
|
import Hasura.Backends.Postgres.SQL.Value
|
|
import Hasura.Backends.Postgres.Types.Column
|
|
import Hasura.Base.Error
|
|
import Hasura.Prelude
|
|
import Hasura.RQL.Types
|
|
import Hasura.SQL.Types
|
|
|
|
-- | Note:- Using sorted columns is necessary to enable casting the rows returned by VALUES expression to table type.
|
|
-- For example, let's consider the table, `CREATE TABLE test (id serial primary key, name text not null, age int)`.
|
|
-- The generated values expression should be in order of columns;
|
|
-- `SELECT ("row"::table).* VALUES (1, 'Robert', 23) AS "row"`.
|
|
mkSelectExpFromColumnValues ::
|
|
forall pgKind m.
|
|
MonadError QErr m =>
|
|
QualifiedTable ->
|
|
[ColumnInfo ('Postgres pgKind)] ->
|
|
[ColumnValues ('Postgres pgKind) TxtEncodedVal] ->
|
|
m S.Select
|
|
mkSelectExpFromColumnValues qt allCols = \case
|
|
[] -> return selNoRows
|
|
colVals -> do
|
|
tuples <- mapM mkTupsFromColVal colVals
|
|
let fromItem = S.FIValues (S.ValuesExp tuples) (S.Alias rowAlias) Nothing
|
|
return
|
|
S.mkSelect
|
|
{ S.selExtr = [extractor],
|
|
S.selFrom = Just $ S.FromExp [fromItem]
|
|
}
|
|
where
|
|
rowAlias = Identifier "row"
|
|
extractor = S.selectStar' $ S.QualifiedIdentifier rowAlias $ Just $ S.TypeAnn $ toSQLTxt qt
|
|
sortedCols = sortCols allCols
|
|
mkTupsFromColVal colVal =
|
|
fmap S.TupleExp $
|
|
forM sortedCols $ \ci -> do
|
|
let pgCol = pgiColumn ci
|
|
val <-
|
|
onNothing (Map.lookup pgCol colVal) $
|
|
throw500 $ "column " <> pgCol <<> " not found in returning values"
|
|
pure $ txtEncodedToSQLExp (pgiType ci) val
|
|
|
|
selNoRows =
|
|
S.mkSelect
|
|
{ S.selExtr = [S.selectStar],
|
|
S.selFrom = Just $ S.mkSimpleFromExp qt,
|
|
S.selWhere = Just $ S.WhereFrag $ S.BELit False
|
|
}
|
|
|
|
txtEncodedToSQLExp colTy = \case
|
|
TENull -> S.SENull
|
|
TELit textValue ->
|
|
S.withTyAnn (unsafePGColumnToBackend colTy) $ S.SELit textValue
|