From 638cc0061e064e9f8d7d3f8535e0598af7ee18ca Mon Sep 17 00:00:00 2001 From: karezche <64801825+karenc-bq@users.noreply.github.com> Date: Fri, 24 Apr 2026 12:58:03 -0700 Subject: [PATCH] fix: add pg_catalog to gdb status queries --- aws_advanced_python_wrapper/database_dialect.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/aws_advanced_python_wrapper/database_dialect.py b/aws_advanced_python_wrapper/database_dialect.py index 86c74acb..7b3d2f27 100644 --- a/aws_advanced_python_wrapper/database_dialect.py +++ b/aws_advanced_python_wrapper/database_dialect.py @@ -628,15 +628,15 @@ def get_host_list_provider_supplier(self, plugin_service: PluginService) -> Call class GlobalAuroraPgDialect(AuroraPgDialect, GlobalAuroraTopologyDialect): - _GLOBAL_STATUS_TABLE_EXISTS_QUERY = "select 'aurora_global_db_status'::regproc" - _GLOBAL_INSTANCE_STATUS_EXISTS_QUERY = "select 'aurora_global_db_instance_status'::regproc" + _GLOBAL_STATUS_TABLE_EXISTS_QUERY = "select 'pg_catalog.aurora_global_db_status'::regproc" + _GLOBAL_INSTANCE_STATUS_EXISTS_QUERY = "select 'pg_catalog.aurora_global_db_instance_status'::regproc" _TOPOLOGY_QUERY = \ ("SELECT SERVER_ID, CASE WHEN SESSION_ID = 'MASTER_SESSION_ID' THEN TRUE ELSE FALSE END, " "VISIBILITY_LAG_IN_MSEC, AWS_REGION " - "FROM aurora_global_db_instance_status()") - _REGION_COUNT_QUERY = "SELECT count(1) FROM aurora_global_db_status()" + "FROM pg_catalog.aurora_global_db_instance_status()") + _REGION_COUNT_QUERY = "SELECT count(1) FROM pg_catalog.aurora_global_db_status()" _REGION_BY_INSTANCE_ID_QUERY = \ - "SELECT AWS_REGION FROM aurora_global_db_instance_status() WHERE SERVER_ID = %s" + "SELECT AWS_REGION FROM pg_catalog.aurora_global_db_instance_status() WHERE SERVER_ID = %s" @property def dialect_update_candidates(self) -> Optional[Tuple[DialectCode, ...]]: