Skip to content

Commit ff47fd8

Browse files
committed
fix: bugs in host list provider
1 parent 7d8e9e8 commit ff47fd8

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

aws_advanced_python_wrapper/host_list_provider.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -199,6 +199,8 @@ def _initialize(self):
199199
else:
200200
self._cluster_instance_template = HostInfo(
201201
host=self._rds_utils.get_rds_instance_host_pattern(self._initial_host_info.host),
202+
host_id=self._initial_host_info.host_id,
203+
port=self._initial_host_info.port,
202204
host_availability_strategy=host_availability_strategy)
203205
self._validate_host_pattern(self._cluster_instance_template.host)
204206

@@ -216,14 +218,14 @@ def _initialize(self):
216218
self._cluster_id = cluster_id_suggestion.cluster_id
217219
self._is_primary_cluster_id = cluster_id_suggestion.is_primary_cluster_id
218220
else:
219-
cluster_url = self._rds_utils.get_rds_cluster_host_url(self._initial_host_info.url)
221+
cluster_url = self._rds_utils.get_rds_cluster_host_url(self._initial_host_info.host)
220222
if cluster_url is not None:
221-
self._cluster_id = cluster_url
223+
self._cluster_id = f"{cluster_url}:{self._cluster_instance_template.port}" if self._cluster_instance_template.is_port_specified() else cluster_url
222224
self._is_primary_cluster_id = True
223225
self._is_primary_cluster_id_cache.put(self._cluster_id, True,
224226
self._suggested_cluster_id_refresh_ns)
225227

226-
self._is_initialized = True
228+
self._is_initialized = True
227229

228230
def _validate_host_pattern(self, host: str):
229231
if not self._rds_utils.is_dns_pattern_valid(host):

0 commit comments

Comments
 (0)