解决saltstack超时问题之借用subprocess gevent multiprocessing

新版的saltstack真是个坑!  前段时间为了做测试,直接从openstack那边搞了几台服务器,接入运维平台开始部署客户端minion的环境,直接yum的saltstack,当时也没有注意,就直接跑lvs的增删改查测试。  当天没出啥问题,今天再测试的时候,感觉前端总是拿不到数据,扯了半天,发现saltstack的超时不好用了,一开始还不知掉问题,后来通过日志,定位到salt超时问题。   


    以前用16和17的时候,虽然也不准,但是最少还是会主动的切断链接,2014版本倒好,只有结束才给你返回。

该回来看看啦,原文地址是: blog.xiaorui.cc

对于2014这个版本,我只能是服气,从一开是就很是蛋疼的~    

     看了下saltstack的源码,在对于zeromq操作的地方,有对时间参数的用法,但是貌似出问题,把源码里面的时间又减小了点,效果还是不稳,找了下saltstack对于任务的多进程分发,但是multiprocessing对于时间处理,貌似没问题。 但还是强改了下,结果。。。。。哥哥我真心有点怕了。。。  既然saltstack 可通过multiprocessing来控制链接的中断停止,那我就改源代码了,我重新写个模块定义超时中断,方法微微有点粗暴 !


下面是我一开始的效果,去minion执行sleep的命令,结果无效果。。。我估计会是版本的问题。


咱们先用一个常用的方式,可以用subprocess的老方法, signal 的方式搞定超时控制.

import subprocess,os, time, signal
#xiaorui.cc
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()


在这里提一下,用saltstack  api的超时,默认控制在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])

父进程被停止,直到子进程被执行完毕。当timeout为None时没有超时,否则有超时。进程可以被join很多次,但不能join自己 !


再给salt他们提提bug,给他们提过好几次类似的问题,还好都解决了。这次看看是啥问题 ~

上面的方法也只是临时的解决,我这里还是希望saltstack快点解决这个bug  ~  另外的是 别总是出新版和大量的新功能更新,还是一步步的来。 步子迈的太大,容易扯到蛋 !



大家觉得文章对你有些作用! 如果想赏钱,可以用微信扫描下面的二维码,感谢!
另外再次标注博客原地址  xiaorui.cc

发表评论

邮箱地址不会被公开。 必填项已用*标注