mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-17 04:24:35 +03:00
58ba01f31c
* Improve performance of replace_metadata on tracking tables. Closes #3802 On the 1000 table track case this went from >20min to 8 seconds, the bottleneck being all the former calls (for each table) to buildSchemaCache. * refactor replace metadata Only save metadata in hdb_catalog schema and build schema cache (strictly) * remove `withBuildSchemaCache` function Co-authored-by: rakeshkky <12475069+rakeshkky@users.noreply.github.com> Co-authored-by: Vamshi Surabhi <0x777@users.noreply.github.com>
124 lines
3.8 KiB
Haskell
124 lines
3.8 KiB
Haskell
module Hasura.RQL.DDL.RemoteSchema
|
|
( runAddRemoteSchema
|
|
, runRemoveRemoteSchema
|
|
, removeRemoteSchemaFromCatalog
|
|
, runReloadRemoteSchema
|
|
, fetchRemoteSchemas
|
|
, addRemoteSchemaP1
|
|
, addRemoteSchemaP2Setup
|
|
, addRemoteSchemaToCatalog
|
|
) where
|
|
|
|
import Hasura.EncJSON
|
|
import Hasura.Prelude
|
|
|
|
import qualified Data.Aeson as J
|
|
import qualified Data.HashMap.Strict as Map
|
|
import qualified Data.HashSet as S
|
|
import qualified Database.PG.Query as Q
|
|
|
|
import Hasura.GraphQL.RemoteServer
|
|
import Hasura.RQL.Types
|
|
import Hasura.Server.Version (HasVersion)
|
|
import Hasura.SQL.Types
|
|
|
|
runAddRemoteSchema
|
|
:: ( HasVersion
|
|
, QErrM m
|
|
, CacheRWM m
|
|
, MonadTx m
|
|
, MonadIO m
|
|
, HasHttpManager m
|
|
)
|
|
=> AddRemoteSchemaQuery -> m EncJSON
|
|
runAddRemoteSchema q = do
|
|
addRemoteSchemaP1 name
|
|
addRemoteSchemaP2 q
|
|
buildSchemaCacheFor $ MORemoteSchema name
|
|
pure successMsg
|
|
where
|
|
name = _arsqName q
|
|
|
|
addRemoteSchemaP1
|
|
:: (QErrM m, CacheRM m)
|
|
=> RemoteSchemaName -> m ()
|
|
addRemoteSchemaP1 name = do
|
|
remoteSchemaMap <- scRemoteSchemas <$> askSchemaCache
|
|
onJust (Map.lookup name remoteSchemaMap) $ const $
|
|
throw400 AlreadyExists $ "remote schema with name "
|
|
<> name <<> " already exists"
|
|
|
|
addRemoteSchemaP2Setup
|
|
:: (HasVersion, QErrM m, MonadIO m, HasHttpManager m)
|
|
=> AddRemoteSchemaQuery -> m RemoteSchemaCtx
|
|
addRemoteSchemaP2Setup (AddRemoteSchemaQuery name def _) = do
|
|
httpMgr <- askHttpManager
|
|
rsi <- validateRemoteSchemaDef def
|
|
gCtx <- fetchRemoteSchema httpMgr name rsi
|
|
pure $ RemoteSchemaCtx name gCtx rsi
|
|
|
|
addRemoteSchemaP2
|
|
:: (HasVersion, MonadTx m, MonadIO m, HasHttpManager m) => AddRemoteSchemaQuery -> m ()
|
|
addRemoteSchemaP2 q = do
|
|
void $ addRemoteSchemaP2Setup q
|
|
liftTx $ addRemoteSchemaToCatalog q
|
|
|
|
runRemoveRemoteSchema
|
|
:: (QErrM m, UserInfoM m, CacheRWM m, MonadTx m)
|
|
=> RemoteSchemaNameQuery -> m EncJSON
|
|
runRemoveRemoteSchema (RemoteSchemaNameQuery rsn) = do
|
|
removeRemoteSchemaP1 rsn
|
|
liftTx $ removeRemoteSchemaFromCatalog rsn
|
|
withNewInconsistentObjsCheck buildSchemaCache
|
|
pure successMsg
|
|
|
|
removeRemoteSchemaP1
|
|
:: (UserInfoM m, QErrM m, CacheRM m)
|
|
=> RemoteSchemaName -> m ()
|
|
removeRemoteSchemaP1 rsn = do
|
|
sc <- askSchemaCache
|
|
let rmSchemas = scRemoteSchemas sc
|
|
void $ onNothing (Map.lookup rsn rmSchemas) $
|
|
throw400 NotExists "no such remote schema"
|
|
|
|
runReloadRemoteSchema
|
|
:: (QErrM m, CacheRWM m)
|
|
=> RemoteSchemaNameQuery -> m EncJSON
|
|
runReloadRemoteSchema (RemoteSchemaNameQuery name) = do
|
|
remoteSchemas <- getAllRemoteSchemas <$> askSchemaCache
|
|
unless (name `elem` remoteSchemas) $ throw400 NotExists $
|
|
"remote schema with name " <> name <<> " does not exist"
|
|
|
|
let invalidations = mempty { ciRemoteSchemas = S.singleton name }
|
|
withNewInconsistentObjsCheck $ buildSchemaCacheWithOptions CatalogUpdate invalidations
|
|
pure successMsg
|
|
|
|
addRemoteSchemaToCatalog
|
|
:: AddRemoteSchemaQuery
|
|
-> Q.TxE QErr ()
|
|
addRemoteSchemaToCatalog (AddRemoteSchemaQuery name def comment) =
|
|
Q.unitQE defaultTxErrorHandler [Q.sql|
|
|
INSERT into hdb_catalog.remote_schemas
|
|
(name, definition, comment)
|
|
VALUES ($1, $2, $3)
|
|
|] (name, Q.AltJ $ J.toJSON def, comment) True
|
|
|
|
removeRemoteSchemaFromCatalog :: RemoteSchemaName -> Q.TxE QErr ()
|
|
removeRemoteSchemaFromCatalog name =
|
|
Q.unitQE defaultTxErrorHandler [Q.sql|
|
|
DELETE FROM hdb_catalog.remote_schemas
|
|
WHERE name = $1
|
|
|] (Identity name) True
|
|
|
|
fetchRemoteSchemas :: Q.TxE QErr [AddRemoteSchemaQuery]
|
|
fetchRemoteSchemas =
|
|
map fromRow <$> Q.listQE defaultTxErrorHandler
|
|
[Q.sql|
|
|
SELECT name, definition, comment
|
|
FROM hdb_catalog.remote_schemas
|
|
ORDER BY name ASC
|
|
|] () True
|
|
where
|
|
fromRow (name, Q.AltJ def, comment) =
|
|
AddRemoteSchemaQuery name def comment
|