如何使用Shell管理一个服务

针对一个服务，比如说C++服务工作在Linux环境环境下，因为它的启动往往伴随着一些环境变量的设置，或者依赖库的设置、或者特殊的启动方式，我们可能需要专门针对它写一个自动化管理的脚本，实现一键操作；

这里针对实际工作的需要与经验总结，特归纳一个管理服务生命周期的脚本，使用shell实现，实现了服务的启动、停止、周期性地监控，包括日志的转存的设置、watchdog脚本的自动生成等；

脚本的维护的地址是：

https://github.com/changshoumeng/projectMgr

#!/bin/bash
##########################################################
#	Teach Wisedom To Machine.
#	Please Call Me Programming devil.
#--------------------------------------------------------
#	ModuleName:server.sh
#	Note:/etc/rc.local
#-------------------------------------------------------
#   Usage:
#   if you project directory like this:
#   --project_root
#   ------lib
#   ------conf
#   ------bin
#   then you put this script into bin/;
#   ./server start	(will set dump_log,process_name,watch_dog)
#   ./server stop   (will try kill pid,then kill -9 pid if stop failed,and stop wath_dog)
#   ./server monit  (will check  the process running,start it if check process stopped)
#########################################################

#you must change it
SRC_PROCESS_NAME="loop_test"
SERVER_LOG_DIR=""

#you cannot change it
LOG_FILE="run.log"
PID_FILE="run.pid"
BASE_DIR=""
RUN_DIR=""
LIB_DIR=""
PROCESS_NAME=$SRC_PROCESS_NAME


function timeStamp(){
	date +'%Y/%m/%d %H:%M:%S'
}

function timeStamp2(){
   date +'%Y%m%d%H%M%S'
}

function logMessage(){
	echo $(timeStamp) $@
	echo $(timeStamp) $@>>$RUN_DIR/$LOG_FILE
}

function setEnv(){
  if [ -z "$BASE_DIR" ] ; then
	PRG="$0"
	while [ -h "$PRG" ] ; do
		ls=`ls -ld "$PRG"`
		link=`expr "$ls" : '.*-> \(.*\)$'`
		if expr "$link" : '/.*' > /dev/null; then
		  PRG="$link"
		else
		  PRG="`dirname "$PRG"`/$link"
		fi
	  done
	  BASE_DIR=`dirname "$PRG"`/..
	  BASE_DIR=`cd "$BASE_DIR" && pwd`
   fi
   RUN_DIR=$BASE_DIR/bin/run
   BIN_DIR=$BASE_DIR/bin
   LIB_DIR=$BASE_DIR/lib
   mkdir -p  $RUN_DIR
}

function setProcessName(){
	IS_CHECK_NAME=$1	
	if [ -z "$IS_CHECK_NAME" ] ;then
		IS_CHECK_NAME=1
	fi
	
	if [ $IS_CHECK_NAME -eq 0 ];then
		PROCESS_NAME=${BASE_DIR##*/}
		return
	fi
	
	if [ -f "$BIN_DIR/$SRC_PROCESS_NAME" ] ;then		
		NEWFILENAME=${BASE_DIR##*/}
		if [ "$NEWFILENAME" != "$SRC_PROCESS_NAME" ] ; then
			[ -f "$BIN_DIR/$NEWFILENAME" ] && mv "$BIN_DIR/$NEWFILENAME"  "${BIN_DIR}/${NEWFILENAME}_$(timeStamp2)"
			mv "$BIN_DIR/$SRC_PROCESS_NAME"  "$BIN_DIR/$NEWFILENAME"
			PROCESS_NAME=$NEWFILENAME
		fi
	else		
		PROCESS_NAME=${BASE_DIR##*/}
	fi
}


function setWatchDog(){
	DOGNAME="${PROCESS_NAME}_dog.sh"
	echo "set watchdog to $DOGNAME"
	if [ -f "$DOGNAME" ] ;then
		return 0
	fi
(
cat <<EOF
#!/bin/bash
#Teach wisedom to my machine
#zhangtao
#watch dog
#watchdog.sh

APPNAME=server.sh

while (true)
  do
		./server.sh monit
		sleep 10
  done
EOF
) > $DOGNAME
	chmod 777  "$DOGNAME"
}

function setServerDumpLog(){
	if [ -z "$SERVER_LOG_DIR" ] ;then
		if [ ! -d "log" ] ;then
			mkdir log
		fi
		return 1
	fi

	MYSERVER_LOG_DIR="${SERVER_LOG_DIR}/${PROCESS_NAME}"
	if [ -d "$MYSERVER_LOG_DIR" ] ;then
		echo "server_log_dir is $MYSERVER_LOG_DIR"
	else
		mkdir -p  "$MYSERVER_LOG_DIR"
		echo "create server_log_dir as: $MYSERVER_LOG_DIR"
	fi

	if [ ! -d "log" ] ;then
		ln -s  "$MYSERVER_LOG_DIR" log
	fi
}




function excuteCmdAndreportLog(){
	`eval $@`
	logMessage $@
}


function running(){
	if [ -f "$RUN_DIR/$PID_FILE" ]; then
			pid=$(cat "$RUN_DIR/$PID_FILE")
			if [ -z "$pid" ] ;then
				return 1
			fi

			pidlist=`pidof "$PROCESS_NAME"`
			if [ -z "$pidlist" ] ;then
				return 1
			fi

			if [[ "$pidlist" =~  "$pid" ]];then
					return 0
			else
					return 1
			fi
	else
			return 1
	fi
}



function start_server(){
	setProcessName
	setServerDumpLog
	if running ;then
		logMessage "$PROCESS_NAME is running ..."
		exit 1
	fi
	setWatchDog
	logMessage "----------------------------------> start_server	$PROCESS_NAME "
	logMessage "nohup $BIN_DIR/$PROCESS_NAME  2>&1 1>&/dev/null  &"
	export LD_LIBRARY_PATH=$LIB_DIR:$LD_LIBRARY_PATH
	echo $LIB_DIR
	chmod a+x $BIN_DIR/$PROCESS_NAME
	nohup $BIN_DIR/$PROCESS_NAME  2>&1 1>&/dev/null  &
	echo $! >$RUN_DIR/$PID_FILE
	excuteCmdAndreportLog "chmod 777 $RUN_DIR/$PID_FILE"
	sleep 2
	if running ;then
		logMessage "$PROCESS_NAME is running ...   "
		exit 1
	fi
	logMessage "$PROCESS_NAME start failed !"
}


function start_server_only(){
	if running ;then
		logMessage "$PROCESS_NAME is running ..."
		exit 1
	fi
	logMessage "----------------------------------> start_server_only	$PROCESS_NAME "
	logMessage "nohup $BIN_DIR/$PROCESS_NAME  2>&1 1>&/dev/null  &"
	export LD_LIBRARY_PATH=$LIB_DIR:$LD_LIBRARY_PATH
	echo $LIB_DIR
	chmod a+x $BIN_DIR/$PROCESS_NAME
	nohup $BIN_DIR/$PROCESS_NAME  2>&1 1>&/dev/null  &
	echo $! >$RUN_DIR/$PID_FILE
	excuteCmdAndreportLog "chmod 777 $RUN_DIR/$PID_FILE"
	sleep 2
	if running ;then
		logMessage "$PROCESS_NAME is running ...   "
		exit 1
	fi
	logMessage "$PROCESS_NAME start failed !"
}


function stop_server(){
	if ! running;then
		logMessage "$PROCESS_NAME was not running"
		exit 1
	fi

	DOGNAME="${PROCESS_NAME}_dog.sh"
	DOGPID=`ps -ef|egrep -i "bash"|egrep -i "$DOGNAME"|egrep -v "egrep"|awk '{print $2}'`
	if [ -n "$DOGPID" ] ;then
		 logMessage "$DOGNAME Work at $DOGPID"
		 excuteCmdAndreportLog "kill -9 $DOGPID"
	fi

	count=0
	pid=$( cat $RUN_DIR/$PID_FILE)
	while running; do
		let count=$count+1
		logMessage "stopping $PROCESS_NAME $count times !!!"
		if [ $count -gt 5 ] ;then
			excuteCmdAndreportLog "kill -9 $pid"
		else
			sleep 1
			excuteCmdAndreportLog "kill  $pid"
		fi
		sleep 2
	done
	logMessage "-----------> stop $PROCESS_NAME successfully <------------"
	excuteCmdAndreportLog "rm $RUN_DIR/$PID_FILE"
}

function monit_server(){
		if running ;then
				exit 1
		fi
		start_server_only $@
}

function status(){
	if running; then
	   logMessage "$PROCESS_NAME is running.";
	   exit 0;
	else
	   logMessage "$PROCESS_NAME was stopped.";
	   exit 1;
	fi
}



function help() {
	echo "------------------------------------------------------------------------------"
	echo "Usage: server.sh {start|status|stop|restart|logback}" >&2
	echo "		 start:				start the $PROCESS_NAME server"
	echo "		 stop:				stop the $PROCESS_NAME server"
	echo "		 monit:				monit the $PROCESS_NAME server"
	echo "		 restart:			restart the $PROCESS_NAME server"
	echo "		 logback:			reload logback config file"
	echo "		 status:			get $PROCESS_NAME current status,running or stopped."
	echo "-----------------------------------------------------------------------------"
}

function getOpts(){
	command=$1
	shift 1
	case $command in
		start)
			start_server $@;
			;;
		stop)
			stop_server $@;
			;;
		logback)
			reload_logback_config $@;
			;;
		status)
			status $@;
			;;
		monit)
			monit_server $@;
			;;
		restart)
			$0 stop $@
			$0 start $@
		  ;;
		help)
			help;
			;;
		*)
			help;
			exit 1;
			;;
	esac
}

function main(){
	setEnv $@
	setProcessName 0
	getOpts $@
}



main $@