检查lvs连接数脚本(nagios插件版)-python

目标：监控lvs的ActiveConn和InActConn连接数
准备工作：由于使用ipvsadm -L -n命令查看lvs连接数一般配置需要root权限，所以需要使用visuo命名在/etc/sudoers中添加下行内容（注意这么使用nagios需要是可登陆用户）：
nagios         ALL=(root)      NOPASSWD: /sbin/ipvsadm -L -n
输入：活跃连接数报警值、非活跃连接数报警值
输出：相关数值大于报警值则进行报警，脚本退出装套为2，将所需报警内容打印在stdout中。
使用方法：
./check_lvs_conn.py  -a 10000 -i 10000
可能输出：
lvs warn!:;ip_prot:111.111.111.116:80 active_conn:16980 inact_conn:12590 
check_result函数稍作修改就可以改成crontab中的监控形式

#! /usr/bin/python

import subprocess
import re
import sys
import optparse

class ip_info():
    def __init__(self,ip_port,active_conn,inact_conn):
        self.ip_port=ip_port
        self.active_conn=active_conn
        self.inact_conn=inact_conn
    def __str__(self):
        return ('ip_prot:%s active_conn:%s inact_conn:%s' % (
        self.ip_port,self.active_conn,self.inact_conn))
    def str(self):
        return self.__str__()

def get_lvs_status():
	#这里需要linux使用visudo命令为脚本执行者执行该命令的权限
	#例如:nagios         ALL=(root)      NOPASSWD: /sbin/ipvsadm -L -n
    command=['/usr/bin/sudo','/sbin/ipvsadm','-L','-n']
    lvs_status_p=subprocess.Popen(command,stdout=subprocess.PIPE)
    return lvs_status_p.stdout

def check_lvs_status(lvs_status,warn_active_conn,warn_inact_conn):
    warn_content=list()
	# 针对ipvsadm的输出去掉不需要处理的行
    require_pattern=re.compile('->')
    exclusive_pattern=re.compile('RemoteAddress')
    separate_pattern=re.compile('\s+')
    for line in lvs_status:
        if require_pattern.search(line) and not exclusive_pattern.search(line):
			# 2字段为ip:port，5字段为活跃连接数，6字段为非活跃连接数
            info_per_line=separate_pattern.split(line)
            active_conn=int(info_per_line[5]) 
            inact_conn=int(info_per_line[6])
            if active_conn >= warn_active_conn or inact_conn >= warn_inact_conn:
                ip_port=info_per_line[2]
                info_per_line=ip_info(ip_port,active_conn,inact_conn)
                warn_content.append(info_per_line)
    return warn_content

def check_result(warn_content):
    warn_message='lvs warn!:'
    if len(warn_content):
        for warn_ip in warn_content:
            warn_message+=(';'+warn_ip.str())
        print(warn_message)
        sys.exit(2)
    else:
        sys.exit(0)

def exec_check(warn_active_conn,warn_inact_conn):
    lvs_status=get_lvs_status()
    warn_content=check_lvs_status(lvs_status,warn_active_conn,warn_inact_conn)
    check_result(warn_content)

def generate_arguments():
	#为兼容旧版使用optparse
    parser=optparse.OptionParser()
    parser.add_option('-a',action='store',type='int',dest='warn_active_conn',help='active conn warn counts')
    parser.add_option('-i',action='store',type='int',dest='warn_inact_conn',help='inactive conn warn counts')
    (options,others)=parser.parse_args()
    return (options.warn_active_conn,options.warn_inact_conn)

if __name__ == '__main__':
    (warn_active_conn,warn_inact_conn)=generate_arguments()
    exec_check(warn_active_conn,warn_inact_conn)

附：
一次sudo /sbin/ipvsadm -L -n命令的可能输出如下：
IP Virtual Server version 1.2.1 (size=4096)
Prot LocalAddress:Port Scheduler Flags
  -> RemoteAddress:Port          Forward     Weight     ActiveConn         InActConn    
TCP  111.111.111.111:80 wlc
  -> 111.111.111.112:80           Route           2              240                     260        
  -> 111.111.111.113:80           Route           2              240                     390        
  -> 111.111.111.114:80           Route           2              240                     280        
  -> 111.111.111.115:80           Route           60            6990                   1520      
  -> 111.111.111.116:80           Route           60            16980                12590
...
...