def _resize_cluster(self, cluster_name, node_count, **kwargs):
"""Resize the cluster of a given name to given number of worker nodes.
:param str cluster_name: Name of the cluster
:param int node_count: New size of the worker nodes
"""
result = {}
cluster_api = ClusterApi(api_client=self.pks_client)
LOGGER.debug(f"Sending request to PKS:{self.pks_host_uri} to resize "
f"the cluster with name: {cluster_name} to "
f"{node_count} worker nodes")
resize_params = UpdateClusterParameters(
kubernetes_worker_instances=node_count)
try:
cluster_api.update_cluster(cluster_name, body=resize_params)
except ApiException as err:
LOGGER.debug(f"Resizing cluster {cluster_name} failed with "
f"error:\n {err}")
raise PksServerError(err.status, err.body)
LOGGER.debug(f"PKS: {self.pks_host_uri} accepted the request to resize"
f" the cluster: {cluster_name}")
result['cluster_name'] = cluster_name
result['task_status'] = 'in progress'
return result
def resize_cluster(self, name, num_worker_nodes):
"""Resize the cluster of a given name to given number of worker nodes.
:param str name: Name of the cluster
:param int num_worker_nodes: New size of the worker nodes
(should be greater than the current number).
:return: None
"""
result = {}
result['body'] = []
cluster_api = ClusterApi(api_client=self.pks_client)
LOGGER.debug(f'Sending request to PKS:{self.host} to resize the '
f'cluster with name: {name} to '
f'{num_worker_nodes} worker nodes')
resize_params = UpdateClusterParameters(
kubernetes_worker_instances=num_worker_nodes)
cluster_api.update_cluster(name, body=resize_params)
LOGGER.debug(f'PKS: {self.host} accepted the request to resize the '
f'cluster: {name}')
result['status_code'] = ACCEPTED
return result
def resize_cluster(self, **kwargs):
"""Resize the cluster of a given name to given number of worker nodes.
System administrator can resize the given cluster regardless of
who is the owner of the cluster. Other users can only resize
the cluster they own.
:return: response status
:rtype: dict
"""
data = kwargs[KwargKey.DATA]
cluster_name = data[RequestKey.CLUSTER_NAME]
num_workers = data[RequestKey.NUM_WORKERS]
qualified_cluster_name = self._append_user_id(cluster_name)
if (self.context.client.is_sysadmin()
or self.context.user.has_org_admin_rights):
cluster_info = self._get_cluster_info(data)
qualified_cluster_name = cluster_info['pks_cluster_name']
self._check_cluster_isolation(cluster_name, qualified_cluster_name)
result = {}
cluster_api = ClusterApi(api_client=self.pks_client)
self.pks_wire_logger.debug(f"Sending request to"
f" PKS:{self.pks_host_uri} to resize"
f" the cluster with name:"
f"{qualified_cluster_name} to"
f" {num_workers} worker nodes")
resize_params = \
UpdateClusterParameters(kubernetes_worker_instances=num_workers)
try:
cluster_api.update_cluster(qualified_cluster_name,
body=resize_params)
except ApiException as err:
SERVER_LOGGER.debug(f"Resizing cluster {qualified_cluster_name}"
f" failed with error:\n {err}")
raise PksServerError(err.status, err.body)
self.pks_wire_logger.debug(f"PKS: {self.pks_host_uri} accepted the"
f" request to resize the cluster: "
f" {qualified_cluster_name}")
result['name'] = qualified_cluster_name
result['task_status'] = 'in progress'
self._restore_original_name(result)
if not self.context.client.is_sysadmin():
self._filter_sensitive_pks_properties(result)
return result