Add a new cleanup job for GKE, AKS, and EKS clusters

This job ensures clusters without the 'protected' label are deleted. Signed-off-by: Shuyang Xin <[email protected]>
antrea-io · Jan 17, 2025 · eff86f7 · eff86f7
1 parent 71cd6e0
commit eff86f7
Show file tree

Hide file tree

Showing 3 changed files with 101 additions and 3 deletions.
diff --git a/ci/test-conformance-aks.sh b/ci/test-conformance-aks.sh
@@ -34,7 +34,7 @@ KUBE_CONFORMANCE_IMAGE_VERSION=auto
 
 _usage="Usage: $0 [--cluster-name <AKSClusterNameToUse>] [--kubeconfig <KubeconfigSavePath>] [--k8s-version <ClusterVersion>]\
                   [--azure-app-id <AppID>] [--azure-tenant-id <TenantID>] [--azure-password <Password>] \
-                  [--aks-region <Region>] [--log-mode <SonobuoyResultLogLevel>] [--setup-only] [--cleanup-only]
+                  [--aks-region <Region>] [--log-mode <SonobuoyResultLogLevel>] [--setup-only] [--cleanup-only] [--cleanup-all]
 
 Setup a AKS cluster to run K8s e2e community tests (Conformance & Network Policy).
 
@@ -47,7 +47,8 @@ Setup a AKS cluster to run K8s e2e community tests (Conformance & Network Policy
         --aks-region             The Azure region where the cluster will be initiated. Defaults to westus.
         --log-mode               Use the flag to set either 'report', 'detail', or 'dump' level data for sonobuoy results.
         --setup-only             Only perform setting up the cluster and run test.
-        --cleanup-only           Only perform cleaning up the cluster."
+        --cleanup-only           Only perform cleaning up the cluster.
+        --cleanup-all            Cleaning up all clusters without protected tag."
 
 function print_usage {
     echoerr "$_usage"
@@ -104,6 +105,10 @@ case $key in
     RUN_ALL=false
     shift
     ;;
+    --cleanup-all)
+    RUN_CLEANUP_ALL=true
+    shift
+    ;;
     -h|--help)
     print_usage
     exit 0
@@ -300,6 +305,29 @@ function cleanup_cluster() {
     echo "=== Cleanup cluster ${CLUSTER} succeeded ==="
 }
 
+function cleanup_all_clusters() {
+    echo '=== Cleaning up all AKS clusters without tag protected ==='
+    clusters=$(az aks list \
+      --query "[!(tags.protected && tags.protected=='true') && resourceGroup=='${RESOURCE_GROUP}'].{name:name,rg:resourceGroup}" \
+      -o tsv)
+    if [[ -z "$clusters" ]]; then
+      echo "Unprotected cluster not found."
+      exit
+    fi
+    while read -r clusterName resourceGroup; do
+        [[ -z "$clusterName" ]] && continue
+        echo "Deleting Cluster: $clusterName in $resourceGroup"
+        az aks delete --name "$clusterName" --resource-group "$resourceGroup" --yes
+    done <<< "$clusters"
+    resource=$(az aks list \
+      --query "[resourceGroup=='${RESOURCE_GROUP}'].{name:name,rg:resourceGroup}" \
+      -o tsv)
+    if [[ -z "$resource" ]]; then
+        az group delete --name ${RESOURCE_GROUP} --yes --no-wait
+    fi
+    echo "=== Cleanup AKS clusters succeeded ==="
+}
+
 # ensures that the script can be run from anywhere
 THIS_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
 GIT_CHECKOUT_DIR=${THIS_DIR}/..
@@ -317,6 +345,10 @@ if [[ "$RUN_ALL" == true || "$RUN_CLEANUP_ONLY" == true ]]; then
     cleanup_cluster
 fi
 
+if [[ "$RUN_CLEANUP_ALL" == true ]]; then
+    cleanup_all_clusters
+fi
+
 if [[ "$RUN_CLEANUP_ONLY" == false && $TEST_SCRIPT_RC -ne 0 ]]; then
     exit 1
 fi
diff --git a/ci/test-conformance-eks.sh b/ci/test-conformance-eks.sh
@@ -40,7 +40,7 @@ AWS_SERVICE_USER_NAME=""
 _usage="Usage: $0 [--cluster-name <EKSClusterNameToUse>] [--kubeconfig <KubeconfigSavePath>] [--k8s-version <ClusterVersion>]\
                   [--aws-access-key <AccessKey>] [--aws-secret-key <SecretKey>] [--aws-region <Region>] [--aws-service-user <ServiceUserName>]\
                   [--aws-service-user-role-arn <ServiceUserRoleARN>] [--ssh-key <SSHKey] [--ssh-private-key <SSHPrivateKey] [--log-mode <SonobuoyResultLogLevel>]\
-                  [--setup-only] [--cleanup-only]
+                  [--setup-only] [--cleanup-only] [--cleanup-all]
 
 Setup a EKS cluster to run K8s e2e community tests (Conformance & Network Policy).
 
@@ -56,6 +56,7 @@ Setup a EKS cluster to run K8s e2e community tests (Conformance & Network Policy
         --log-mode                    Use the flag to set either 'report', 'detail', or 'dump' level data for sonobuoy results.
         --setup-only                  Only perform setting up the cluster and run test.
         --cleanup-only                Only perform cleaning up the cluster.
+        --cleanup-all                 Cleaning up all clusters without protected tag.
         --skip-eksctl-install         Do not install the latest eksctl version. Eksctl must be installed already."
 
 function print_usage {
@@ -125,6 +126,10 @@ case $key in
     RUN_ALL=false
     shift
     ;;
+    --cleanup-all)
+    RUN_CLEANUP_ALL=true
+    shift
+    ;;
     --skip-eksctl-install)
     INSTALL_EKSCTL=false
     shift
@@ -354,6 +359,32 @@ function cleanup_cluster() {
     echo "=== Cleanup cluster ${CLUSTER} succeeded ==="
 }
 
+function cleanup_all_clusters() {
+    echo '=== Cleaning up all EKS clusters without tag protected ==='
+    clusters=$(eksctl get cluster --output json | jq -r '.[].metadata.name')
+    if [[ -z "$clusters" ]]; then
+      echo "Unprotected cluster not found."
+      exit
+    fi
+    for cluster in $clusters; do
+        cluster_arn=$(aws eks describe-cluster --name "$cluster" --query "cluster.arn" --output text 2>/dev/null)
+        if [[ "$cluster_arn" == "None" ]] || [[ -z "$cluster_arn" ]]; then
+            log "Warning: Unable to retrieve ARN for cluster '$cluster'. Skipping this cluster."
+            continue
+        fi
+        tags=$(aws eks list-tags-for-resource --resource-arn "$cluster_arn" --query "tags" --output json 2>/dev/null)
+        if [[ $? -ne 0 ]]; then
+            echo "Warning: Unable to retrieve tags for cluster '$cluster'. Skipping this cluster."
+            continue
+        fi
+        has_tag=$(echo "$tags" | jq -r --arg key "protected" '. | has($key)')
+        if [[ "$has_tag" == "false" ]]; then
+            eksctl delete cluster --name ${cluster} --region $REGION
+        fi
+    done
+    echo "=== Cleanup cluster ${cluster} succeeded ==="
+}
+
 # ensures that the script can be run from anywhere
 THIS_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
 GIT_CHECKOUT_DIR=${THIS_DIR}/..
@@ -371,6 +402,10 @@ if [[ "$RUN_ALL" == true || "$RUN_CLEANUP_ONLY" == true ]]; then
     cleanup_cluster
 fi
 
+if [[ "$RUN_CLEANUP_ALL" == true ]]; then
+    cleanup_all_clusters
+fi
+
 if [[ "$RUN_CLEANUP_ONLY" == false && $TEST_SCRIPT_RC -ne 0 ]]; then
     exit 1
 fi
diff --git a/ci/test-conformance-gke.sh b/ci/test-conformance-gke.sh
@@ -128,6 +128,10 @@ case $key in
     RUN_ALL=false
     shift
     ;;
+    --cleanup-all)
+    RUN_CLEANUP_ALL=true
+    shift
+    ;;
     -h|--help)
     print_usage
     exit 0
@@ -329,6 +333,29 @@ function cleanup_cluster() {
     echo "=== Cleanup cluster ${CLUSTER} succeeded ==="
 }
 
+function cleanup_all_clusters() {
+    echo '=== Cleaning up all unprotected GKE clusters ==='
+    gcloud container clusters list --zone "${GKE_ZONE}" --format="json(name, location, resourceLabels)" | \
+        jq -r '.[] | select(.resourceLabels.protected != "true") | "\(.name) \(.location)"' | \
+        while read CLUSTER_NAME ZONE; do
+            echo "Deleting cluster ${CLUSTER_NAME} in zone '${GKE_ZONE}'..."
+            retry=5
+            while [[ "${retry}" -gt 0 ]]; do
+                gcloud container clusters delete ${CLUSTER_NAME} --zone ${GKE_ZONE}
+                if [[ $? -eq 0 ]]; then
+                    break
+                fi
+                sleep 10
+                retry=$((retry-1))
+            done
+            if [[ "${retry}" -eq 0 ]]; then
+                echo "=== Failed to delete GKE cluster ${CLUSTER}! ==="
+                continue
+            fi
+        done
+    echo "=== Cleanup GKE clusters succeeded ==="
+}
+
 if [[ "$RUN_ALL" == true || "$RUN_SETUP_ONLY" == true ]]; then
     setup_gke
     deliver_antrea_to_gke
@@ -339,6 +366,10 @@ if [[ "$RUN_ALL" == true || "$RUN_CLEANUP_ONLY" == true ]]; then
     cleanup_cluster
 fi
 
+if [[ "$RUN_CLEANUP_ALL" == true ]]; then
+    cleanup_all_clusters
+fi
+
 if [[ "$RUN_CLEANUP_ONLY" == false && $TEST_SCRIPT_RC -ne 0 ]]; then
     exit 1
 fi