From 74b932506dd4e1381dcbfe175d4b9efaf61e7667 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gardais=20J=C3=A9r=C3=A9my?=
 <jeremy.gardais@univ-rennes1.fr>
Date: Fri, 5 Jun 2020 10:53:51 +0200
Subject: [PATCH] Disable script now manage multiple queues

---
 cluster/disable.host.queue.sh | 49 +++++++++++++++++++----------------
 1 file changed, 27 insertions(+), 22 deletions(-)

diff --git a/cluster/disable.host.queue.sh b/cluster/disable.host.queue.sh
index 696e507..5ed7b7b 100755
--- a/cluster/disable.host.queue.sh
+++ b/cluster/disable.host.queue.sh
@@ -6,7 +6,7 @@ readonly PROGDIR=$(readlink -m $(dirname "${0}"))
 readonly ARGS="${*}"
 readonly NBARGS="${#}"
 ## Test if DEBUG is already defined (by parent script,…)
-[ -z "${DEBUG}" ] && readonly DEBUG=0
+[ -z "${DEBUG}" ] && readonly DEBUG=1
 
 ## Colors
 readonly PURPLE='\033[1;35m'
@@ -39,14 +39,14 @@ usage() {                                                       # {{{
 	cat <<- EOF
 usage: $PROGNAME [hostname]
 
-Try to disable the SGE queue of the current host (default)
+Try to disable all SGE queues of the current host (default)
 or the one passed as first argument.
 
 EXAMPLES :
-    - Disable SGE's queue of the current host
+    - Disable SGE's queue(s) of the current host
         ${PROGNAME}
 
-    - Disable SGE's queue of the "marvin.domain.tld" host
+    - Disable SGE's queue(s) of "marvin.domain.tld" host
         ${PROGNAME} marvin.domain.tld
 	EOF
 
@@ -54,24 +54,24 @@ EXAMPLES :
 # }}}
 debug_message() {                                               # {{{
 
-	local_message="${1}"
+	local_debug_message="${1}"
 
 	## Print message if DEBUG is enable (=0)
-	[ "${DEBUG}" -eq "0" ] && printf '\e[1;35m%-6b\e[m\n' "DEBUG − ${PROGNAME} : ${local_message}"
+	[ "${DEBUG}" -eq "0" ] && printf '\e[1;35m%-6b\e[m\n' "DEBUG − ${PROGNAME} : ${local_debug_message}"
 
 }
 # }}}
 is_queue_enable() {                                            # {{{
 
-	local_sge_hostname="${1}"
-	local_sge_queue_name="${2}"
+	local_queue_enable_hostname="${1}"
+	local_queue_enable_name="${2}"
 	## List all queues with 'disable' state and filter to the expected queue name
 	## with a fake_user to avoid pending jobs for this queue
 	### And count returned lines
-	local_sge_queue_test=$(qstat -f -qs d -q "${local_sge_queue_name:=/dev/null}@${local_sge_hostname:=/dev/null}" -u fake_user \
+	local_queue_enable_test=$(qstat -f -qs d -q "${local_queue_enable_name:=/dev/null}@${local_queue_enable_hostname:=/dev/null}" -u fake_user \
 		| wc -l)
 
-	case "${local_sge_queue_test}" in
+	case "${local_queue_enable_test}" in
 		0 ) ## No result so the queue is enable
 			local_sge_queue_state="enable"
 			return_queue_enable="0"
@@ -81,14 +81,14 @@ is_queue_enable() {                                            # {{{
 			return_queue_enable="1"
 			;;
 		* ) ## Unexpected result
-			printf '%b\n' "${RED}Not able to determine the state of ${local_sge_queue_name:=/dev/null}@${local_sge_hostname:=/dev/null} queue (command return ${local_sge_queue_test} lines).${RESET}"
+			printf '%b\n' "${RED}Not able to determine the state of ${local_sge_queue_name:=/dev/null}@${local_queue_enable_hostname:=/dev/null} queue (command return ${local_queue_enable_test} lines).${RESET}"
 			exit 2
 			;;
 esac
 
 	## Simple debug message to valid current variables
 	debug_message "is_queue_enable − \
-SGE queue: ${RED}${local_sge_queue_name:=/dev/null}${COLOR_DEBUG} \
+SGE queue: ${RED}${local_queue_enable_name:=/dev/null}${COLOR_DEBUG} \
 state is: ${RED}${local_sge_queue_state:=/dev/null}${COLOR_DEBUG}."
 
 	return "${return_queue_enable}"
@@ -185,21 +185,26 @@ main() {                                                        # {{{
 SGE queue(s): ${RED}${sge_queues_name_print:=/dev/null}${COLOR_DEBUG}\
 for host: ${RED}${sge_hostname:=/dev/null}${COLOR_DEBUG}."
 
-	## If the queue are already disable
+	## If the queue(s) are already disable
 	### Exit
 	is_all_queue_disable "${sge_hostname}" "${sge_queues_name}" \
 		&& exit 0
 
-	## If the queue is enable
-	### Try to disable it
-	is_queue_enable "${sge_hostname}" "${sge_queue_name}" \
-		&& disable_sge_queue "${sge_hostname}" "${sge_queue_name}"
+	## Test all queues one by one
+	for loop_queue in ${sge_queues_name}; do
 
-	## If the queue is still enable
-	### Exit with error
-	is_queue_enable "${sge_hostname}" "${sge_queue_name}" \
-		&& printf '%b\n' "${RED}ERROR ${sge_queue_name}@${sge_hostname} is still enable.${RESET}" \
-		&& exit 4
+		## If the queue is enable
+		### Try to disable it
+		is_queue_enable "${sge_hostname}" "${loop_queue}" \
+			&& disable_sge_queue "${sge_hostname}" "${loop_queue}"
+
+		## If the queue is still enable
+		### Exit with error
+		is_queue_enable "${sge_hostname}" "${loop_queue}" \
+			&& printf '%b\n' "${RED}ERROR ${loop_queue}@${sge_hostname} is still enable.${RESET}" \
+			&& exit 4
+
+	done
 
 }
 # }}}