ansible api2.0 多進程執行不同的playbook

1. 自動化運維工具：ansible
2. 多進程調用ansible api的應用場景：   應用系統檢查
   • 一個應用系統可能具有20—50台服務器的集群，初步的系統層面檢查可以用一個統一的playbook來檢查，比如（df -h這種命令）。但是深入到應用層的話，就有些個性化的應用了，比如有2台http服務器，有20台中間件服務器在加4台數據庫服務器等等，檢查項都是不同的。如果現在還要進行批量的系統檢查，就不能用同一個playbook了。另外，如果順序執行的話，等待時間就會非常長，而且沒有必要，因為中間件的檢查和數據庫的檢查沒有必然的先后順序的要求，可以同步執行。
   • 所以因為上述的場景需求，導致我們需要多進程調用ansible api來動態生成Inventory並執行不同的playbook。
3. 實現步驟
   1. 基於ansible python api 2.0 二次開發，封裝接口
   2. 創建多個ansible 對象，不同的inventory 不同的playbook
   3. 創建多進程，執行程序　
4. 注意事項
   1. ansible api2.0 相較於1.0復雜了許多，也更加靈活。這個需要去閱讀原代碼，需要多花點時間和耐心
   2. 在用多進程之前，我嘗試使用多線程，這塊好像ansible不支持threading ,所以就用multiprocessing代替
5. 代碼

 AnsibleAPI.py

#!/usr/bin/env python
# coding=utf-8
import json
import logging
from ansible.parsing.dataloader import DataLoader
from ansible.vars import VariableManager
from ansible.inventory import Inventory
from ansible.playbook.play import Play
from ansible.executor.task_queue_manager import TaskQueueManager
from ansible.executor.playbook_executor import PlaybookExecutor
from ansible.plugins.callback import CallbackBase
from collections import namedtuple
from ansible import constants as C
import ansible.executor.task_result
import multiprocessing


class ResultsCollector(CallbackBase):

    def v2_runner_on_ok(self, result):
        host = result._host
        logging.warning('===v2_runner_on_ok====host=%s===result=%s' % (host, result._result))


    def v2_runner_on_failed(self, result, ignore_errors=False):
        host = result._host
        logging.warning('===v2_runner_on_failed====host=%s===result=%s' % (host, result._result))

    def v2_runner_on_unreachable(self, result):
        host = result._host
        logging.warning('===v2_runner_on_unreachable====host=%s===result=%s' % (host, result._result))


class AnsibleAPI(object):
    def __init__(self, hostlist, playbooks, *args, **kwargs):
        self.hostlist = hostlist
        self.playbooks = playbooks
        self.passwords = None
        self.callback = None
        Options = namedtuple('Options', ['connection', 'remote_user', 'ask_sudo_pass', 'verbosity', 'ack_pass',
                                         'module_path', 'forks', 'become', 'become_method', 'become_user',
                                         'check', 'listhosts', 'listtasks', 'listtags', 'syntax',
                                         'sudo_user', 'sudo', 'display_skipped_hosts'])

        self.options = Options(connection='smart', remote_user='root', ack_pass=None, sudo_user='root',
                               forks=200, sudo='yes', ask_sudo_pass=False, verbosity=5, module_path=None,
                               become=True, become_method='su', become_user='root', check=None, listhosts=False,
                               listtasks=False, listtags=None, syntax=None, display_skipped_hosts=False)
        self.loader = DataLoader()
        self.variable_manager = VariableManager()
        self.inventory = Inventory(loader=self.loader, variable_manager=self.variable_manager,
                                   host_list=self.hostlist)
        self.variable_manager.set_inventory(self.inventory)
        self.variable_manager.extra_vars = {"ansible_ssh_user": "root"}  # 額外參數，包括playbook參數 key:value

    def runplaybook(self):
        playbook = PlaybookExecutor( 
            playbooks=self.playbooks,
            inventory=self.inventory,
            variable_manager=self.variable_manager,
            loader=self.loader,
            options=self.options,
            passwords=None)
        playbook._tqm._stdout_callback = ResultsCollector()
        playbook.run()

if __name__ == '__main__':
    #創建對象
    anl = AnsibleAPI(['10.26.222.216'], ['/root/ansibleInventory_cl/df.yml'])
    an2 = AnsibleAPI(['10.26.222.210', '10.26.222.213'], ['/root/ansibleInventory_cl/ifconfig.yml'])
    #多進程
    processes = []
    p1 = multiprocessing.Process(name='process_one', target=anl.runplaybook)
    p2 = multiprocessing.Process(name='process_two', target=an2.runplaybook)
    processes.append(p1)
    processes.append(p2)
    for p in processes:
        p.start()

    print('tasks completed')