Skip to content

Commit 9a1a4d9

Browse files
authored
feat: add support for blue/green deployments (#911)
1 parent c13fd4f commit 9a1a4d9

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

45 files changed

+6739
-1317
lines changed

aws_advanced_python_wrapper/aurora_initial_connection_strategy_plugin.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ def _get_verified_writer_connection(self, props: Properties, is_initial_connecti
8686
self._plugin_service.force_refresh_host_list(writer_candidate_conn)
8787
writer_candidate = self._plugin_service.identify_connection(writer_candidate_conn)
8888

89-
if writer_candidate is not None and writer_candidate.role != HostRole.WRITER:
89+
if writer_candidate is None or writer_candidate.role != HostRole.WRITER:
9090
self._close_connection(writer_candidate_conn)
9191
self._delay(retry_delay_ms)
9292
continue
@@ -133,6 +133,11 @@ def _get_verified_reader_connection(self, props: Properties, is_initial_connecti
133133
self._plugin_service.force_refresh_host_list(reader_candidate_conn)
134134
reader_candidate = self._plugin_service.identify_connection(reader_candidate_conn)
135135

136+
if reader_candidate is None:
137+
self._close_connection(reader_candidate_conn)
138+
self._delay(retry_delay_ms)
139+
continue
140+
136141
if reader_candidate is not None and reader_candidate.role != HostRole.READER:
137142
if self._has_no_readers():
138143
# Cluster has no readers. Simulate Aurora reader cluster endpoint logic and return the current writer connection.

0 commit comments

Comments
 (0)