collectord

#!/bin/sh
#
# collectord: Starts/stops the labstats "collector" daemon
#
# chkconfig: 35 82 16
# description: Starts/stops the labstats "collector" daemon
# processname: collectord
# pidfile: /var/run/labstats/collector.pid

### BEGIN INIT INFO
# Provides: collectord
# Required-Start: $syslog $local_fs $network
# Required-Stop: $syslog $local_fs $network
# Default-Stop: 0 1 2 6
# Default-Start: 3 5
# Short-Description: Starts/stops the labstats "collector" daemon
# Description: Starts a daemonized version of the labstats collector script. 
#              Listens to port 5555 and publishes a JSON file containing the 
#              current machine's statistics (eg. CPU load, used memory, 
#              model etc.) to port 5556.
### END INIT INFO

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

# Storage of python script and its directory.
dir=/afs/umich.edu/user/a/d/adrikim/newLabStats # directory for now
prog="collector"
progdir=$dir$prog.py

# Storage of pidfile and its directory.
pidfile=/var/run/labstats/$prog.pid
pidname="$prog.pid"
PID=""

start() {
    update_pid
    # Checks your privilege.
    if [ $UID -ne 0 ] ; then
	failure
	echo "User has insufficient privilege."
	exit 4
    elif [ ! -z "$PID" ] ; then
	failure
	echo "Instance of collector already seems to be running."
	exit 4
    fi
    
    # Note: no need to try and process other flags
    python $dir/collector.py -d

    # check that it's running by check pid was created
    if [ -e "$pidfile" ] ; then
	success
	echo "Started $prog"
    else
        failure
	echo "$prog failed to start"
    fi
}

stop() {
    # checks sudo/root
    if [ $UID -ne 0 ] ; then
	failure
	echo "Can't stop collector; user has insufficient privilege."
	exit 4
    fi

    # Check that collector is actually running (check its pid)
    update_pid 
    if [ -z $PID ] ; then
	failure
	echo "PID not found. Collector not running?"
	exit 4
    fi
    
    # Stops collector
    kill "$PID"
    success 
    echo "Stopped $prog"
    # Script handles erasure of the pidfile
}

restart() {
    stop
    start
}

reload() {
    restart
}

force_reload() {
    restart
}

rh_status() {
    update_pid 
    if [ -z "$PID" ] ; then # PID is null; collector not running
            echo "collector is not running"
    else
	echo "collector is running with pid" $PID
    fi
}

update_pid() {
    if [ -e "$pidfile" ] ; then
        PID=$(head -n 1 $pidfile)
    else       
        PID=`ps -ef | grep collector | grep python | awk '{print $2}'`
    fi
}

rh_status_q() {
    rh_status >/dev/null 2>&1
}

case "$1" in
    start)
	start
        ;;
    stop)
        stop
        ;;
    restart)
        restart
        ;;
    reload)
        reload
        ;;
    force-reload)
        force_reload
        ;;
    status)
        rh_status
        ;;
    condrestart|try-restart)
        rh_status_q || exit 0
        restart
        ;;
    *)
        echo $"Usage: $0 {start|stop|status|restart|condrestart|try-restart|reload|force-reload}"
        exit 2
esac
exit $?