def execute(self, context):
eks_hook = EKSHook(
aws_conn_id=self.aws_conn_id,
region_name=self.region,
)
if self.force_delete_compute:
self.delete_any_nodegroups(eks_hook)
self.delete_any_fargate_profiles(eks_hook)
eks_hook.delete_cluster(name=self.cluster_name)
def execute(self, context):
eks_hook = EKSHook(
aws_conn_id=self.aws_conn_id,
region_name=self.region,
)
eks_hook.create_cluster(
name=self.cluster_name,
roleArn=self.cluster_role_arn,
resourcesVpcConfig=self.resources_vpc_config,
)
if not self.compute:
return None
self.log.info("Waiting for EKS Cluster to provision. This will take some time.")
countdown = TIMEOUT_SECONDS
while eks_hook.get_cluster_state(clusterName=self.cluster_name) != ClusterStates.ACTIVE:
if countdown >= CHECK_INTERVAL_SECONDS:
countdown -= CHECK_INTERVAL_SECONDS
self.log.info(
"Waiting for cluster to start. Checking again in %d seconds", CHECK_INTERVAL_SECONDS
)
sleep(CHECK_INTERVAL_SECONDS)
else:
message = (
"Cluster is still inactive after the allocated time limit. "
"Failed cluster will be torn down."
)
self.log.error(message)
# If there is something preventing the cluster for activating, tear it down and abort.
eks_hook.delete_cluster(name=self.cluster_name)
raise RuntimeError(message)
if self.compute == 'nodegroup':
eks_hook.create_nodegroup(
clusterName=self.cluster_name,
nodegroupName=self.nodegroup_name,
subnets=self.resources_vpc_config.get('subnetIds'),
nodeRole=self.nodegroup_role_arn,
)
elif self.compute == 'fargate':
eks_hook.create_fargate_profile(
clusterName=self.cluster_name,
fargateProfileName=self.fargate_profile_name,
podExecutionRoleArn=self.fargate_pod_execution_role_arn,
selectors=self.fargate_selectors,
)
def execute(self, context):
eks_hook = EKSHook(
aws_conn_id=self.aws_conn_id,
region_name=self.region,
)
if self.force_delete_compute:
nodegroups = eks_hook.list_nodegroups(
clusterName=self.cluster_name)
nodegroup_count = len(nodegroups)
if nodegroup_count:
self.log.info(
"A cluster can not be deleted with attached nodegroups. Deleting %d nodegroups.",
nodegroup_count,
)
for group in nodegroups:
eks_hook.delete_nodegroup(clusterName=self.cluster_name,
nodegroupName=group)
# Scaling up the timeout based on the number of nodegroups that are being processed.
additional_seconds = 5 * 60
countdown = TIMEOUT_SECONDS + (nodegroup_count *
additional_seconds)
while eks_hook.list_nodegroups(clusterName=self.cluster_name):
if countdown >= CHECK_INTERVAL_SECONDS:
countdown -= CHECK_INTERVAL_SECONDS
sleep(CHECK_INTERVAL_SECONDS)
self.log.info(
"Waiting for the remaining %s nodegroups to delete. "
"Checking again in %d seconds.",
nodegroup_count,
CHECK_INTERVAL_SECONDS,
)
else:
raise RuntimeError(
"Nodegroups are still inactive after the allocated time limit. Aborting."
)
self.log.info("No nodegroups remain, deleting cluster.")
eks_hook.delete_cluster(name=self.cluster_name)
