process_monitor.zip

#!/bin/sh

# http://code.google.com/p/mooon
# 进程监控脚本，当指定进程不存在时，执行重启脚本将它拉起
# 特色：
# 1.本监控脚本可重复执行，它会自动做互斥，保证总是只有一个监控脚本进程存在
# 2.互斥不仅依据监控脚本文件名，而且包含了它的命令行参数，只有整体相同时互斥才生效
# 3.对于被监控的进程，可以只指定进程名，也可以包含命令行参数
# 4.不管是监控脚本还是被监控进程，总是只针对属于当前用户下的进程
 
# 需要指定个数的命令行参数
# 参数1：被监控的进程名
# 参数2：重启被监控进程的脚本
if test $# -ne 2; then
printf "\033[1;33musage: $0 process_cmdline restart_script\033[m\n"
exit 1
fi
 
process_cmdline="$1" # 需要监控的进程名，或完整的命令行，也可以为部分命令行
restart_script="$2"  # 用来重启进程的脚本，要求具有可执行权限
monitor_interval=10  # 定时检测时间间隔，单位为秒
cur_user=`whoami`    # 执行本监控脚本的用户名
 
# 取指定网卡上的IP地址
#eth=1&&netstat -ie|awk -F'[: ]' 'begin{found=0;} { if (match($0,"eth'"$eth"'")) found=1; else if ((1==found) && match($0,"eth")) found=0; if ((1==found) && match($0,"inet addr:") && match($0,"Bcast:")) print $13; }'
 
# 下面这段脚本，用来阻止多个监控脚本进程出现
uid=`id -u $cur_user`
self_name=`basename $0`
self_cmdline="$0 $*"
process_name=$(basename `echo "$process_cmdline"|cut -d" " -f1`)
 
# 以死循环方式，定时检测指定的进程是否存在
while true; do
self_count=`ps -C $self_name h -o euser,args| awk 'BEGIN { num=0; } { if (($1==uid || $1==cur_user) && match($0, self_cmdline)) {++num;}} END { printf("%d",num); }' uid=$uid cur_user=$cur_user self_cmdline="$self_cmdline"`
if test $self_count -gt 2; then
# 如果监控脚本已经运行，则退出不重复运行
printf "\033[0;32;31m$0 is running[$self_count], current user is $cur_user.\033[m\n"
exit
fi
 
# 检查被监控的进程是否存在，如果不存在则重启
process_count=`ps -C $process_name h -o euser,args| awk 'BEGIN { num=0; } { if (($1==uid || $1==cur_user) && match($0, process_cmdline)) {++num;}} END { printf("%d",num); }' uid=$uid cur_user=$cur_user process_cmdline="$process_cmdline"`
if test $process_count -lt 1; then
# 执行重启脚本，要求这个脚本能够将指定的进程拉起来
printf "\033[0;32;34mrestart \"$process_cmdline\"\033[m\n"
sh -c "$restart_script" # 注意一定要以“sh -c”方式执行
fi
 
sleep $monitor_interval
done