mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-21 06:21:39 +03:00
2ff8aa8f7d
PR-URL: https://github.com/hasura/graphql-engine-mono/pull/5550 GitOrigin-RevId: f53f88b64617500353bce5a699b89fcbb591d80f
56 lines
2.2 KiB
Haskell
56 lines
2.2 KiB
Haskell
-- | This module defines the type class 'PostgresAnnotatedFieldJSON'.
|
|
module Hasura.Backends.Postgres.Translate.Select.AnnotatedFieldJSON
|
|
( PostgresAnnotatedFieldJSON (..),
|
|
)
|
|
where
|
|
|
|
import Data.Text qualified as T
|
|
import Hasura.Backends.Postgres.SQL.DML qualified as S
|
|
import Hasura.Backends.Postgres.Translate.Select.Internal.Helpers (withJsonBuildObj)
|
|
import Hasura.Prelude
|
|
import Hasura.RQL.Types.Common (FieldName (getFieldNameTxt))
|
|
import Hasura.SQL.Backend (PostgresKind (..))
|
|
|
|
class PostgresAnnotatedFieldJSON (pgKind :: PostgresKind) where
|
|
annRowToJson :: FieldName -> [(FieldName, S.SQLExp)] -> (S.ColumnAlias, S.SQLExp)
|
|
|
|
instance PostgresAnnotatedFieldJSON 'Vanilla where
|
|
annRowToJson = pgAnnRowToJson
|
|
|
|
instance PostgresAnnotatedFieldJSON 'Citus where
|
|
annRowToJson fieldAlias fieldExps =
|
|
-- Due to the restrictions Citus imposes on joins between tables of various
|
|
-- distribution types we cannot use row_to_json and have to only rely on
|
|
-- json_build_object.
|
|
withJsonBuildObj fieldAlias $ concatMap toJsonBuildObjectExps fieldExps
|
|
where
|
|
toJsonBuildObjectExps (fieldName, fieldExp) =
|
|
[S.SELit $ getFieldNameTxt fieldName, fieldExp]
|
|
|
|
instance PostgresAnnotatedFieldJSON 'Cockroach where
|
|
annRowToJson = pgAnnRowToJson
|
|
|
|
pgAnnRowToJson :: FieldName -> [(FieldName, S.SQLExp)] -> (S.ColumnAlias, S.SQLExp)
|
|
pgAnnRowToJson fieldAlias fieldExps =
|
|
-- postgres ignores anything beyond 63 chars for an iden
|
|
-- in this case, we'll need to use json_build_object function
|
|
-- json_build_object is slower than row_to_json hence it is only
|
|
-- used when needed
|
|
if any ((> 63) . T.length . getFieldNameTxt . fst) fieldExps
|
|
then withJsonBuildObj fieldAlias $ concatMap toJsonBuildObjectExps fieldExps
|
|
else withRowToJSON fieldAlias $ map toRowToJsonExtr fieldExps
|
|
where
|
|
toJsonBuildObjectExps (fieldName, fieldExp) =
|
|
[S.SELit $ getFieldNameTxt fieldName, fieldExp]
|
|
|
|
toRowToJsonExtr (fieldName, fieldExp) =
|
|
S.Extractor fieldExp $ Just $ S.toColumnAlias fieldName
|
|
|
|
-- uses row_to_json to build a json object
|
|
withRowToJSON ::
|
|
FieldName -> [S.Extractor] -> (S.ColumnAlias, S.SQLExp)
|
|
withRowToJSON parAls extrs =
|
|
(S.toColumnAlias parAls, jsonRow)
|
|
where
|
|
jsonRow = S.applyRowToJson extrs
|