~ritho/rweb

2afb48da02386a57f49bc630e4e78df6e70c37a8 — Ritho 2 months ago aadda76
Simplify and improve the start/stop script.

Signed-off-by: Ritho <palvarez@ritho.net>
1 files changed, 26 insertions(+), 120 deletions(-)

M scripts/web
M scripts/web => scripts/web +26 -120
@@ 6,163 6,69 @@
# Required-Stop:     $local_fs $remote_fs $network $syslog $named
# Default-Start:     2 3 4 5
# Default-Stop:      0 1 6
# Short-Description: starts the web
# Description:       starts web using start-stop-daemon
# Short-Description: starts the ritho's web
### END INIT INFO

set -e

PATH=/usr/local/sbin:/usr/local/bin:/sbin:/bin:/usr/sbin:/usr/bin
DAEMON=/usr/bin/web
NAME=web
DESC=web

STOP_SCHEDULE="${STOP_SCHEDULE:-QUIT/5/TERM/5/KILL/5}"

test -x $DAEMON || exit 0
test -x /usr/bin/web || exit 0

. /lib/init/vars.sh
. /lib/lsb/init-functions

# Try to extract nginx pidfile
PID=$(cat /etc/web/web.yaml | grep -Ev '^\s*#' | awk 'BEGIN { RS="[;{}]" } { if ($1 == "pid") print $2 }' | head -n1)
if [ -z "$PID" ]; then
	PID=/var/run/web.pid
fi

start_web() {
	# Start the daemon/service
	#
	# Returns:
	#   0 if daemon has been started
	#   1 if daemon was already running
	#   2 if daemon could not be started
	start-stop-daemon --start --quiet --pidfile $PID --exec $DAEMON --test > /dev/null \
		|| return 1
	start-stop-daemon --start --quiet --pidfile $PID --exec $DAEMON -- \
		$DAEMON_OPTS 2>/dev/null \
		|| return 2
}

stop_web() {
	# Stops the daemon/service
	#
	# Return
	#   0 if daemon has been stopped
	#   1 if daemon was already stopped
	#   2 if daemon could not be stopped
	#   other if a failure occurred
	start-stop-daemon --stop --quiet --retry=$STOP_SCHEDULE --pidfile $PID --name $NAME
	RETVAL="$?"
	sleep 1
	return "$RETVAL"
}

reload_web() {
	# Function that sends a SIGHUP to the daemon/service
	start-stop-daemon --stop --signal HUP --quiet --pidfile $PID --name $NAME
	start-stop-daemon --stop --signal HUP --quiet --pidfile /run/web.pid --name $NAME
	return 0
}

rotate_logs() {
	# Rotate log files
	start-stop-daemon --stop --signal USR1 --quiet --pidfile $PID --name $NAME
	start-stop-daemon --stop --signal USR1 --quiet --pidfile /run/web.pid --name $NAME
	return 0
}

upgrade_web() {
	# Online upgrade nginx executable
	# http://nginx.org/en/docs/control.html
	#
	# Return
	#   0 if web has been successfully upgraded
	#   1 if web is not running
	#   2 if the pid files were not created on time
	#   3 if the old master could not be killed
	if start-stop-daemon --stop --signal USR2 --quiet --pidfile $PID --name $NAME; then
		# Wait for both old and new master to write their pid file
		while [ ! -s "${PID}.oldbin" ] || [ ! -s "${PID}" ]; do
			cnt=`expr $cnt + 1`
			if [ $cnt -gt 10 ]; then
				return 2
			fi
			sleep 1
		done
		# Everything is ready, gracefully stop the old master
		if start-stop-daemon --stop --signal QUIT --quiet --pidfile "${PID}.oldbin" --name $NAME; then
			return 0
		else
			return 3
		fi
	else
		return 1
	fi
}

case "$1" in
	start)
		log_daemon_msg "Starting $DESC" "$NAME"
		start_web
		case "$?" in
			0|1) log_end_msg 0 ;;
			2)   log_end_msg 1 ;;
		esac
		if start-stop-daemon --start --quiet --oknodo --background --make-pidfile --chuid 0:0 --pidfile /run/web.pid --exec /usr/bin/web; then
			log_end_msg 0 || true
		else
			log_end_msg 1 || true
		fi
		;;
	stop)
		log_daemon_msg "Stopping $DESC" "$NAME"
		stop_web
		case "$?" in
			0|1) log_end_msg 0 ;;
			2)   log_end_msg 1 ;;
		esac
		if start-stop-daemon --stop --quiet --oknodo --pidfile /run/web.pid --exec /usr/bin/web; then
			log_end_msg 0 || true
		else
			log_end_msg 1 || true
		fi
		;;
	restart)
		log_daemon_msg "Restarting $DESC" "$NAME"

		# Check configuration before stopping nginx
		if ! test_config; then
			log_end_msg 1 # Configuration error
			exit $?
		if start-stop-daemon --stop --quiet --oknodo --pidfile /run/web.pid --exec /usr/bin/web; then
			if start-stop-daemon --start --quiet --oknodo --background --make-pidfile --chuid 0:0 --pidfile /run/web.pid --exec /usr/bin/web; then
				log_end_msg 0 || true
			else
				log_end_msg 1 || true
			fi
		else
			log_end_msg 1 || true
		fi

		stop_web
		case "$?" in
			0|1)
				start_web
				case "$?" in
					0) log_end_msg 0 ;;
					1) log_end_msg 1 ;; # Old process is still running
					*) log_end_msg 1 ;; # Failed to start
				esac
				;;
			*)
				# Failed to stop
				log_end_msg 1
				;;
		esac
		;;
	reload|force-reload)
		log_daemon_msg "Reloading $DESC configuration" "$NAME"

		# Check configuration before stopping nginx
		#
		# This is not entirely correct since the on-disk nginx binary
		# may differ from the in-memory one, but that's not common.
		# We prefer to check the configuration and return an error
		# to the administrator.
		if ! test_config; then
			log_end_msg 1 # Configuration error
			exit $?
		fi

		reload_web
		log_end_msg $?
		;;
	status)
		status_of_proc -p $PID "$DAEMON" "$NAME" && exit 0 || exit $?
		;;
	upgrade)
		log_daemon_msg "Upgrading binary" "$NAME"
		upgrade_web
		log_end_msg $?
		status_of_proc -p /run/web.pid /usr/bin/web "$NAME" && exit 0 || exit $?
		;;
	rotate)
		log_daemon_msg "Re-opening $DESC log files" "$NAME"


@@ 170,7 76,7 @@ case "$1" in
		log_end_msg $?
		;;
	*)
		echo "Usage: $NAME {start|stop|restart|reload|force-reload|status|rotate|upgrade}" >&2
		echo "Usage: $NAME {start|stop|restart|reload|force-reload|status|rotate}" >&2
		exit 3
		;;
esac