5656from cassandra .cluster import Cluster as ClusterDriver
5757from cassandra .cluster import NoHostAvailable
5858from cassandra .policies import RetryPolicy
59- from cassandra .policies import WhiteListRoundRobinPolicy , HostFilterPolicy , RoundRobinPolicy , RackAwareRoundRobinPolicy , LoadBalancingPolicy
59+ from cassandra .policies import WhiteListRoundRobinPolicy , RackAwareRoundRobinPolicy , LoadBalancingPolicy
6060from cassandra .query import SimpleStatement
6161from argus .common .enums import ResourceState
6262from argus .client .sct .types import LogLink
@@ -2937,10 +2937,6 @@ def print_node_running_nemesis(self, node_ip):
29372937
29382938 return f' ({ node .running_nemesis } nemesis target node)' if node .running_nemesis else ' (not target node)'
29392939
2940- @property
2941- def is_cqlsh_support_cloud_bundle (self ):
2942- return bool (self .parent_cluster .connection_bundle_file )
2943-
29442940 @property
29452941 def is_replacement_by_host_id_supported (self ):
29462942 return ComparableScyllaVersion (self .scylla_version ) > '5.2.0~dev'
@@ -2962,12 +2958,6 @@ def _gen_cqlsh_cmd(self, command, keyspace, timeout, connect_timeout):
29622958 command = '"{}"' .format (command .strip ().replace ('"' , '\\ "' ))
29632959
29642960 cqlsh_cmd = self .add_install_prefix ('/usr/bin/cqlsh' )
2965- if self .is_cqlsh_support_cloud_bundle :
2966- connection_bundle_file = self .parent_cluster .connection_bundle_file
2967- target_connection_bundle_file = str (Path ('/tmp/' ) / connection_bundle_file .name )
2968- self .remoter .send_files (str (connection_bundle_file ), target_connection_bundle_file )
2969-
2970- return f'{ cqlsh_cmd } { options } -e { command } --cloudconf { target_connection_bundle_file } '
29712961 return f'{ cqlsh_cmd } { options } -e { command } { host } '
29722962
29732963 def run_cqlsh (self , cmd , keyspace = None , timeout = 120 , verbose = True , split = False , connect_timeout = 60 ,
@@ -3677,7 +3667,7 @@ def create_ssl_context(keyfile: str, certfile: str, truststore: str):
36773667 return ssl_context
36783668
36793669 def _create_session (self , node , keyspace , user , password , compression , protocol_version , load_balancing_policy = None , port = None , # noqa: PLR0913
3680- ssl_context = None , node_ips = None , connect_timeout = None , verbose = True , connection_bundle_file = None ):
3670+ ssl_context = None , node_ips = None , connect_timeout = None , verbose = True ):
36813671 if not port :
36823672 port = node .CQL_PORT
36833673
@@ -3703,8 +3693,6 @@ def _create_session(self, node, keyspace, user, password, compression, protocol_
37033693 self .log .debug ("ssl_context: %s" , str (ssl_context ))
37043694
37053695 kwargs = dict (contact_points = node_ips , port = port , ssl_context = ssl_context )
3706- if connection_bundle_file :
3707- kwargs = dict (scylla_cloud = connection_bundle_file )
37083696 cluster_driver = ClusterDriver (auth_provider = auth_provider ,
37093697 compression = compression ,
37103698 protocol_version = protocol_version ,
@@ -3798,43 +3786,24 @@ def cql_connection(self, node, keyspace=None, user=None,
37983786 - If a connection bundle file is available in the parent cluster, it will be used for the connection.
37993787 - If no connection bundle file is provided, the method will use the WhiteListRoundRobinPolicy with the specified nodes.
38003788 """
3801- if connection_bundle_file := node .parent_cluster .connection_bundle_file :
3802- wlrr = None
3803- node_ips = []
3804- else :
3805- wlrr , node_ips = self .get_load_balancing_policy (whitelist_nodes = whitelist_nodes )
38063789
3790+ wlrr , node_ips = self .get_load_balancing_policy (whitelist_nodes = whitelist_nodes )
38073791 return self ._create_session (node = node , keyspace = keyspace , user = user , password = password , compression = compression ,
38083792 protocol_version = protocol_version , load_balancing_policy = wlrr , port = port , ssl_context = ssl_context ,
3809- node_ips = node_ips , connect_timeout = connect_timeout , verbose = verbose ,
3810- connection_bundle_file = connection_bundle_file )
3793+ node_ips = node_ips , connect_timeout = connect_timeout , verbose = verbose )
38113794
38123795 def cql_connection_exclusive (self , node , keyspace = None , user = None ,
38133796 password = None , compression = True ,
38143797 protocol_version = None , port = None ,
38153798 ssl_context = None , connect_timeout = 100 , verbose = True ):
3816- if connection_bundle_file := node .parent_cluster .connection_bundle_file :
3817- # TODO: handle the case of multiple datacenters
3818- bundle_yaml = yaml .safe_load (connection_bundle_file .open ('r' , encoding = 'utf-8' ))
3819- node_domain = None
3820- for _ , connection_data in bundle_yaml .get ('datacenters' , {}).items ():
3821- node_domain = connection_data .get ('nodeDomain' ).strip ()
3822- assert node_domain , f"didn't found nodeDomain in bundle [{ connection_bundle_file } ]"
3823-
3824- def host_filter (host ):
3825- return str (host .host_id ) == str (node .host_id ) or node_domain == host .endpoint ._server_name
3826- wlrr = HostFilterPolicy (child_policy = RoundRobinPolicy (), predicate = host_filter )
3827- node_ips = []
3828- else :
3829- # Use WhiteListRoundRobinPolicy with a single node IP.
3830- # RackAwareRoundRobinPolicy is not applicable for exclusive node connections,
3831- # as it operates based on rack and datacenter, not individual nodes.
3832- node_ips = [node .cql_address ]
3833- wlrr = WhiteListRoundRobinPolicy (node_ips )
3799+ # Use WhiteListRoundRobinPolicy with a single node IP.
3800+ # RackAwareRoundRobinPolicy is not applicable for exclusive node connections,
3801+ # as it operates based on rack and datacenter, not individual nodes.
3802+ node_ips = [node .cql_address ]
3803+ wlrr = WhiteListRoundRobinPolicy (node_ips )
38343804 return self ._create_session (node = node , keyspace = keyspace , user = user , password = password , compression = compression ,
38353805 protocol_version = protocol_version , load_balancing_policy = wlrr , port = port , ssl_context = ssl_context ,
3836- node_ips = node_ips , connect_timeout = connect_timeout , verbose = verbose ,
3837- connection_bundle_file = connection_bundle_file )
3806+ node_ips = node_ips , connect_timeout = connect_timeout , verbose = verbose )
38383807
38393808 @retrying (n = 8 , sleep_time = 15 , allowed_exceptions = (NoHostAvailable ,))
38403809 def cql_connection_patient (self , node , keyspace = None ,
@@ -4316,11 +4285,6 @@ def proposed_scylla_yaml(self) -> ScyllaYaml:
43164285 )
43174286 return ScyllaYaml (** cluster_params_builder .model_dump (exclude_unset = True , exclude_none = True ))
43184287
4319- @property
4320- def connection_bundle_file (self ) -> Path | None :
4321- bundle_file = self .params .get ("k8s_connection_bundle_file" )
4322- return Path (bundle_file ) if bundle_file else None
4323-
43244288 @property
43254289 def racks (self ) -> Set [int ]:
43264290 return {node .rack for node in self .nodes }
0 commit comments