@(python之路)[saltstack源码详解一]

saltstack源码详解

初识源码流程

环境准备：

# 安装

yum install salt-ssh

pip3 install salt-ssh

# 命令

salt-ssh

# 配置

/etc/salt/roster

# 功能

salt-ssh "*" cmd.run "ls"

下面我们查看源码

入口

yum install salt-ssh后，他就会把代码放到python中的lib中。他会触发这个脚本

from salt.scripts import salt_ssh

if __name__ == '__main__':

    salt_ssh()

他会调用这里边的东西并执行。

解析salt-ssh "*" cmd.run "ls" -i 命令

源码流程：

• 启动命令时python脚本
• 读取默认配置文件
  
  obj = salt.clissh.SaltSSH()
  
  obj.run()
• 执行命令
  
  obj = SSH(配置文件)
  
  obj.run()
• 创建进程间供享的队列；[]
• 读取配置文件中的主机，给每个符合条件的主机创建一个进程；
  
  host = next(target_iter)
  
  routine = MultiprocessingProcess(target=self.handle_routine,args=args)
  
  每个进程执行得到结果后，将结果放入队列中
  
  ret = self.handle_routing(……)
  
   触发Single
  
    触发Shell
  
    触发terminal
  
   que.put(ret)

salt日常使用

from salt.client.ssh.shell import Shell

shell = Shell({_ssh_version:(4，9 )})

result = shell.exec_cmd("ls")

print(result)

预留python接口：

from salt.client.ssh.client import SSHClient

client = SSHClient()

ret = client.cmd('*','cmd.run',('ls',),30)

print(ret)

1.grains.items

salt-ssh '*' grains.items

grains一般用于获取服务的静态信息

自定义：

# /srv/salt/_grains/xx.py

预留python接口

from salt.client.ssh.client import SSHClient

client = SSHClient()

ret = client.cmd("*",'grains.items')

print(ret)

2.pillar.items

salt-ssh "*" pillar.items

pillar一般用于获取服务器动态信息

方式一：

# /srv/pillar/top.sls

base:

  "*":          # "*" 表示所有的分组

    - xxx       # 文件名

# /srv/pillar/xxx.sls

x1:

	{% if grains['os_family'] == 'Debian' %}

	  xxx1: 111111111111111

	  {% elif grains['os_family'] == 'RedHat' %}

	  xxx1: 222222222222222

	  {% elif grains['os'] == 'Arch' %}

	  xxx1: 333333333333333

	{% endif %}

2/3: 是否可以用python脚本实现

# /etc/salt/master

ext_pillar:

  - xiaohua:

	     api: 123

  - tom:

 # /var/cache/salt/master/extmods/pillar/tom.py  # [**推荐使用**]

def ext_pillar(minion_id,pillar,*args,**kwargs):

	import time

	return {'alexcccccccccccccctime':str(time.time())}

# /usr/lib/python2.7/site-packages/salt/pillar/wupeiqi.py

def ext_pillar(minion_id,pillar,*args,**kwargs):

	import time

	return {'alexcccccccccccccctime':str(time.time())}

总结pillar源码分析：

salt/script.py

	client = salt.cli.ssh.SaltSSH()

	client.run()

salt/cli/ssh.py

	# self.config是读取的配置文件

	ssh = salt.client.ssh.SSH(self.config)

	ssh.run()

salt/client/ssh/__init__.py

	class SSH(object):

		def run(self, jid=None):

			if self.opts.get('raw_shell', False):

				fun = 'ssh._raw'

				args = argv

			else:

				fun = argv[0] if argv else ''

				args = argv[1:]

			for ret in self.handle_ssh():

				pass 

		def handle_ssh():

			routine = MultiprocessingProcess(target=self.handle_routine,args=args)

			routine.start()

		def handle_routine(self, que, opts, host, target, mine=False):

			single = Single(...)

			single.run()

salt/client/ssh/__init__.py

	class Single(object):

		def run(self, deploy_attempted=False):

			stdout, retcode = self.run_wfunc()

			return stdout, stderr, retcode

		def run_wfunc(self):

			# 实例化Pillar对象

			pillar = salt.pillar.Pillar(

					opts_pkg,

					opts_pkg['grains'],

					opts_pkg['id'],

					opts_pkg.get('environment', 'base')

					)

			pillar_dirs = {}

			# pillar对象.pillar => 去三个地方获取所有pillar

			pillar_data = pillar.compile_pillar(pillar_dirs=pillar_dirs)

			print(pillar_data)

salt/pillar/__init__.py

	class Pillar(object):

		def __init__(...):

			# 去中找自定义pillar

			# /var/cache/salt/master/extmods/pillar/alex.py  # [**推荐使用**]

			# /usr/lib/python2.7/site-packages/salt/pillar/wupeiqi.py

			# ['xiaohua','tom','cmd_json',.....]

			self.ext_pillars = salt.loader.pillars(ext_pillar_opts, self.functions)

	def compile_pillar(self, ext=True, pillar_dirs=None):

		# 1. top.sls

		# 去中找自定义pillar

		# 2. /var/cache/salt/master/extmods/pillar/alex.py  # [**推荐使用**]

		# 3. /usr/lib/python2.7/site-packages/salt/pillar/wupeiqi.py

		pillar1 = self.render_pillar('top.cls')

		pillar2 = self.ext_pillar(pillar1,...)

		return pillar1 + pillar2

	def self.ext_pillar(...):

		return self._external_pillar_data(....)

	def _external_pillar_data(self, pillar, val, pillar_dirs, key):

		ext = self.ext_pillars[key](self.minion_id,pillar,*val)

		return ext