CyberPanel/CPScripts/watchdog.sh

#!/bin/bash

show_help() {
echo -e "\nrun command: \e[31mnohup bash /etc/cyberpanel/watchdog.sh SERVICE_NAME >/dev/null 2>&1 &\e[39m"
echo -e "\nreplace \e[31mSERVICE_NAME\e[39m to the service name, acceptable word: \e[31mmariadb\e[39m or \e[31mlsws\e[39m"
echo -e "\nWatchdog will check service status every 60 seconds and tries to restart if it is not running and also send an email to designated address"
echo -e "\nto exit watchdog , run command \e[31mbash /etc/cyberpanel/watchdog.sh kill\e[39m"
echo -e "\n\nplease also create \e[31m/etc/cyberpanel/watchdog.flag\e[39m file with following format:"
echo -e "TO=address@email.com"
echo -e "SENDER=sender name"
echo -e "FROM=sender@email.com"
echo -e "You may proceed without flag file , but that will make email sending failed."
}

watchdog_check() {
echo -e "\nChecking LiteSpeed ..."
pid=$(ps aux | grep "watchdog lsws"  | grep -v grep | awk '{print $2}')
	if [[ "$pid" == "" ]] ; then
		echo -e "\nWatchDog for LSWS is gone , restarting..."
		nohup watchdog lsws > /dev/null 2>&1 &
		echo -e "\nWatchDog for LSWS has been started..."
	else
		echo -e "\nWatchDog for LSWS is running...\n"
		echo $(ps aux | grep "watchdog lsws"  | grep -v grep)
	fi
echo -e "\nChecking MariaDB ..."
pid=$(ps aux | grep "watchdog mariadb"  | grep -v grep | awk '{print $2}')
	if [[ "$pid" == "" ]] ; then
		echo -e "\nWatchDog for MariaDB is gone , restarting..."
		nohup watchdog mariadb > /dev/null 2>&1 &
		echo -e "\nWatchDog for MariaDB has been started..."
	else
		echo -e "\nWatchDog for MariaDB is running...\n"
		echo $(ps aux | grep "watchdog mariadb"  | grep -v grep)
	fi
}

if [[ $1 == "mariadb" ]] || [[ $1 == "database" ]] || [[ $1 == "mysql" ]] ; then
	NAME="mariadb"
	echo "Watchdog on MariaDB is started up ..."
elif [[ $1 == "web" ]] || [[ $1 == "lsws" ]] || [[ $1 == "litespeed" ]] || [[ $1 == "openlitespeed" ]] ; then
	NAME="lsws"
	echo "Watchdog on LiteSpeed is started up ..."
elif [[ $1 == "help" ]] || [[ $1 == "-h" ]] || [[ $1 == "--help" ]] || [[ $1 == "" ]] ; then
	show_help
exit
elif [[ $1 == "check" ]] || [[ $1 == "status" ]] ; then
	watchdog_check
	exit
elif [[ $1 == "kill" ]] ; then
pid=$(ps aux | grep "watchdog lsws"  | grep -v grep | awk '{print $2}')
if [[ "$pid" != "" ]] ; then
kill -15 $pid
fi
pid=$(ps aux | grep "watchdog mariadb"  | grep -v grep | awk '{print $2}')
if [[ "$pid" != "" ]] ; then
kill -15 $pid
fi
echo "watchdo has been killed..."
exit
else
show_help

echo -e "\n\n\nunknown service name..."
exit
fi



while [ true = true ]
	do
		systemctl status $NAME 2>&1>/dev/null
			if [[ $? == "0" ]] ; then
				if [[ $NAME == "mariadb" ]] ; then
				pid=$(ps aux | grep "/usr/sbin/mysqld"  | grep -v grep | awk '{print $2}')
				echo "-1000" > /proc/$pid/oom_score_adj
				fi
				echo "$NAME service is running..."
			else
				echo "$NAME is down , try to restart it..."
				if [[ $NAME == "lsws" ]] ; then
					pkill lsphp
				fi
				if [[ $NAME == "mariadb" ]] ; then
				pid=$(ps aux | grep "/usr/sbin/mysqld"  | grep -v grep | awk '{print $2}')
				echo "-1000" > /proc/$pid/oom_score_adj
				fi
				systemctl stop $NAME
				systemctl start $NAME
				if [ -f /etc/cyberpanel/watchdog.flag ] ; then
				flag="/etc/cyberpanel/watchdog.flag"
				LINE3=$(awk 'NR==3' $flag)
				LINE2=$(awk 'NR==2' $flag)
				LINE1=$(awk 'NR==1' $flag)

				FROM=${LINE3#*=}
				SENDER=${LINE2#*=}
				TO=${LINE1#*=}
				sendmail -F $SENDER -f $FROM -i $TO <<MAIL_END
Subject: $NAME is down...
To: $TO
$NAME is down , watchdog attempted to restarting it...

MAIL_END
				fi
			fi
	sleep 60
done