码迷,mamicode.com
首页 > 其他好文 > 详细

指定saltstack的任务id

时间:2014-09-04 19:27:30      阅读:285      评论:0      收藏:0      [点我收藏+]

标签:saltstack   改造源码   jid   

最近在看saltstack的源代码,了解里面的流程,看了下salt命令的执行流程。想想自己能不能改造下源代码,于是自己在那想需求。后来想想能不能在执行salt的时候指定jid啊。


先说下需求:

指定salt命令的jid

类似这样的:

bubuko.com,布布扣

开始动手实现。


第一步

修改salt.utils.parsers.SaltCMDOptionParser类,增加命令行选项。

self.add_option(
            ‘-j‘,‘--jid‘,
            dest=‘jid‘,
            default=‘‘,
            help=(‘Set the job\‘s id,‘
                ‘like 20140903143821048459‘)
        )


第二步

修改salt.utils.parsers.OptionParser类,用于判断jid的合法性。

# 在parse_args方法返回前,调用这个方法
if hasattr(options,‘jid‘):
     self._check_jid(options.jid)
return options, args

# 检查jid的合法性
def _check_jid(self,jid=‘‘):
        try:
            if jid == ‘‘:
                pass
            elif len(jid) != 20:
                raise optparse.OptionValueError(‘jid needs 20 length and is numberic str,like "20140903143821048459"‘)
            else:
                import re
                if not re.match(r‘\d{20}‘,jid):
                    raise optparse.OptionValueError(‘jid needs 20 length and is numberic str,like "20140903143821048459"‘)
        except optparse.OptionValueError,e:
            print(e)
            sys.exit(1)


第三步

修改salt.cli.SaltCMD类

try:
    local = salt.client.LocalClient(self.get_config_file_path(),self.options.jid) #修改
except SaltClientError as exc:
    self.exit(2, ‘{0}\n‘.format(exc))
    return

if self.options.batch:
    batch = salt.cli.batch.Batch(self.config,jid=self.options.jid) # 修改


第四步

修改salt.client.LocalClient类

def __init__(self,
                 c_path=os.path.join(syspaths.CONFIG_DIR, ‘master‘),
                 jid=‘‘,mopts=None): # 增加jid参数
     ......
     self.jid = jid # 增加属性
     self.serial = salt.payload.Serial(self.opts)

def run_job(self,tgt,fun,....):
    # jid = ‘‘
    jid = self.jid


第五步

修改salt.cli.batch.Batch类

def __init__(self, opts, quiet=False,jid=‘‘): # 增加jid
        self.opts = opts
        self.quiet = quiet
        self.local = salt.client.LocalClient(opts[‘conf_file‘],jid) # 增加jid
        self.minions = self.__gather_minions()


所有步骤完成,重启master和minion。

bubuko.com,布布扣

当当当。。。

修改成功。

为什么要加这个功能,因为后台管理系统在执行任务的时候,需要先由web页面产生一个jid。通过这个jid去查询一些信息。

本文出自 “fly天地” 博客,请务必保留此出处http://liuping0906.blog.51cto.com/2516248/1548765

指定saltstack的任务id

标签:saltstack   改造源码   jid   

原文地址:http://liuping0906.blog.51cto.com/2516248/1548765

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!