improve cleanup

mafredri · mafredri · commit d0fbd2ce98b6 · 2024-01-24T15:25:28.000+02:00
diff --git a/scaletest/templates/scaletest-runner/scripts/cleanup.sh b/scaletest/templates/scaletest-runner/scripts/cleanup.sh
@@ -12,32 +12,49 @@ if [[ -z $event ]]; then
 	event=manual
 fi
 
-if [[ $event = manual ]]; then
-	echo -n 'WARNING: This will clean up all scaletest resources, continue? (y/n) '
-	read -r -n 1
-	if [[ $REPLY != [yY] ]]; then
-		echo $'\nAborting...'
-		exit 1
-	fi
-fi
-
-if [[ $event != shutdown_scale_down_only ]]; then
+do_cleanup() {
 	start_phase "Cleanup (${event})"
 	coder exp scaletest cleanup \
 		--cleanup-job-timeout 2h \
-		--cleanup-timeout 5h \
-		| tee "${SCALETEST_RESULTS_DIR}/cleanup-${event}.txt"
+		--cleanup-timeout 5h |
+		tee "${SCALETEST_RESULTS_DIR}/cleanup-${event}.txt"
 	end_phase
-fi
+}
 
-if [[ $event != prepare ]]; then
-	start_phase "Scale down provisioners"
+do_scaledown() {
+	start_phase "Scale down provisioners (${event})"
 	maybedryrun "$DRY_RUN" kubectl scale deployment/coder-provisioner --replicas 1
 	maybedryrun "$DRY_RUN" kubectl rollout status deployment/coder-provisioner
 	end_phase
-fi
+}
+
+case "${event}" in
+manual)
+	echo -n 'WARNING: This will clean up all scaletest resources, continue? (y/n) '
+	read -r -n 1
+	if [[ $REPLY != [yY] ]]; then
+		echo $'\nAborting...'
+		exit 1
+	fi
+
+	do_cleanup
+	do_scaledown
 
-if [[ $event = manual ]]; then
 	echo 'Press any key to continue...'
 	read -s -r -n 1
-fi
+	;;
+prepare)
+	do_cleanup
+	;;
+shutdown)
+	do_cleanup
+	do_scaledown
+	;;
+shutdown_scale_down_only)
+	do_scaledown
+	;;
+*)
+	echo "Unknown event: ${event}" >&2
+	exit 1
+	;;
+esac
