mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-17 12:31:52 +03:00
95 lines
3.9 KiB
Haskell
95 lines
3.9 KiB
Haskell
-- | Functions for loading and modifying the catalog. See the module documentation for
|
|
-- "Hasura.RQL.DDL.Schema" for more details.
|
|
module Hasura.RQL.DDL.Schema.Catalog
|
|
( fetchCatalogData
|
|
, saveTableToCatalog
|
|
, updateTableIsEnumInCatalog
|
|
, updateTableConfig
|
|
, deleteTableFromCatalog
|
|
, getTableConfig
|
|
, purgeDependentObject
|
|
) where
|
|
|
|
import Hasura.Prelude
|
|
|
|
import qualified Data.Text as T
|
|
import qualified Database.PG.Query as Q
|
|
|
|
import Data.Aeson
|
|
|
|
import Hasura.Db
|
|
import Hasura.RQL.DDL.ComputedField
|
|
import Hasura.RQL.DDL.EventTrigger
|
|
import Hasura.RQL.DDL.Permission.Internal
|
|
import Hasura.RQL.DDL.Relationship
|
|
import Hasura.RQL.DDL.Schema.Function
|
|
import Hasura.RQL.Types.Catalog
|
|
import Hasura.RQL.Types.Common
|
|
import Hasura.RQL.Types.Error
|
|
import Hasura.RQL.Types.SchemaCache
|
|
import Hasura.SQL.Types
|
|
|
|
import Debug.Trace
|
|
|
|
fetchCatalogData :: (MonadTx m) => m CatalogMetadata
|
|
fetchCatalogData = do
|
|
liftTx . liftIO $ traceEventIO "START fetch"
|
|
liftTx . liftIO $ traceEventIO "START fetch/query"
|
|
metadataBytes <- (liftTx $ runIdentity . Q.getRow <$> Q.withQE defaultTxErrorHandler $(Q.sqlFromFile "src-rsr/catalog_metadata.sql") () True)
|
|
liftTx . liftIO $ traceEventIO "STOP fetch/query"
|
|
liftTx . liftIO $ traceEventIO "START fetch/decode"
|
|
let !decodedValue = force (eitherDecodeStrict' metadataBytes)
|
|
liftTx . liftIO $ traceEventIO "STOP fetch/decode"
|
|
liftTx . liftIO $ traceEventIO "STOP fetch"
|
|
decodedValue `onLeft` \err -> throw500 (T.pack err)
|
|
|
|
saveTableToCatalog :: (MonadTx m) => QualifiedTable -> SystemDefined -> Bool -> TableConfig -> m ()
|
|
saveTableToCatalog (QualifiedObject sn tn) systemDefined isEnum config = liftTx $
|
|
Q.unitQE defaultTxErrorHandler [Q.sql|
|
|
INSERT INTO "hdb_catalog"."hdb_table"
|
|
(table_schema, table_name, is_system_defined, is_enum, configuration)
|
|
VALUES ($1, $2, $3, $4, $5)
|
|
|] (sn, tn, systemDefined, isEnum, configVal) False
|
|
where
|
|
configVal = Q.AltJ $ toJSON config
|
|
|
|
updateTableIsEnumInCatalog :: (MonadTx m) => QualifiedTable -> Bool -> m ()
|
|
updateTableIsEnumInCatalog (QualifiedObject sn tn) isEnum = liftTx $
|
|
Q.unitQE defaultTxErrorHandler [Q.sql|
|
|
UPDATE "hdb_catalog"."hdb_table" SET is_enum = $3
|
|
WHERE table_schema = $1 AND table_name = $2
|
|
|] (sn, tn, isEnum) False
|
|
|
|
updateTableConfig :: (MonadTx m) => QualifiedTable -> TableConfig -> m ()
|
|
updateTableConfig (QualifiedObject sn tn) config = liftTx $
|
|
Q.unitQE defaultTxErrorHandler [Q.sql|
|
|
UPDATE "hdb_catalog"."hdb_table"
|
|
SET configuration = $1
|
|
WHERE table_schema = $2 AND table_name = $3
|
|
|] (configVal, sn, tn) False
|
|
where
|
|
configVal = Q.AltJ $ toJSON config
|
|
|
|
deleteTableFromCatalog :: (MonadTx m) => QualifiedTable -> m ()
|
|
deleteTableFromCatalog (QualifiedObject sn tn) = liftTx $ Q.unitQE defaultTxErrorHandler [Q.sql|
|
|
DELETE FROM "hdb_catalog"."hdb_table"
|
|
WHERE table_schema = $1 AND table_name = $2
|
|
|] (sn, tn) False
|
|
|
|
getTableConfig :: MonadTx m => QualifiedTable -> m TableConfig
|
|
getTableConfig (QualifiedObject sn tn) = liftTx $
|
|
Q.getAltJ . runIdentity . Q.getRow <$> Q.withQE defaultTxErrorHandler
|
|
[Q.sql|
|
|
SELECT configuration::json FROM hdb_catalog.hdb_table
|
|
WHERE table_schema = $1 AND table_name = $2
|
|
|] (sn, tn) True
|
|
|
|
purgeDependentObject :: (MonadTx m) => SchemaObjId -> m ()
|
|
purgeDependentObject = \case
|
|
SOTableObj tn (TOPerm rn pt) -> liftTx $ dropPermFromCatalog tn rn pt
|
|
SOTableObj qt (TORel rn) -> liftTx $ delRelFromCatalog qt rn
|
|
SOFunction qf -> liftTx $ delFunctionFromCatalog qf
|
|
SOTableObj _ (TOTrigger trn) -> liftTx $ delEventTriggerFromCatalog trn
|
|
SOTableObj qt (TOComputedField ccn) -> dropComputedFieldFromCatalog qt ccn
|
|
schemaObjId -> throw500 $ "unexpected dependent object: " <> reportSchemaObj schemaObjId
|