mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-18 04:51:35 +03:00
3cef692dd7
PR-URL: https://github.com/hasura/graphql-engine-mono/pull/8649 GitOrigin-RevId: 4bb9311d1d7ab4e8ee641bde5df2babcb1b8b306
281 lines
12 KiB
Haskell
281 lines
12 KiB
Haskell
{-# LANGUAGE QuasiQuotes #-}
|
|
{-# LANGUAGE UndecidableInstances #-}
|
|
{-# OPTIONS_GHC -fno-warn-orphans #-}
|
|
|
|
-- | Postgres Instances Metadata
|
|
--
|
|
-- Defines a 'Hasura.RQL.Types.Metadata.Backend.BackendMetadata' type class instance for Postgres.
|
|
module Hasura.Backends.Postgres.Instances.Metadata () where
|
|
|
|
import Data.HashMap.Strict qualified as Map
|
|
import Data.HashMap.Strict.InsOrd qualified as InsOrd
|
|
import Data.String.Interpolate (i)
|
|
import Data.Text.Extended
|
|
import Database.PG.Query.PTI qualified as PTI
|
|
import Database.PG.Query.Pool (fromPGTxErr)
|
|
import Database.PG.Query.Transaction (Query)
|
|
import Database.PG.Query.Transaction qualified as Query
|
|
import Database.PostgreSQL.LibPQ qualified as PQ
|
|
import Hasura.Backends.Postgres.DDL qualified as Postgres
|
|
import Hasura.Backends.Postgres.Execute.Types (runPgSourceReadTx)
|
|
import Hasura.Backends.Postgres.Instances.LogicalModels as Postgres (validateLogicalModel)
|
|
import Hasura.Backends.Postgres.SQL.Types (QualifiedObject (..), QualifiedTable)
|
|
import Hasura.Backends.Postgres.SQL.Types qualified as Postgres
|
|
import Hasura.Backends.Postgres.Types.CitusExtraTableMetadata
|
|
import Hasura.Base.Error
|
|
import Hasura.Prelude
|
|
import Hasura.RQL.Types.Backend (Backend)
|
|
import Hasura.RQL.Types.Metadata.Backend
|
|
import Hasura.RQL.Types.Relationships.Local
|
|
import Hasura.RQL.Types.SchemaCache (askSourceConfig)
|
|
import Hasura.RQL.Types.Table
|
|
import Hasura.SQL.Backend
|
|
|
|
--------------------------------------------------------------------------------
|
|
-- PostgresMetadata
|
|
|
|
-- | We differentiate the handling of metadata between Citus and Vanilla
|
|
-- Postgres because Citus imposes limitations on the types of joins that it
|
|
-- permits, which then limits the types of relations that we can track.
|
|
class PostgresMetadata (pgKind :: PostgresKind) where
|
|
-- TODO: find a better name
|
|
validateRel ::
|
|
MonadError QErr m =>
|
|
TableCache ('Postgres pgKind) ->
|
|
QualifiedTable ->
|
|
Either (ObjRelDef ('Postgres pgKind)) (ArrRelDef ('Postgres pgKind)) ->
|
|
m ()
|
|
|
|
-- | A query for getting the list of all tables on a given data source. This
|
|
-- is primarily used by the console to display tables for tracking etc.
|
|
listAllTablesSql :: Query
|
|
|
|
-- | A mapping from pg scalar types with clear oid equivalent to oid.
|
|
--
|
|
-- This is a insert order hash map so that when we invert it
|
|
-- duplicate oids will point to a more "general" type.
|
|
pgTypeOidMapping :: InsOrd.InsOrdHashMap Postgres.PGScalarType PQ.Oid
|
|
pgTypeOidMapping =
|
|
InsOrd.fromList $
|
|
[ (Postgres.PGSmallInt, PTI.int2),
|
|
(Postgres.PGSerial, PTI.int4),
|
|
(Postgres.PGInteger, PTI.int4),
|
|
(Postgres.PGBigSerial, PTI.int8),
|
|
(Postgres.PGBigInt, PTI.int8),
|
|
(Postgres.PGFloat, PTI.float4),
|
|
(Postgres.PGDouble, PTI.float8),
|
|
(Postgres.PGMoney, PTI.numeric),
|
|
(Postgres.PGNumeric, PTI.numeric),
|
|
(Postgres.PGBoolean, PTI.bool),
|
|
(Postgres.PGChar, PTI.bpchar),
|
|
(Postgres.PGVarchar, PTI.varchar),
|
|
(Postgres.PGText, PTI.text),
|
|
(Postgres.PGDate, PTI.date),
|
|
(Postgres.PGTimeStamp, PTI.timestamp),
|
|
(Postgres.PGTimeStampTZ, PTI.timestamptz),
|
|
(Postgres.PGTimeTZ, PTI.timetz),
|
|
(Postgres.PGJSON, PTI.json),
|
|
(Postgres.PGJSONB, PTI.jsonb),
|
|
(Postgres.PGUUID, PTI.uuid),
|
|
(Postgres.PGArray Postgres.PGSmallInt, PTI.int2_array),
|
|
(Postgres.PGArray Postgres.PGSerial, PTI.int4_array),
|
|
(Postgres.PGArray Postgres.PGInteger, PTI.int4_array),
|
|
(Postgres.PGArray Postgres.PGBigSerial, PTI.int8_array),
|
|
(Postgres.PGArray Postgres.PGBigInt, PTI.int8_array),
|
|
(Postgres.PGArray Postgres.PGFloat, PTI.float4_array),
|
|
(Postgres.PGArray Postgres.PGDouble, PTI.float8_array),
|
|
(Postgres.PGArray Postgres.PGMoney, PTI.numeric_array),
|
|
(Postgres.PGArray Postgres.PGNumeric, PTI.numeric_array),
|
|
(Postgres.PGArray Postgres.PGBoolean, PTI.bool_array),
|
|
(Postgres.PGArray Postgres.PGChar, PTI.char_array),
|
|
(Postgres.PGArray Postgres.PGVarchar, PTI.varchar_array),
|
|
(Postgres.PGArray Postgres.PGText, PTI.text_array),
|
|
(Postgres.PGArray Postgres.PGDate, PTI.date_array),
|
|
(Postgres.PGArray Postgres.PGTimeStamp, PTI.timestamp_array),
|
|
(Postgres.PGArray Postgres.PGTimeStampTZ, PTI.timestamptz_array),
|
|
(Postgres.PGArray Postgres.PGTimeTZ, PTI.timetz_array),
|
|
(Postgres.PGArray Postgres.PGJSON, PTI.json_array),
|
|
(Postgres.PGArray Postgres.PGJSON, PTI.jsonb_array),
|
|
(Postgres.PGArray Postgres.PGUUID, PTI.uuid_array)
|
|
]
|
|
|
|
instance PostgresMetadata 'Vanilla where
|
|
validateRel _ _ _ = pure ()
|
|
|
|
listAllTablesSql =
|
|
Query.fromText
|
|
[i|
|
|
WITH partitions as (
|
|
SELECT array(
|
|
SELECT
|
|
child.relname AS partition
|
|
FROM pg_inherits
|
|
JOIN pg_class child ON pg_inherits.inhrelid = child.oid
|
|
JOIN pg_namespace nmsp_child ON nmsp_child.oid = child.relnamespace
|
|
) as names
|
|
)
|
|
SELECT info_schema.table_name, info_schema.table_schema
|
|
FROM information_schema.tables as info_schema, partitions
|
|
WHERE
|
|
info_schema.table_schema NOT IN ('information_schema', 'pg_catalog', 'hdb_catalog', '_timescaledb_internal')
|
|
AND NOT (info_schema.table_name = ANY (partitions.names))
|
|
|]
|
|
|
|
instance PostgresMetadata 'Citus where
|
|
validateRel ::
|
|
forall m.
|
|
MonadError QErr m =>
|
|
TableCache ('Postgres 'Citus) ->
|
|
QualifiedTable ->
|
|
Either (ObjRelDef ('Postgres 'Citus)) (ArrRelDef ('Postgres 'Citus)) ->
|
|
m ()
|
|
validateRel tableCache sourceTable relInfo = do
|
|
sourceTableInfo <- lookupTableInfo sourceTable
|
|
case relInfo of
|
|
Left (RelDef _ obj _) ->
|
|
case obj of
|
|
RUFKeyOn (SameTable _) -> pure ()
|
|
RUFKeyOn (RemoteTable targetTable _) -> checkObjectRelationship sourceTableInfo targetTable
|
|
RUManual RelManualConfig {} -> pure ()
|
|
Right (RelDef _ obj _) ->
|
|
case obj of
|
|
RUFKeyOn (ArrRelUsingFKeyOn targetTable _col) -> checkArrayRelationship sourceTableInfo targetTable
|
|
RUManual RelManualConfig {} -> pure ()
|
|
where
|
|
lookupTableInfo tableName =
|
|
Map.lookup tableName tableCache
|
|
`onNothing` throw400 NotFound ("no such table " <>> tableName)
|
|
|
|
checkObjectRelationship sourceTableInfo targetTable = do
|
|
targetTableInfo <- lookupTableInfo targetTable
|
|
let notSupported = throwNotSupportedError sourceTableInfo targetTableInfo "object"
|
|
case ( _tciExtraTableMetadata $ _tiCoreInfo sourceTableInfo,
|
|
_tciExtraTableMetadata $ _tiCoreInfo targetTableInfo
|
|
) of
|
|
(Distributed {}, Local) -> notSupported
|
|
(Distributed {}, Reference) -> pure ()
|
|
(Distributed {}, Distributed {}) -> pure ()
|
|
(_, Distributed {}) -> notSupported
|
|
(_, _) -> pure ()
|
|
|
|
checkArrayRelationship sourceTableInfo targetTable = do
|
|
targetTableInfo <- lookupTableInfo targetTable
|
|
let notSupported = throwNotSupportedError sourceTableInfo targetTableInfo "array"
|
|
case ( _tciExtraTableMetadata $ _tiCoreInfo sourceTableInfo,
|
|
_tciExtraTableMetadata $ _tiCoreInfo targetTableInfo
|
|
) of
|
|
(Distributed {}, Distributed {}) -> pure ()
|
|
(Distributed {}, _) -> notSupported
|
|
(_, Distributed {}) -> notSupported
|
|
(_, _) -> pure ()
|
|
|
|
showDistributionType :: ExtraTableMetadata -> Text
|
|
showDistributionType = \case
|
|
Local -> "local"
|
|
Distributed _ -> "distributed"
|
|
Reference -> "reference"
|
|
|
|
throwNotSupportedError :: TableInfo ('Postgres 'Citus) -> TableInfo ('Postgres 'Citus) -> Text -> m ()
|
|
throwNotSupportedError sourceTableInfo targetTableInfo t =
|
|
let tciSrc = _tiCoreInfo sourceTableInfo
|
|
tciTgt = _tiCoreInfo targetTableInfo
|
|
in throw400
|
|
NotSupported
|
|
( showDistributionType (_tciExtraTableMetadata tciSrc)
|
|
<> " tables ("
|
|
<> toTxt (_tciName tciSrc)
|
|
<> ") cannot have an "
|
|
<> t
|
|
<> " relationship against a "
|
|
<> showDistributionType (_tciExtraTableMetadata $ _tiCoreInfo targetTableInfo)
|
|
<> " table ("
|
|
<> toTxt (_tciName tciTgt)
|
|
<> ")"
|
|
)
|
|
|
|
listAllTablesSql =
|
|
Query.fromText
|
|
[i|
|
|
WITH partitions as (
|
|
SELECT array(
|
|
SELECT
|
|
child.relname AS partition
|
|
FROM pg_inherits
|
|
JOIN pg_class child ON pg_inherits.inhrelid = child.oid
|
|
JOIN pg_namespace nmsp_child ON nmsp_child.oid = child.relnamespace
|
|
) as names
|
|
)
|
|
SELECT info_schema.table_name, info_schema.table_schema
|
|
FROM information_schema.tables as info_schema, partitions
|
|
WHERE
|
|
info_schema.table_schema NOT IN ('pg_catalog', 'citus', 'information_schema', 'columnar', 'columnar_internal', 'guest', 'INFORMATION_SCHEMA', 'sys', 'db_owner', 'db_securityadmin', 'db_accessadmin', 'db_backupoperator', 'db_ddladmin', 'db_datawriter', 'db_datareader', 'db_denydatawriter', 'db_denydatareader', 'hdb_catalog', '_timescaledb_internal')
|
|
AND NOT (info_schema.table_name = ANY (partitions.names))
|
|
AND info_schema.table_name NOT IN ('citus_tables')
|
|
|]
|
|
|
|
instance PostgresMetadata 'Cockroach where
|
|
validateRel _ _ _ = pure ()
|
|
|
|
pgTypeOidMapping =
|
|
InsOrd.fromList
|
|
[ (Postgres.PGInteger, PTI.int8),
|
|
(Postgres.PGSerial, PTI.int8),
|
|
(Postgres.PGJSON, PTI.jsonb)
|
|
]
|
|
`InsOrd.union` pgTypeOidMapping @'Vanilla
|
|
|
|
listAllTablesSql =
|
|
Query.fromText
|
|
[i|
|
|
WITH partitions as (
|
|
SELECT array(
|
|
SELECT
|
|
child.relname AS partition
|
|
FROM pg_inherits
|
|
JOIN pg_class child ON pg_inherits.inhrelid = child.oid
|
|
JOIN pg_namespace nmsp_child ON nmsp_child.oid = child.relnamespace
|
|
) as names
|
|
)
|
|
SELECT info_schema.table_name, info_schema.table_schema
|
|
FROM information_schema.tables as info_schema, partitions
|
|
WHERE
|
|
info_schema.table_schema NOT IN ('pg_catalog', 'crdb_internal', 'information_schema', 'columnar', 'guest', 'INFORMATION_SCHEMA', 'sys', 'db_owner', 'db_securityadmin', 'db_accessadmin', 'db_backupoperator', 'db_ddladmin', 'db_datawriter', 'db_datareader', 'db_denydatawriter', 'db_denydatareader', 'hdb_catalog', '_timescaledb_internal', 'pg_extension')
|
|
AND NOT (info_schema.table_name = ANY (partitions.names));
|
|
|]
|
|
|
|
----------------------------------------------------------------
|
|
-- BackendMetadata instance
|
|
|
|
instance
|
|
( Backend ('Postgres pgKind),
|
|
PostgresMetadata pgKind,
|
|
Postgres.FetchTableMetadata pgKind,
|
|
Postgres.FetchFunctionMetadata pgKind,
|
|
Postgres.ToMetadataFetchQuery pgKind
|
|
) =>
|
|
BackendMetadata ('Postgres pgKind)
|
|
where
|
|
prepareCatalog = Postgres.prepareCatalog
|
|
buildComputedFieldInfo = Postgres.buildComputedFieldInfo
|
|
fetchAndValidateEnumValues = Postgres.fetchAndValidateEnumValues
|
|
resolveSourceConfig = Postgres.resolveSourceConfig
|
|
resolveDatabaseMetadata = Postgres.resolveDatabaseMetadata
|
|
parseBoolExpOperations = Postgres.parseBoolExpOperations
|
|
buildFunctionInfo = Postgres.buildFunctionInfo
|
|
updateColumnInEventTrigger = Postgres.updateColumnInEventTrigger
|
|
parseCollectableType = Postgres.parseCollectableType
|
|
postDropSourceHook = Postgres.postDropSourceHook
|
|
validateRelationship = validateRel @pgKind
|
|
buildComputedFieldBooleanExp = Postgres.buildComputedFieldBooleanExp
|
|
validateLogicalModel = Postgres.validateLogicalModel (pgTypeOidMapping @pgKind)
|
|
supportsBeingRemoteRelationshipTarget _ = True
|
|
|
|
listAllTables _ sourceName = do
|
|
sourceConfig <- askSourceConfig @('Postgres pgKind) sourceName
|
|
|
|
results <-
|
|
runPgSourceReadTx sourceConfig (Query.multiQE fromPGTxErr (listAllTablesSql @pgKind))
|
|
`onLeftM` \err -> throwError (prefixQErr "failed to fetch source tables: " err)
|
|
|
|
pure [QualifiedObject {..} | (qName, qSchema) <- results]
|