5757from cassandra .cluster import Cluster as ClusterDriver
5858from cassandra .cluster import NoHostAvailable
5959from cassandra .policies import RetryPolicy
60- from cassandra .policies import WhiteListRoundRobinPolicy , HostFilterPolicy , RoundRobinPolicy
60+ from cassandra .policies import WhiteListRoundRobinPolicy
6161from cassandra .query import SimpleStatement
6262from argus .common .enums import ResourceState
6363from argus .client .sct .types import LogLink
@@ -2874,10 +2874,6 @@ def print_node_running_nemesis(self, node_ip):
28742874
28752875 return f' ({ node .running_nemesis } nemesis target node)' if node .running_nemesis else ' (not target node)'
28762876
2877- @property
2878- def is_cqlsh_support_cloud_bundle (self ):
2879- return bool (self .parent_cluster .connection_bundle_file )
2880-
28812877 @property
28822878 def is_replacement_by_host_id_supported (self ):
28832879 return ComparableScyllaVersion (self .scylla_version ) > '5.2.0~dev'
@@ -2899,12 +2895,6 @@ def _gen_cqlsh_cmd(self, command, keyspace, timeout, connect_timeout):
28992895 command = '"{}"' .format (command .strip ().replace ('"' , '\\ "' ))
29002896
29012897 cqlsh_cmd = self .add_install_prefix ('/usr/bin/cqlsh' )
2902- if self .is_cqlsh_support_cloud_bundle :
2903- connection_bundle_file = self .parent_cluster .connection_bundle_file
2904- target_connection_bundle_file = str (Path ('/tmp/' ) / connection_bundle_file .name )
2905- self .remoter .send_files (str (connection_bundle_file ), target_connection_bundle_file )
2906-
2907- return f'{ cqlsh_cmd } { options } -e { command } --cloudconf { target_connection_bundle_file } '
29082898 return f'{ cqlsh_cmd } { options } -e { command } { host } '
29092899
29102900 def run_cqlsh (self , cmd , keyspace = None , timeout = 120 , verbose = True , split = False , connect_timeout = 60 ,
@@ -3594,7 +3584,7 @@ def create_ssl_context(keyfile: str, certfile: str, truststore: str):
35943584 return ssl_context
35953585
35963586 def _create_session (self , node , keyspace , user , password , compression , protocol_version , load_balancing_policy = None , port = None , # noqa: PLR0913
3597- ssl_context = None , node_ips = None , connect_timeout = None , verbose = True , connection_bundle_file = None ):
3587+ ssl_context = None , node_ips = None , connect_timeout = None , verbose = True ):
35983588 if not port :
35993589 port = node .CQL_PORT
36003590
@@ -3620,8 +3610,6 @@ def _create_session(self, node, keyspace, user, password, compression, protocol_
36203610 self .log .debug ("ssl_context: %s" , str (ssl_context ))
36213611
36223612 kwargs = dict (contact_points = node_ips , port = port , ssl_context = ssl_context )
3623- if connection_bundle_file :
3624- kwargs = dict (scylla_cloud = connection_bundle_file )
36253613 cluster_driver = ClusterDriver (auth_provider = auth_provider ,
36263614 compression = compression ,
36273615 protocol_version = protocol_version ,
@@ -3671,40 +3659,23 @@ def cql_connection(self, node, keyspace=None, user=None,
36713659 - If a connection bundle file is available in the parent cluster, it will be used for the connection.
36723660 - If no connection bundle file is provided, the method will use the WhiteListRoundRobinPolicy with the specified nodes.
36733661 """
3674- if connection_bundle_file := node .parent_cluster .connection_bundle_file :
3675- wlrr = None
3676- node_ips = []
3677- else :
3678- node_ips = self .get_node_cql_ips (nodes = whitelist_nodes )
3679- wlrr = WhiteListRoundRobinPolicy (node_ips )
3662+ node_ips = self .get_node_cql_ips (nodes = whitelist_nodes )
3663+ wlrr = WhiteListRoundRobinPolicy (node_ips )
3664+
36803665 return self ._create_session (node = node , keyspace = keyspace , user = user , password = password , compression = compression ,
36813666 protocol_version = protocol_version , load_balancing_policy = wlrr , port = port , ssl_context = ssl_context ,
3682- node_ips = node_ips , connect_timeout = connect_timeout , verbose = verbose ,
3683- connection_bundle_file = connection_bundle_file )
3667+ node_ips = node_ips , connect_timeout = connect_timeout , verbose = verbose )
36843668
36853669 def cql_connection_exclusive (self , node , keyspace = None , user = None ,
36863670 password = None , compression = True ,
36873671 protocol_version = None , port = None ,
36883672 ssl_context = None , connect_timeout = 100 , verbose = True ):
3689- if connection_bundle_file := node .parent_cluster .connection_bundle_file :
3690- # TODO: handle the case of multiple datacenters
3691- bundle_yaml = yaml .safe_load (connection_bundle_file .open ('r' , encoding = 'utf-8' ))
3692- node_domain = None
3693- for _ , connection_data in bundle_yaml .get ('datacenters' , {}).items ():
3694- node_domain = connection_data .get ('nodeDomain' ).strip ()
3695- assert node_domain , f"didn't found nodeDomain in bundle [{ connection_bundle_file } ]"
3696-
3697- def host_filter (host ):
3698- return str (host .host_id ) == str (node .host_id ) or node_domain == host .endpoint ._server_name
3699- wlrr = HostFilterPolicy (child_policy = RoundRobinPolicy (), predicate = host_filter )
3700- node_ips = []
3701- else :
3702- node_ips = [node .cql_address ]
3703- wlrr = WhiteListRoundRobinPolicy (node_ips )
3673+
3674+ node_ips = [node .cql_address ]
3675+ wlrr = WhiteListRoundRobinPolicy (node_ips )
37043676 return self ._create_session (node = node , keyspace = keyspace , user = user , password = password , compression = compression ,
37053677 protocol_version = protocol_version , load_balancing_policy = wlrr , port = port , ssl_context = ssl_context ,
3706- node_ips = node_ips , connect_timeout = connect_timeout , verbose = verbose ,
3707- connection_bundle_file = connection_bundle_file )
3678+ node_ips = node_ips , connect_timeout = connect_timeout , verbose = verbose )
37083679
37093680 @retrying (n = 8 , sleep_time = 15 , allowed_exceptions = (NoHostAvailable ,))
37103681 def cql_connection_patient (self , node , keyspace = None ,
@@ -4172,11 +4143,6 @@ def proposed_scylla_yaml(self) -> ScyllaYaml:
41724143 )
41734144 return ScyllaYaml (** cluster_params_builder .dict (exclude_unset = True , exclude_none = True ))
41744145
4175- @property
4176- def connection_bundle_file (self ) -> Path | None :
4177- bundle_file = self .params .get ("k8s_connection_bundle_file" )
4178- return Path (bundle_file ) if bundle_file else None
4179-
41804146 @property
41814147 def racks (self ) -> Set [int ]:
41824148 return {node .rack for node in self .nodes }
0 commit comments