使用salt models 自定義

使用salt models 自定義獲取supervisor服務狀態

參考文檔：http://www.supervisord.org/api.html

saltstack提供的模塊沒有提供獲取supervisor的服務狀態信息，使用cmd.run命令返回的結果需要字符串整理，太low，那麼你需要了解下salt的 models使用。

salt models功能可以讓你自己定義salt沒有的功能，這裏編寫python腳本從supervisor XML-RPC API來獲取想要的數據。

官網給出的方式是使用http的方式獲取supervisor的服務狀態信息，supervisor默認不開啓http服務，配置如下

[rpcinterface:supervisor]
supervisor.rpcinterface_factory = supervisor.rpcinterface:make_main_rpcinterface
;[inet_http_server]       
;port=127.0.0.1:9001

xmlrpclib要求傳遞的URL以http或https開頭，解決這個問題的方法是定義一個忽略該URL的自定義傳輸。以下是使用來自主管的傳輸的一些代碼：

import supervisor.xmlrpc
import xmlrpclib
proxy = xmlrpclib.ServerProxy('http://127.0.0.1',
                               transport=supervisor.xmlrpc.SupervisorTransport(
                                    None, None, serverurl='unix://'+socketpath))
proxy.supervisor.getState()

如果這是沒有用的，這裏發現的代碼的更新版本在這裏：

import httplib
import socket
import xmlrpclib

class UnixStreamHTTPConnection(httplib.HTTPConnection, object):
    def __init__(self, *args, **kwargs):
        self.socketpath = kwargs.pop('socketpath')
        super(UnixStreamHTTPConnection, self).__init__(*args, **kwargs)

    def connect(self):
        self.sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
        self.sock.connect(self.socketpath)

class UnixStreamTransport(xmlrpclib.Transport, object):
    def __init__(self, *args, **kwargs):
        self.socketpath = kwargs.pop('socketpath')
        super(UnixStreamTransport, self).__init__(*args, **kwargs)

    def make_connection(self, host):
        return UnixStreamHTTPConnection(host, socketpath=self.socketpath)
def all_process_info():
    server = xmlrpclib.ServerProxy('http://localhost', transport=UnixStreamTransport(socketpath="/var/run/supervisor/supervisor.sock"))
    grains = {"supervisor": server.supervisor.getAllProcessInfo()}
    return grains