借用subprocess和gevent、multiprocessing解决saltstack超时不准的问题
新版的saltstack真是个坑！前段时间为了做测试，直接从openstack那边搞了几台服务器，接入运维平台开始部署客户端minion的环境，直接yum的saltstack，当时也没有注意，就直接跑lvs的增删改查测试。当天没出啥问题，今天再测试的时候，感觉前端总是拿不到数据，扯了半天，发现saltstack的超时不好用了，一开始还不知掉问题，后来通过日志，定位到salt超时问题。以前用16和17的时候，虽然也不准，但是最少还是会主动的切断链接，2014版本倒好，只有结束才给你返回。
对于2014这个版本，我只能是服气，从一开是就很是蛋疼的～
看了下saltstack的源码，在对于zeromq操作的地方，有对时间参数的用法，但是貌似出问题，把源码里面的时间又减小了点，效果还是不稳，找了下saltstack对于任务的多进程分发，但是multiprocessing对于时间处理，貌似没问题。 但还是强改了下，结果。。。。。哥哥我真心有点怕了。。。既然saltstack 可通过multiprocessing来控制链接的中断停止，那我就改源代码了，我重新写个模块定义超时中断，方法微微有点粗暴 ！
下面是我一开始的效果，去minion执行sleep的命令，结果无效果。。。我估计会是版本的问题。
原文：http://rfyiamcool.blog.51cto.com/1030776/1377429
咱们先用一个常用的方式，可以用subprocess的老方法， signal 的方式搞定超时控制。
import subprocess,os, time, signal
#
def timeout_cmd(cmd, timeout):
timeout=timeout-1
command = cmd.split(" ")
import subprocess, datetime, os, time, signal
start = datetime.datetime.now()
process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
while process.poll() is None:
time.sleep(0.1)
now = datetime.datetime.now()
if (now - start).seconds> timeout:
os.kill(process.pid, signal.SIGKILL)
os.waitpid(-1, os.WNOHANG)
return None
return process.stdout.read()
在这里提一下，用saltstackapi的超时，默认控制在10s的，但我抓到一个微妙的特征，10左右，时间算是精确中断，但是时间一高，你的超时精确就有点不准了。
下面的方法是用gevent的控制超时，如果没有结果返回，会直接切掉连接，并返回一个None 空值。
import gevent
import gevent.monkey
import urllib
import salt.client
gevent.monkey.patch_all()
def test():
#就这么用，7是超时时间，后面的False表示不抛出其他异常了！
with gevent.Timeout(7, False) as timeout:
local = salt.client.LocalClient()
local.cmd('*', 'cmd.run', ['ip a'])
if __name__ == "__main__":
g = gevent.spawn(test)
g.join()
当然从前端过来的任务用multiprocess的join做超时时间也是可以的。
from multiprocessing import Process
import os
#xiaorui.cc
def f(mip):
info('function f')
print 'hello', mip
import salt.client
local = salt.client.LocalClient()
local.cmd('*', 'cmd.run', ['ip a'])
if __name__ == '__main__':
info('main line')
p = Process(target=f, args=('10.58.11.231',))
p.start()
p.join(5)
join()
父进程被停止，直到子进程被执行完毕。当timeout为None时没有超时，否则有超时。进程可以被join很多次，但不能join自己 !
再给salt他们提提bug，给他们提过好几次类似的问题，还好都解决了。这次看看是啥问题 ～
原文：http://rfyiamcool.blog.51cto.com/1030776/1377429
上面的方法也只是临时的解决，我这里还是希望saltstack快点解决这个bug～另外的是 别总是出新版和大量的新功能更新，还是一步步的来。 步子迈的太大，容易扯到蛋 ！
页:
[1]