control

#!/bin/bash

WORKSPACE=$(
	cd $(dirname $0)/
	pwd
)
cd $WORKSPACE

mkdir -p var

module=swcollector2
app=octopux-$module
conf=cfg.json
dist=$(rpm --showrc | grep dist | grep el | awk '($2=="dist"){print $3 }' | awk -F . '{print $2}')
logfile=/var/log/supervisor/${app}.log
iteration=$(date +%s)
progpath="usr/local/octopux/${app}/"
file_list="control cfg.example.json ${app} perfcounter.json ipt_ser_traffic.sh"

function check_pid() {
	running=$(ps -C $app | grep -v "PID TTY" | wc -l)
	return $running
}

function supervisorconf() {
	cat <<EOF >${app}.conf
[program:${app}]
command=/${progpath}${app}
numprocs=1
directory=/${progpath}
autostart=true
autorestart=true
startsecs=1
startretries=1000000
minfds=64000
exitcodes=0
stopsignal=SIGTERM
stopwaitsecs=300
user=root
redirect_stderr=true
stdout_logfile=/var/log/supervisor/${app}.log
stdout_logfile_maxbytes=10MB
stdout_logfile_backups=10
stdout_capture_maxbytes=10MB
stdout_events_enabled=true
stderr_logfile=/var/log/supervisor/${app}.err.log
stderr_logfile_maxbytes=10MB
stderr_logfile_backups=10
stderr_capture_maxbytes=10MB
stderr_events_enabled=true

EOF


cat << EOF > postinsl.in
supervisorctl reread
supervisorctl update
supervisorctl restart ${app}
EOF

	echo "OK"
	echo ""

}

function run() {
	check_pid
	running=$?
	if [ ${running} -gt 0 ]; then
		echo -n "${app} now is running already, pid="
		ps -C $app | grep -v "PID TTY" | awk '{print $1}'
		stop
		sleep 1
	fi

	if ! [ -f ${conf} ]; then
		echo "Config file $conf doesn't exist, creating one."
		cp cfg.example.json ${conf}
	fi
	ulimit -HSn 65536
	./$app -c ${conf}
}

function start() {
	check_pid
	running=$?
	if [ $running -gt 0 ]; then
		echo -n "${app} now is running already, pid="
		ps -C ${app} | grep -v "PID TTY" | awk '{print $1}'
		return 1
	fi

	if ! [ -f $conf ]; then
		echo "Config file $conf doesn't exist, creating one."
		cp cfg.example.json $conf
	fi
	ulimit -HSn 65536
	nohup ./${app} -c ${conf} >>${logfile} 2>&1 &
	sleep 1
	running=$(ps -C ${app} | grep -v "PID TTY" | wc -l)
	if [ ${running} -gt 0 ]; then
		echo "${app} started..., pid=" $(ps -C ${app} | grep -v "PID TTY" | awk '{print $1}')
	else
		echo "${app} failed to start."
		return 1
	fi
}

function stop() {
	pid=$(ps -C ${app} | grep -v "PID TTY" | awk '{print $1}')
	kill $(pidof ${app})
	echo "${app} (${pid}) stoped..."
}

function restart() {
	stop
	sleep 1
	start
}

function status() {
	check_pid
	running=$?
	if [ $running -gt 0 ]; then
		echo started
	else
		echo stoped
	fi
}

function tailf() {
	tail -f ${logfile}
}

function build() {
	go build -o ${app}
	if [ $? -ne 0 ]; then
		exit $?
	fi

	./$app -v
}

function rpm() {
	build

	if [ "$dist" == "" ]; then
		echo "cant build rpm on this os!"
		exit 1
	fi
	version=$(./${app} -v)
	supervisorconf
	mkdir -p "rpm/${progpath}"
	mkdir -p rpm/etc/supervisor/conf.d/
	#mkdir -p rpm/etc/supervisord.d/

	cp -R ${file_list} rpm/${progpath}
	cp cfg.example.json rpm/${progpath}cfg.json

	cp ${app}.conf rpm/etc/supervisor/conf.d
	#cp ${app}.conf rpm/etc/supervisord.d/${app}.ini

	fpm -s dir -t rpm -n ${app} -m dotwoo_test -v ${version} --iteration ${iteration} --rpm-dist ${dist} --after-install postinsl.in --after-upgrade postinsl.in --after-remove postinsl.in -C rpm --config-files ${progpath}cfg.json --config-files ${progpath}perfcounter.json -f --url http://www.baishancloud.com/ --provides dotwoo@baishancloud.com --vendor dotwoo@baishancloud.com

	rm -fr rpm
}

function help() {
	echo "$0 build|start|stop|restart|status|tail|run|rpm"
}

if [ "$1" == "" ]; then
	help
elif [ "$1" == "stop" ]; then
	stop
elif [ "$1" == "start" ]; then
	start
elif [ "$1" == "restart" ]; then
	restart
elif [ "$1" == "status" ]; then
	status
elif [ "$1" == "tail" ]; then
	tailf
elif [ "$1" == "run" ]; then
	run
elif [ "$1" == "build" ]; then
	build
elif [ "$1" == "rpm" ]; then
	rpm
else
	help
fi