salt命令源码分析

/usr/bin/salt

#!/usr/bin/python
‘‘‘
Publish commands to the salt system from the command line on the master.
‘‘‘

from salt.scripts import salt_main

if __name__ == ‘__main__‘:
    salt_main()

调用scripts程序中的salt_main函数

然后再调用/usr/lib/python2.6/site-packages/salt/cli/__init__.py 中的SaltCMD类,然后再调用这个类中的run函数

def salt_main():
    ‘‘‘
    Publish commands to the salt system from the command line on the
    master.
    ‘‘‘
    if ‘‘ in sys.path:
        sys.path.remove(‘‘)
    try:
        client = salt.cli.SaltCMD()
        client.run()
    except KeyboardInterrupt:
        raise SystemExit(‘\nExiting gracefully on Ctrl-c‘)

SaltCMD类调用了/usr/lib/python2.6/site-packages/salt/utils/parsers.py中的SaltCMDOptionParser类

class SaltCMD(parsers.SaltCMDOptionParser):
    ‘‘‘
    The execution of a salt command happens here
    ‘‘‘

    def run(self):
        ‘‘‘
        Execute the salt command line
        ‘‘‘
        self.parse_args()

        if self.config[‘verify_env‘]:
            if not self.config[‘log_file‘].startswith((‘tcp://‘,
                                                       ‘udp://‘,
                                                       ‘file://‘)):
                # Logfile is not using Syslog, verify
                verify_files(
                    [self.config[‘log_file‘]],
                    self.config[‘user‘]
                )

        # Setup file logging!
        self.setup_logfile_logger()

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

        if self.options.batch:
            batch = salt.cli.batch.Batch(self.config)
            # Printing the output is already taken care of in run() itself
            for res in batch.run():
                pass
        else:
            if self.options.timeout <= 0:
                self.options.timeout = local.opts[‘timeout‘]

            kwargs = {
                ‘tgt‘: self.config[‘tgt‘],
                ‘fun‘: self.config[‘fun‘],
                ‘arg‘: self.config[‘arg‘],
                ‘timeout‘: self.options.timeout,
                ‘show_timeout‘: self.options.show_timeout}

            if ‘token‘ in self.config:
                try:
                    with salt.utils.fopen(os.path.join(self.config[‘cachedir‘], ‘.root_key‘), ‘r‘) as fp_:
                        kwargs[‘key‘] = fp_.readline()
                except IOError:
                    kwargs[‘token‘] = self.config[‘token‘]

            if self.selected_target_option:
                kwargs[‘expr_form‘] = self.selected_target_option
            else:
                kwargs[‘expr_form‘] = ‘glob‘

            if getattr(self.options, ‘return‘):
                kwargs[‘ret‘] = getattr(self.options, ‘return‘)

            # If using eauth and a token hasn‘t already been loaded into
            # kwargs, prompt the user to enter auth credentials
            if ‘token‘ not in kwargs and self.options.eauth:
                resolver = salt.auth.Resolver(self.config)
                res = resolver.cli(self.options.eauth)
                if self.options.mktoken and res:
                    tok = resolver.token_cli(
                            self.options.eauth,
                            res
                            )
                    if tok:
                        kwargs[‘token‘] = tok.get(‘token‘, ‘‘)
                if not res:
                    sys.exit(2)
                kwargs.update(res)
                kwargs[‘eauth‘] = self.options.eauth

            if self.config[‘async‘]:
                jid = local.cmd_async(**kwargs)
                print(‘Executed command with job ID: {0}‘.format(jid))
                return
            try:
                # local will be None when there was an error
                if local:
                    if self.options.subset:
                        cmd_func = local.cmd_subset
                        kwargs[‘sub‘] = self.options.subset
                        kwargs[‘cli‘] = True
                    else:
                        cmd_func = local.cmd_cli
                    if self.options.static:
                        if self.options.verbose:
                            kwargs[‘verbose‘] = True
                        full_ret = local.cmd_full_return(**kwargs)
                        ret, out = self._format_ret(full_ret)
                        self._output_ret(ret, out)
                    elif self.config[‘fun‘] == ‘sys.doc‘:
                        ret = {}
                        out = ‘‘
                        for full_ret in local.cmd_cli(**kwargs):
                            ret_, out = self._format_ret(full_ret)
                            ret.update(ret_)
                        self._output_ret(ret, out)
                    else:
                        if self.options.verbose:
                            kwargs[‘verbose‘] = True
                        for full_ret in cmd_func(**kwargs):
                            ret, out = self._format_ret(full_ret)
                            self._output_ret(ret, out)
            except (SaltInvocationError, EauthAuthenticationError) as exc:
                ret = str(exc)
                out = ‘‘
                self._output_ret(ret, out)

    def _output_ret(self, ret, out):
        ‘‘‘
        Print the output from a single return to the terminal
        ‘‘‘
        # Handle special case commands
        if self.config[‘fun‘] == ‘sys.doc‘ and not isinstance(ret, Exception):
            self._print_docs(ret)
        else:
            # Determine the proper output method and run it
            salt.output.display_output(ret, out, self.config)
        if not ret:
            sys.exit(2)

    def _format_ret(self, full_ret):
        ‘‘‘
        Take the full return data and format it to simple output
        ‘‘‘
        ret = {}
        out = ‘‘
        for key, data in full_ret.items():
            ret[key] = data[‘ret‘]
            if ‘out‘ in data:
                out = data[‘out‘]
        return ret, out

    def _print_docs(self, ret):
        ‘‘‘
        Print out the docstrings for all of the functions on the minions
        ‘‘‘
        docs = {}
        if not ret:
            self.exit(2, ‘No minions found to gather docs from\n‘)
        if isinstance(ret, str):
            self.exit(2, ‘{0}\n‘.format(ret))
        for host in ret:
            for fun in ret[host]:
                if fun not in docs:
                    if ret[host][fun]:
                        docs[fun] = ret[host][fun]
        for fun in sorted(docs):
            salt.output.display_output(fun + ‘:‘, ‘text‘, self.config)
            print(docs[fun])
            print(‘‘)

salt --help    查看帮助信息

salt ‘*‘ test.ping

时间: 2024-10-11 11:18:53

salt命令源码分析的相关文章

memcached学习笔记——存储命令源码分析下篇

上一篇回顾:<memcached学习笔记——存储命令源码分析上篇>通过分析memcached的存储命令源码的过程,了解了memcached如何解析文本命令和mencached的内存管理机制. 本文是延续上一篇,继续分析存储命令的源码.接上一篇内存分配成功后,本文主要讲解:1.memcached存储方式:2.add和set命令的区别. memcached存储方式 哈希表(HashTable) 哈希表在实践中使用的非常广泛,例如编译器通常会维护的一个符号表来保存标记,很多高级语言中也显式的支持哈希

memcached学习笔记——存储命令源码分析上

原创文章,转载请标明,谢谢. 上一篇分析过memcached的连接模型,了解memcached是如何高效处理客户端连接,这一篇分析memcached源码中的process_update_command函数,探究memcached客户端的set命令,解读memcached是如何解析客户端文本命令,剖析memcached的内存管理,LRU算法是如何工作等等. 解析客户端文本命令 客户端向memcached server发出set操作,memcached server读取客户端的命令,客户端的连接状态

docker stats 命令源码分析

本文是基于docker 1.10.3版本的源码,对docker stats命令进行源码分析,看看docker stats命令输出的数据是从cgroups fs中怎么怎么计算出来的. docker client相关代码入口可参考:/docker/docker/api/client/stats.go#141 docker daemon相关代码入口可参考:/docker/docker/daemon/daemon.go#1474 源码分析结果 Cpu数据: docker daemon会记录这次读取/sy

redis源码分析之事务Transaction(下)

接着上一篇,这篇文章分析一下redis事务操作中multi,exec,discard三个核心命令. 原文地址:http://www.jianshu.com/p/e22615586595 看本篇文章前需要先对上面文章有所了解: redis源码分析之事务Transaction(上) 一.redis事务核心命令简介 redis事务操作核心命令: //用于开启事务 {"multi",multiCommand,1,"sF",0,NULL,0,0,0,0,0}, //用来执行事

redis源码分析之事务Transaction(上)

这周学习了一下redis事务功能的实现原理,本来是想用一篇文章进行总结的,写完以后发现这块内容比较多,而且多个命令之间又互相依赖,放在一篇文章里一方面篇幅会比较大,另一方面文章组织结构会比较乱,不容易阅读.因此把事务这个模块整理成上下两篇文章进行总结. 原文地址:http://www.jianshu.com/p/acb97d620ad7 这篇文章我们重点分析一下redis事务命令中的两个辅助命令:watch跟unwatch. 一.redis事务辅助命令简介 依然从server.c文件的命令表中找

zeromq源码分析笔记之线程间收发命令(2)

在zeromq源码分析笔记之架构说到了zmq的整体架构,可以看到线程间通信包括两类,一类是用于收发命令,告知对象该调用什么方法去做什么事情,命令的结构由command_t结构体确定:另一类是socket_base_t实例与session的消息通信,消息的结构由msg_t确定.命令的发送与存储是通过mailbox_t实现的,消息的发送和存储是通过pipe_t实现的,这两个结构都会详细说到,今天先说一下线程间的收发命令. zeromq的线程可分为两类,一类是io线程,像reaper_t.io_thr

Linux c 开发 - Memcached源码分析之命令解析(2)

前言 从我们上一章<Linux c 开发 - Memcached源码分析之基于Libevent的网络模型>我们基本了解了Memcached的网络模型.这一章节,我们需要详细解读Memcached的命令解析. 我们回顾上一章发现Memcached会分成主线程和N个工作线程.主线程主要用于监听accpet客户端的Socket连接,而工作线程主要用于接管具体的客户端连接. 主线程和工作线程之间主要通过基于Libevent的pipe的读写事件来监听,当有连接练上来的时候,主线程会将连接交个某一个工作线

Docker源码分析之——Docker Client的启动与命令执行

在上文Docker源码分析之--Docker Daemon的启动 中,介绍了Docker Daemon进程的启动.Docker Daemon可以认为是一个Docker作为Server的运行载体,而真正发送关于docker container操作的请求的载体,在于Docker Client.本文从Docker源码的角度,分析Docker Client启动与执行请求的过程. Docker Client启动的流程与Docker Daemon启动的过程相仿.首先执行reexec.Init():随后解析f

nginx源码分析--nginx外部信号 命令参数

nginx命令行参数 不像许多其他软件系统,Nginx 仅有几个命令行参数,完全通过配置文件来配置 -c </path/to/config> 为 Nginx 指定一个配置文件,来代替缺省的. -t 不运行,而仅仅测试配置文件.nginx 将检查配置文件的语法的正确性,并尝试打开配置文件中所引用到的文件. -v 显示 nginx 的版本. -V 显示 nginx 的版本,编译器版本和配置参数. nginx控制信号 可以使用信号系统来控制主进程.默认,nginx 将其主进程的 pid 写入到 /u