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


先说下需求:

指定salt命令的jid

类似这样的:

wKioL1QIDsTRoHWYAADPA53FEF4308.jpg

开始动手实现。


第一步

修改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。

wKiom1QIFLnBc3fIAAJi78OjehQ794.jpg

当当当。。。

修改成功。

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