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
59+ from cassandra .policies import WhiteListRoundRobinPolicy
6060from cassandra .query import SimpleStatement
6161from argus .common .enums import ResourceState
6262from argus .client .sct .types import LogLink
@@ -2882,10 +2882,6 @@ def print_node_running_nemesis(self, node_ip):
28822882
28832883 return f' ({ node .running_nemesis } nemesis target node)' if node .running_nemesis else ' (not target node)'
28842884
2885- @property
2886- def is_cqlsh_support_cloud_bundle (self ):
2887- return bool (self .parent_cluster .connection_bundle_file )
2888-
28892885 @property
28902886 def is_replacement_by_host_id_supported (self ):
28912887 return ComparableScyllaVersion (self .scylla_version ) > '5.2.0~dev'
@@ -2907,12 +2903,6 @@ def _gen_cqlsh_cmd(self, command, keyspace, timeout, connect_timeout):
29072903 command = '"{}"' .format (command .strip ().replace ('"' , '\\ "' ))
29082904
29092905 cqlsh_cmd = self .add_install_prefix ('/usr/bin/cqlsh' )
2910- if self .is_cqlsh_support_cloud_bundle :
2911- connection_bundle_file = self .parent_cluster .connection_bundle_file
2912- target_connection_bundle_file = str (Path ('/tmp/' ) / connection_bundle_file .name )
2913- self .remoter .send_files (str (connection_bundle_file ), target_connection_bundle_file )
2914-
2915- return f'{ cqlsh_cmd } { options } -e { command } --cloudconf { target_connection_bundle_file } '
29162906 return f'{ cqlsh_cmd } { options } -e { command } { host } '
29172907
29182908 def run_cqlsh (self , cmd , keyspace = None , timeout = 120 , verbose = True , split = False , connect_timeout = 60 ,
@@ -3620,7 +3610,7 @@ def create_ssl_context(keyfile: str, certfile: str, truststore: str):
36203610 return ssl_context
36213611
36223612 def _create_session (self , node , keyspace , user , password , compression , protocol_version , load_balancing_policy = None , port = None , # noqa: PLR0913
3623- ssl_context = None , node_ips = None , connect_timeout = None , verbose = True , connection_bundle_file = None ):
3613+ ssl_context = None , node_ips = None , connect_timeout = None , verbose = True ):
36243614 if not port :
36253615 port = node .CQL_PORT
36263616
@@ -3646,8 +3636,6 @@ def _create_session(self, node, keyspace, user, password, compression, protocol_
36463636 self .log .debug ("ssl_context: %s" , str (ssl_context ))
36473637
36483638 kwargs = dict (contact_points = node_ips , port = port , ssl_context = ssl_context )
3649- if connection_bundle_file :
3650- kwargs = dict (scylla_cloud = connection_bundle_file )
36513639 cluster_driver = ClusterDriver (auth_provider = auth_provider ,
36523640 compression = compression ,
36533641 protocol_version = protocol_version ,
@@ -3697,40 +3685,21 @@ def cql_connection(self, node, keyspace=None, user=None,
36973685 - If a connection bundle file is available in the parent cluster, it will be used for the connection.
36983686 - If no connection bundle file is provided, the method will use the WhiteListRoundRobinPolicy with the specified nodes.
36993687 """
3700- if connection_bundle_file := node .parent_cluster .connection_bundle_file :
3701- wlrr = None
3702- node_ips = []
3703- else :
3704- node_ips = self .get_node_cql_ips (nodes = whitelist_nodes )
3705- wlrr = WhiteListRoundRobinPolicy (node_ips )
3688+ node_ips = self .get_node_cql_ips (nodes = whitelist_nodes )
3689+ wlrr = WhiteListRoundRobinPolicy (node_ips )
37063690 return self ._create_session (node = node , keyspace = keyspace , user = user , password = password , compression = compression ,
37073691 protocol_version = protocol_version , load_balancing_policy = wlrr , port = port , ssl_context = ssl_context ,
3708- node_ips = node_ips , connect_timeout = connect_timeout , verbose = verbose ,
3709- connection_bundle_file = connection_bundle_file )
3692+ node_ips = node_ips , connect_timeout = connect_timeout , verbose = verbose )
37103693
37113694 def cql_connection_exclusive (self , node , keyspace = None , user = None ,
37123695 password = None , compression = True ,
37133696 protocol_version = None , port = None ,
37143697 ssl_context = None , connect_timeout = 100 , verbose = True ):
3715- if connection_bundle_file := node .parent_cluster .connection_bundle_file :
3716- # TODO: handle the case of multiple datacenters
3717- bundle_yaml = yaml .safe_load (connection_bundle_file .open ('r' , encoding = 'utf-8' ))
3718- node_domain = None
3719- for _ , connection_data in bundle_yaml .get ('datacenters' , {}).items ():
3720- node_domain = connection_data .get ('nodeDomain' ).strip ()
3721- assert node_domain , f"didn't found nodeDomain in bundle [{ connection_bundle_file } ]"
3722-
3723- def host_filter (host ):
3724- return str (host .host_id ) == str (node .host_id ) or node_domain == host .endpoint ._server_name
3725- wlrr = HostFilterPolicy (child_policy = RoundRobinPolicy (), predicate = host_filter )
3726- node_ips = []
3727- else :
3728- node_ips = [node .cql_address ]
3729- wlrr = WhiteListRoundRobinPolicy (node_ips )
3698+ node_ips = [node .cql_address ]
3699+ wlrr = WhiteListRoundRobinPolicy (node_ips )
37303700 return self ._create_session (node = node , keyspace = keyspace , user = user , password = password , compression = compression ,
37313701 protocol_version = protocol_version , load_balancing_policy = wlrr , port = port , ssl_context = ssl_context ,
3732- node_ips = node_ips , connect_timeout = connect_timeout , verbose = verbose ,
3733- connection_bundle_file = connection_bundle_file )
3702+ node_ips = node_ips , connect_timeout = connect_timeout , verbose = verbose )
37343703
37353704 @retrying (n = 8 , sleep_time = 15 , allowed_exceptions = (NoHostAvailable ,))
37363705 def cql_connection_patient (self , node , keyspace = None ,
@@ -4198,11 +4167,6 @@ def proposed_scylla_yaml(self) -> ScyllaYaml:
41984167 )
41994168 return ScyllaYaml (** cluster_params_builder .model_dump (exclude_unset = True , exclude_none = True ))
42004169
4201- @property
4202- def connection_bundle_file (self ) -> Path | None :
4203- bundle_file = self .params .get ("k8s_connection_bundle_file" )
4204- return Path (bundle_file ) if bundle_file else None
4205-
42064170 @property
42074171 def racks (self ) -> Set [int ]:
42084172 return {node .rack for node in self .nodes }
0 commit comments