Skip to content

Commit da90f4f

Browse files
authored
fix: gdb initial connection support (#1223)
1 parent 5d69eb0 commit da90f4f

3 files changed

Lines changed: 4 additions & 2 deletions

File tree

aws_advanced_python_wrapper/aurora_initial_connection_strategy_plugin.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def connect(self, target_driver_func: Callable, driver_dialect: DriverDialect, h
5454
if not url_type.is_rds_cluster:
5555
return connect_func()
5656

57-
if url_type == RdsUrlType.RDS_WRITER_CLUSTER:
57+
if url_type == RdsUrlType.RDS_WRITER_CLUSTER or url_type == RdsUrlType.RDS_GLOBAL_WRITER_CLUSTER:
5858
writer_candidate_conn: Optional[Connection] = self._get_verified_writer_connection(props, is_initial_connection, connect_func)
5959
if writer_candidate_conn is None:
6060
return connect_func()

aws_advanced_python_wrapper/simple_read_write_splitting_plugin.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,7 @@ def get_verified_initial_connection(
118118

119119
if (
120120
url_type == RdsUrlType.RDS_WRITER_CLUSTER
121+
or url_type == RdsUrlType.RDS_GLOBAL_WRITER_CLUSTER
121122
or self._verify_initial_connection_type == HostRole.WRITER
122123
):
123124
conn = self._get_verified_connection(host_info, HostRole.WRITER, plugin_service_connect_func, connect_func)

aws_advanced_python_wrapper/stale_dns_plugin.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,8 @@ def get_verified_connection(self, is_initial_connection: bool, host_list_provide
5858
:param connect_func:
5959
:return:
6060
"""
61-
if not self._rds_helper.is_writer_cluster_dns(host_info.host):
61+
if not self._rds_helper.is_writer_cluster_dns(host_info.host) \
62+
and not self._rds_helper.is_global_db_writer_cluster_dns(host_info.host):
6263
return connect_func()
6364

6465
conn: Connection = connect_func()

0 commit comments

Comments
 (0)