进程内存占用监控脚本

最近做系统维护的时候，发现系统里面总有那么几个进程的内存占用量不断的在往上涨，直接导致整个系统崩溃。因为该服务器上同时还运行着其他几个服务进程，为了不让其受到影响，就写了个shell脚本，发现有进程的内存占用率达到上限时，就将其杀死。

该脚本可以既可以根据指定的进程号进行监控，也可以根据正则表达式指定进程名进行监控。

具体代码如下：

#! /bin/bash

PNAME=
PID=
MEM=
SEC=1
USAGE="$(basename $0) -[n 进程名] -[p 进程ID] -[s 监视间隔时间(单位：秒)] -[m 内存占用上限(单位：字节，可以通过k,m,g,K,M,G,%指定单位)]"

while getopts n:p:s:m: OPT
do
        case $OPT in
        n)PNAME=$OPTARG
        ;;
        p)PID=$OPTARG
        ;;
        s)SEC=$OPTARG
        ;;
        m)MEM=$OPTARG
        ;;
        \?) echo $USAGE >&2;exit 1
        ;;
        esac 
done

echo $MEM|grep ^[0-9,.]*[0-9,k,m,g,K,M,G,%]$ >/dev/null

if [ $? -gt 0 ];then
        echo "$MEM词法错误" >&2
        echo $USAGE >&2
        exit 1
fi

MEM=$(echo $MEM|tr "[A-Z]" "[a-z]")

len=${#MEM}
unit=$(echo $MEM|cut -c $len)

case $unit in
	k)
	MEM=$(echo $MEM|cut -c 1-$[len-1])
	MEM=$[MEM*1024]
	;;
	m)
	MEM=$(echo $MEM|cut -c 1-$[len-1])
	MEM=$[MEM*1024*1024]
	;;
	g)
	MEM=$(echo $MEM|cut -c 1-$[len-1])
	MEM=$[MEM*1024*1024*1024]
	;;
	%)
	MEM=$(echo $MEM|cut -c 1-$[len-1])
	;;
	\?)
	;;
esac

while true;do
	date
	
	if [ -n "$PID" ];then
	        ents=$(ps -e -o pid= -o vsz= -o pmem= -o pcpu= -o comm=|grep ^$PID)
	elif [ -n "$PNAME" ];then
	        ents=$(ps -e -o pid= -o vsz= -o pmem= -o pcpu= -o comm=|grep $PNAME$) 
	else
	        echo  "指定进程检索条件">&2
	        echo $USAGE >&2
	        exit 1
	fi
	
	if [ -z "$ents" ];then
	        echo "没有符合检索条件的进程"
	        exit 0
	fi
	
	fields=($ents)
	
	echo "符合检索条件的进程信息："
	echo "【PID】 【VSZ(单位：k)】 【PMEM】 【PCPU】 【COMM】"
	for ((i=0; i<${#fields[@]};i++))
	do
		printf " ${fields[$i]}   "
		if [ $[ i%5 ] -eq 4 ];then
			printf \\n
		fi
	done
	
	#按照内存的占用率进行监控
	if [ $unit == "%" ];then
	        pos=2
	        pmem=${fields[$pos]}
	        while [ -n "$pmem" ];do
	                echo "$pmem $MEM"|awk '($1>=$2){exit 1}'
			if [ $? -eq 1 ];then
	                        pid=${fields[$[pos-2]]}
	                        comm=${fields[$[pos+2]]}
	                        echo "杀死$pid:$comm进程【内存占用率$pmem>=$MEM】"
	                        kill -9 $pid
	                fi
	
	                pos=$[pos+5]
	                pmem=${fields[$pos]}
	        done
	else
	#按照内存的占用量进行监控
	        pos=1
	        vsz=${fields[$pos]}
	        while [ -n "$vsz" ];do
			vsz=$[vsz*1024]
	                if [ $vsz -ge $MEM ];then
	                        pid=${fields[$[pos-1]]}
	                        comm=${fields[$[pos+3]]}
	                        echo "杀死$pid:$comm进程的【内存占用数量$vsz>=$MEM】"
	                        kill -9 $pid
	                fi
	
	                pos=$[pos+5]
	                vsz=${fields[$pos]}
	        done
	fi
	
	echo ""	
	if [ $SEC -gt 0 ];then
		sleep $SEC
	else
		exit 0
	fi
	
done