stop.sh

#!/usr/bin/env bash
#                    Copyright (c) 2016-2017
#   Data Intensive Applications and Systems Labaratory (DIAS)
#            Ecole Polytechnique Federale de Lausanne
#
#                      All Rights Reserved.
#
# Permission to use, copy, modify and distribute this software and its
# documentation is hereby granted, provided that both the copyright notice
# and this permission notice appear in all copies of the software, derivative
# works or modified versions, and any portions thereof, and that both notices
# appear in supporting documentation.
#
# This code is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
# A PARTICULAR PURPOSE. THE AUTHORS AND ECOLE POLYTECHNIQUE FEDERALE DE LAUSANNE
# DISCLAIM ANY LIABILITY OF ANY KIND FOR ANY DAMAGES WHATSOEVER RESULTING FROM THE
# USE OF THIS SOFTWARE.

federation_nodes=""
federation_hosts=""
for h in $(docker node ls --format '{{ .Hostname }}')
do
	federation_nodes="${federation_nodes} $(docker node inspect --format '{{ .Spec.Labels.name }}' ${h})"
	federation_hosts="${federation_hosts} ${h}"
done

usage() {
	( # This is just in case the user wants to check the settings
		. ./settings.sh
	)

	cat <<EOT
usage: $0 [-h|--help] (all|nodename [nodename ...])
	-h, --help: show this message and exit
	all: Stops the federation on all the nodes currently known
	nodename: one or more nodes on which to stop the stack

You can use environment variables, or add them into settings.local.sh
to change the default values.

To see the full list, please refer to settings.default.sh

Please find below the list of known Federation nodes:
${federation_nodes}

Errors: This script will exit with the following error codes:
 1	No arguments provided
 2	Federation node is incorrect
EOT
}

stop_node() {
	(
		FEDERATION_NODE=$1

		# Export the settings to the docker-compose files
		export FEDERATION_NODE

		# Finally stop the stack
		docker stack rm ${FEDERATION_NODE} &
	)
}

stop_nodes() {
	# Make sure we start from empty lists
	nodes="$*"
	hosts=""
	managers=""
	workers=""

	for n in ${nodes}
	do
		for h in ${federation_hosts}
		do
			label=$(docker node inspect --format '{{ .Spec.Labels.name }}' ${h})
			if [ "x${label}" == "x${n}" ];
			then
				hosts="${hosts} ${h}"
				break 1
			fi
		done
	done

	# Sort the nodes based on their roles
	for h in ${hosts}
	do
		if [ "manager" == "$(docker node inspect --format '{{ .Spec.Role }}' ${h})" ];
		then 
			managers="${managers} ${h}"
		else
			workers="${workers} ${h}"
		fi
	done

	# Stop all the worker nodes
	for h in ${workers}
	do
		label=$(docker node inspect --format '{{ .Spec.Labels.name }}' ${h})
		stop_node ${label}
	done

	# Then stop all the manager nodes
	for h in ${managers}
	do
		label=$(docker node inspect --format '{{ .Spec.Labels.name }}' ${h})
		stop_node ${label}
	done
}

stop_all_nodes() {
	stop_nodes ${federation_nodes}
}

stop_one_node() {
	for h in ${federation_hosts}
	do
		label=$(docker node inspect --format '{{ .Spec.Labels.name }}' ${h})
		if [ "x${label}" == "x${FEDERATION_NODE}" ];
		then
			stop_node ${label}
			break
		fi
	done
}

if [ $# -lt 1 ];
then
	usage
	exit 1
fi

if [ $# -eq 1 ];
then
	case $1 in
		-h|--help)
			usage
			exit 0
		;;
	
		*)
			FEDERATION_NODE="$1"
		;;
	esac

	if [ -z "${FEDERATION_NODE}" ]; then
		echo "Invalid federation node name"
		usage
		exit 3
	fi

	case ${FEDERATION_NODE} in
		all)
			stop_all_nodes
		;;

		*)
			stop_one_node ${FEDERATION_NODE}
		;;
	esac
else
	stop_nodes $*
fi

exit 0