Shell脚本编写Nagios插件监控程序资源占用

一般情况下，我们只需要监控程序进程在没在就可以了。但是这次遭遇了这样的事，公司开发的程序，程序进程还在，但是死锁了。导致大范围的影响，更要命的是根本不知道问题出在哪里,还是别的测试部同事帮忙发现的，真是丢尽运维的脸了…
为避免下次再遭遇到这样的情况，分析了这次进程死锁的现象，发现死锁会占用100%的cpu，正常情况下只占用10%以内。决定编写nagios插件，用来监控程序占用的资源，包括cpu,内存等。

一、shell脚本需求分析：

   能设置cpu,mem的阈值，资源占用超过阈值就报警。
   要能判断这个进程是否存在，若有一个不存在，则报警。

二、shell脚本执行效果如下：

1、如果输入格式不正确，则输出帮助信息

[root@center230 libexec]# shcomponent_resource.sh
usage parament:
   component_resource.sh [--cpu] [--mem]
 
example:
   component_resource.sh --cpu 50 --mem 50

2、若没超出阈值，输出资源占用情况，退出值为0

[root@center230 libexec]# shcomponent_resource.sh  --cpu 50 --mem 50
vueserver_cpu_use=5.6% vuecache_cpu_use=1.9%vueagent_cpu_use=0.0% vuecenter_cpu_use=0.0% vuedaemon_cpu_use=0.0%;vueserver_mem_use=0.2% vuecache_mem_use=7.4% vueagent_mem_use=0.5% vuecenter_mem_use=0.1%vuedaemon_mem_use=0.0%
[root@center230 libexec]# echo $?
0

3、若超出阈值，输出资源占用情况，退出值为2

[root@center230 libexec]# shcomponent_resource.sh  --cpu 5 --mem 5
vueserver_cpu_use=9.4% vuecache_cpu_use=0.0%vueagent_cpu_use=0.0% vuecenter_cpu_use=0.0% vuedaemon_cpu_use=0.0%;vueserver_mem_use=0.2% vuecache_mem_use=7.4% vueagent_mem_use=0.5%vuecenter_mem_use=0.1% vuedaemon_mem_use=0.0%
[root@center230 libexec]# echo $?
2

4、若进程不存在，输出down掉的进程，以及正常使用中的进程资源情况，退出值为2

[root@yckj scripts]# sh component_resource.sh--cpu 50 --mem 50
current vuedaemon vuecenter vueagent vuecache vueserver is down.
[root@yckj scripts]# echo $?
2

三、shell脚本代码如下：

[root@center230 libexec]# catcomponent_resource.sh
#!/bin/sh
#author:yangrong
#date:2014-05-20
#mail:[e-mail]
 
#pragrom_list=(vuedaemon vuecenter vueagentvuecache vueserver vueconnector myswitch slirpvde)
pragrom_list=(vuedaemon vuecenter vueagentvuecache vueserver)
 
####获取cpu阈值和mem阈值#######
case $1 in
 --cpu)
   cpu_crit=$2
  ;;
 --mem)
   mem_crit=$2
  ;;
esac
 
case $3 in
 --cpu)
   cpu_crit=$4
  ;;
 --mem)
   mem_crit=$4
  ;;
esac
 
 
 
###判断传参数量,如果不为4，则var值为1，var0则正常####
if [[ $1 == $3  ]];then
       var=1  
elif [ $# -ne 4 ] ;then
       var=1
else
       var=0
fi
 
 
###打印错误提示信息
if [ $var -eq 1 ];then
   echo "usage parament:"
   echo "    $0 [--cpu][--mem]"
   echo ""
   echo "example:"
   echo "    $0 --cpu 50 --mem50"
   exit
fi
 
 
###把不存在的进程放一变量中
num=$(( ${#pragrom_list[@]}-1 ))
 
notexist=""
for digit in `seq 0 $num`
do
 a=`ps -ef|grep -v grep |grep ${pragrom_list[$digit]}|wc -l`
  if[ $a -eq 0 ];then
    notexist="$notexist ${pragrom_list[$digit]}"
    unset pragrom_list[$digit]
  fi
done
#echo"pragrom_list=${pragrom_list[@]}"
 
 
 
####对比进程所占资源与阈值大小
cpu_use_all=""
mem_use_all=""
compare_cpu_temp=0
compare_mem_temp=0
for n in ${pragrom_list[@]}
do
  cpu_use=`top -b -n1|grep $n|awk '{print $9}'`
  mem_use=`top -b -n1|grep $n|awk '{print $10}'`
   if[[ $cpu_use == "" ]];then
       cpu_use=0
   fi
   if[[ $mem_use == "" ]];then
       mem_use=0
   fi
 
  compare_cpu=`echo "$cpu_use > $cpu_crit"|bc`
  compare_mem=`echo "$mem_use > $mem_crit"|bc` 
   if[[ $compare_cpu == 1  ]];then
       compare_cpu_temp=1
   fi
   if[[ $compare_mem == 1  ]];then
       compare_mem_temp=1
   fi
 
  cpu_use_all="${n}_cpu_use=${cpu_use}% ${cpu_use_all}"
  mem_use_all="${n}_mem_use=${mem_use}% ${mem_use_all}"
done
 
 
###如果该变量有值，则代表有进程down。则退出值为2
if [[ "$notexist" != ""]];then
 echo -e "current ${notexist} isdown.$cpu_use_all;$mem_use_all"
 exit 2
###如果cpu比较值为1，则代表有进程占用超过阈值，则退出值为2
elif [[ "$compare_cpu_temp" == 1]];then
   echo -e "$cpu_use_all;$mem_use_all"
   exit 2
 
##如果mem比较值为1，则代表为进程mem占用超过阈值，则退出值为2
elif [[ $compare_mem_temp == 1 ]];then
   echo -e "$cpu_use_all;$mem_use_all"
   exit 2
##否则则正常输出，并输出所占cpu与内存比例
else
   echo -e "$cpu_use_all;$mem_use_all"
   exit 0
fi

四、后话：

  随着近日编写shell脚本越来越多，有时难免会回改以前所写脚本，经常要看一段时间才能看懂。
  为方便后续的维护，在脚本当中，每一个函数，每一段功能，都做备注，方便以后自己或他人来进行维护。