MozDef/initscripts/mozdefmq

#!/bin/bash

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
# Copyright (c) 2014 Mozilla Corporation
#
# Contributors:
# Jeff Bryner jbryner@mozilla.com

# uwsgi - This script starts and stops uwsgi apps consuming the mozdef message queues
#
# chkconfig:   - 85 15
# description: uWSGI is a program to run applications adhering to the
#              Web Server Gateway Interface.
# processname: uwsgi
# config:      /etc/sysconfig/uwsgi

# Source function library.
. /etc/rc.d/init.d/functions

# Source the virtual python environment with uwsgi
source  /home/mozdef/envs/mozdef/bin/activate

# all args are in the .ini file, restarts/quit/reload, etc via the fifo instead signalling the PID
uwsgi="/home/mozdef/envs/mozdef/bin/uwsgi"
prog=$(basename "$uwsgi")
UWSGI_CONF_DIR="/home/mozdef/envs/mozdef/mq/"

each_action() {
    action=$1
    configs=$(find "$UWSGI_CONF_DIR" \
                   -maxdepth 1 \
                   -type f \
                   -regextype posix-extended \
                   -iregex '.*\.(ini)$')

    code=0
    if [ -n "$configs" ]; then
        for f in $configs; do
            case "$action" in
                force-reload|restart)
                    stop "$f"
                    start "$f"
                    ;;
                reload)
                    reload "$f"
                    ;;
                chain-reload)
                    chain-reload "$f"
                    ;;
                worker-reload)
                    worker-reload "$f"
                    ;;
                stats)
                    stats "$f"
                    ;;                    
                start)
                    start "$f"
                    ;;
                status)
                    rh_status "$f"
                    ;;
                status_q)
                    rh_status "$f" >/dev/null 2>&1
                    ;;
                stop)
                    stop "$f"
                    ;;
            esac
            retval=$?
        done

        if [ $retval -gt $code ]; then
            code=$retval
        fi
    fi

    return $code
}

args_for() {
    config_file="$1"
    instance=$(instance_for "$config_file")
    pidfile=$(pidfile_for "$config_file")
    args=""
    case "$1" in
        *.ini)        args="$args --ini $f";;
    esac

    echo "$args"
}

instance_for() {
    config_file="$1"
    instance=$(basename "$config_file")
    instance=${instance%.*}
    echo "$instance"
}

pidfile_for() {
    instance=$(instance_for "$1")
    echo "${UWSGI_CONF_DIR}/${instance}.pid"
}

fifo_for(){
    instance=$(instance_for "$1")
    echo "${UWSGI_CONF_DIR}/${instance}.fifo"
}

reload() {
    config_file="$1"
    instance=$(instance_for "$config_file")
    pidfile=$(pidfile_for "$config_file")
    fifofile=$(fifo_for "$config_file")

    echo -n "Reloading uWSGI for ${instance}... "
    
    killproc -p "$pidfile" "$prog" -HUP
    retval=$?
    echo
    return $retval
}

chain-reload(){
    #use the special uwsgi method to 'chain reload' workers
    #allowing them to finish their jobs and restart one by one,only after one is ready
    #trigger by echo c>fifo
    config_file="$1"
    instance=$(instance_for "$config_file")
    pidfile=$(pidfile_for "$config_file")
    fifofile=$(fifo_for "$config_file")

    echo -n "Chain reloading uWSGI for ${instance}... "
    echo c > "$fifofile"
    retval=$?
    echo
    return $retval
}

worker-reload(){
    #use the special uwsgi method to 'gracefully reload' workers
    #allowing them to finish their jobs and restart one by one, ready or not
    #trigger by echo w>fifo
    config_file="$1"
    instance=$(instance_for "$config_file")
    pidfile=$(pidfile_for "$config_file")
    fifofile=$(fifo_for "$config_file")

    echo -n "Worker reloading uWSGI for ${instance}... "
    echo w > "$fifofile"
    retval=$?
    echo
    return $retval
}

stats(){
    #use the special uwsgi method to request stats to the logs
    #trigger by echo s>fifo
    config_file="$1"
    instance=$(instance_for "$config_file")
    pidfile=$(pidfile_for "$config_file")
    fifofile=$(fifo_for "$config_file")

    echo -n "Requesting stats (to logs) from uWSGI for ${instance}... "
    echo s > "$fifofile"
    retval=$?
    echo
    return $retval
}


start() {
    config_file="$1"
    instance=$(instance_for "$config_file")
    pidfile=$(pidfile_for "$config_file")
    args="$(args_for "$config_file")"

    echo -n "Starting uWSGI for ${instance}... "
    daemon --pidfile="$pidfile" $uwsgi $args
    retval=$?
    echo
    return $retval
}

rh_status() {
    config_file="$1"
    status -p "$(pidfile_for "$config_file")" "$prog"
}

stop() {
    config_file="$1"
    instance=$(instance_for "$config_file")
    pidfile=$(pidfile_for "$config_file")

    echo -n "Stopping uWSGI for ${instance}... "
    killproc -p "$pidfile" "$prog"
    retval=$?
    echo
    return $retval
}

case $1 in
    worker-reload|stats|chain-reload|force_reload|reload|restart|start|status|status_q|stop)
        each_action "$1"
        ;;
    *)
        echo "Usage: $0 {worker-reload|chain-reload|reload|stats|restart|start|status|stop}"
        echo "Restart options in order of niceness: chain-reload,worker-reload,reload,restart"
        exit 2
        ;;
esac

exit $?