Skip to content

Commit 34dbd16

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

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

aws_advanced_python_wrapper/host_list_provider.py

Lines changed: 6 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,15 @@ 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}" \
224+
if self._cluster_instance_template.is_port_specified() else cluster_url
222225
self._is_primary_cluster_id = True
223226
self._is_primary_cluster_id_cache.put(self._cluster_id, True,
224227
self._suggested_cluster_id_refresh_ns)
225228

226-
self._is_initialized = True
229+
self._is_initialized = True
227230

228231
def _validate_host_pattern(self, host: str):
229232
if not self._rds_utils.is_dns_pattern_valid(host):

0 commit comments

Comments
 (0)