Exit with non-zero code when job failed or being canceled (#219)

GoogleCloudPlatform · Apr 9, 2020 · cef9f6c · cef9f6c · Mrart · Apr 15, 2020
1 parent 3f4001a
commit cef9f6c
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/controllers/flinkcluster_submit_job_script.go b/controllers/flinkcluster_submit_job_script.go
@@ -66,9 +66,17 @@ function wait_for_job() {
 	while true; do
 		echo -e "\nWaiting for job to finish..."
 		list_jobs
-		if list_jobs | grep -e "(FINISHED)" -e "(FAILED)" -e "(CANCELED)"; then
-			echo -e "\nJob has terminated, exiting..."
-			break
+		if list_jobs | grep "(FINISHED)"; then
+			echo -e "\nJob has completed successfully, exiting 0"
+			return 0
+		fi
+		if list_jobs | grep "(FAILED)"; then
+			echo -e "\nJob failed, exiting 1"
+			return 1
+		fi
+		if list_jobs | grep "(CANCELED)"; then
+			echo -e "\nJob has been cancelled, exiting 2"
+			return 2
 		fi
 		sleep 30
 	done