对Ansible二次开发来检查服务器的Java程序占用端口数量
一 应用场景描述
有这个么一个需求：对很多台服务器的Java程序所占用的端口数量与正常的值进行对比确认。
可以使用如下命令得到结果：
netstat -tulnp|grep java|wc -l
但是服务器很多，每台手动去执行这条命令是不现实的。于是想到使用Ansible批量去执行，Ansible使用paramiko去ssh登录服务器执行命令，并使用mutilprocessing实现多进程ssh登录。
二 代码实现
如果直接使用Ansible去执行，不作任何处理的话。执行情况是这样的：
# ansible -i qa_servers.txt all --private-key=/root/.ssh/id_rsa -m shell -a "netstat -tulnp|grep java|wc -l" 172.30.25.71 | success | rc=0 >> 3 172.30.25.179 | success | rc=0 >> 19 172.30.25.180 | success | rc=0 >> 86 172.30.25.181 | success | rc=0 >> 82
ansible就是Ansible命令执行的命令， -i 单独指定hosts文件 --private-key指定ssh私钥路径 -m指定需要调用的模块，这里调用shell模块
-a 指定给模块传递的参数，这里既是要执行的命令
这里出现了几个问题：
a.每台服务器上的正常Java端口数量不同，并且需要和正常的值进行比对和确认，最好可以打印输出结果
b.在命令行直接调用ansible虽然对每台服务器都执行了命令但是输出结果很不规范，需要进一步处理。
Ansible本来就是Python开发的，安装完Ansible后所有的代码都位于/usr/lib/python2.6/site-packages/ansible/目录下，所以可以当作阅读正常python代码来查看Ansible源代码。
再通过查看Ansible官网的代码示例得知Ansible执行命令都是通过
/usr/lib/python2.6/site-packages/ansible/runner/__init__.py 代码中的Runner类的run函数去执行
所以想要对Ansible的输出结果进一步处理，首先要获取Ansible调用命令执行的详细信息。
完整代码如下：
import ansible.runner
from ansible.color import stringc
host_list='qa_servers.txt'
private_key_file='/root/.ssh/id_rsa'
pattern='*'
forks=10
timeout=30
module_name='shell'
# construct the ansible runner and execute on all hosts
results = ansible.runner.Runner(
host_list=host_list,
private_key_file=private_key_file,
pattern=pattern,
forks=forks,
timeout=timeout,
module_name=module_name,
module_args=module_args
).run()
#print results
if results is None:
print "No hosts found"
sys.exit(1)
print stringc("+-------------------+----------------------+-------------+-----------+--------+",color='cyan')
print stringc("| HOST | HOSTNAME | CHICKED NUM | RIGHT NUM | STATUS |",color='cyan')
print stringc("+-------------------+----------------------+-------------+-----------+--------+",color='cyan')
for (hostname, result) in results['contacted'].items():
if not 'failed' in result:
num_default=int(subprocess.Popen(''' awk '/%s/{print $3}' %s|sed -n 's/#//p' ''' %(hostname,host_list),shell=True,stdout=subprocess.PIPE).communicate()[0].split('\n')[0])
num=int(result['stdout'].split('\n')[0])
host_name=result['stdout'].split('\n')[1]
if num==num_default:
status=stringc('PASS',color='green')
else:
status=stringc('WARN',color='red')
print stringc("| %-17s | %-20s | %-11d | %-9d | %-12s ",color='cyan') %(hostname,host_name,num,num_default,status) + stringc("|",color='cyan')
print stringc("+-------------------+----------------------+-------------+-----------+--------+",color='cyan')
# print "%s >>> %s" % (hostname, result['stdout'])
#print "FAILED *******"
#for (hostname, result) in results['contacted'].items():
# if 'failed' in result:
# print "%s >>> %s" % (hostname, result['msg'])
#print "DOWN *********"
#for (hostname, result) in results['dark'].items():
# print "%s >>> %s" % (hostname, result)
qa_servers.txt的内容如下：
172.30.25.71 #testt #3 172.30.25.179 #testttttttt179 #30 172.30.25.180 #testttttttt180 #45 172.30.25.181 #testttttttt181 #55 172.30.25.172 #test #68 172.30.25.173 #test #7
第一列是需要执行命令的主机IP或者主机名，第二列是主机名，第三列就是正常的Java程序占用端口数量。第二列和第三列必须要注释掉。Ansible识别主机时只识别第一列。
执行代码的结果如下：
+-------------------+----------------------+-------------+-----------+--------+ | HOST | HOSTNAME | CHICKED NUM | RIGHT NUM | STATUS | +-------------------+----------------------+-------------+-----------+--------+ | 172.30.25.180 | testtesttesttt | 86 | 45 | WARN | +-------------------+----------------------+-------------+-----------+--------+ | 172.30.25.181 | testttttttt181 | 82 | 55 | WARN | +-------------------+----------------------+-------------+-----------+--------+ | 172.30.25.179 | testttttttt179 | 19 | 30 | WARN | +-------------------+----------------------+-------------+-----------+--------+ | 172.30.25.71 | testt | 3 | 3 | PASS | +-------------------+----------------------+-------------+-----------+--------+
代码中要点：
a.直接使用Ansible的代码进行模块调用。
b.使用Ansible的color.py对输出结果进行颜色处理，PASS是绿色显示，WARN是红色显示。这里的color.py很有启示作用，如果自己以后想要对输出进行颜色处理可以直接参照这里的代码
codeCodes = {
'black': '0;30', 'bright gray': '0;37',
'blue': '0;34', 'white': '1;37',
'green': '0;32', 'bright blue': '1;34',
'cyan': '0;36', 'bright green': '1;32',
'red': '0;31', 'bright cyan': '1;36',
'purple': '0;35', 'bright red': '1;31',
'yellow': '0;33', 'bright purple': '1;35',
'dark gray': '1;30', 'bright yellow': '1;33',
'normal': '0'
}
def stringc(text, color):
"""String in color."""
if ANSIBLE_COLOR:
return "\033["+codeCodes[color]+"m"+text+"\033[0m"
else:
return text
c.使用python的subprocess模块调用shell命令
参考资料：
http://docs.ansible.com/ansible/developing_api.html