SaltStack:編寫自己的模塊代碼

Salt的執(zhí)行模塊函數(shù)并沒有什么特別之處,它們實際上就是Python函數(shù),但是當(dāng)它們結(jié)合上Salt的遠程執(zhí)行功能就會變得非常強大。當(dāng)然,我們可以用Python擴展出自己的模塊。

默認(rèn)情況下,我們的定制化模塊存放在/srv/salt/_modules文件夾下,這個文件夾不是默認(rèn)創(chuàng)建的,需要建立對應(yīng)的目錄并添加文件

mkdir -pv /srv/salt/_modules/

創(chuàng)建完之后就可以在該文件下寫第一個我們自己的模塊了

def world():
    """
    This is hello function.
    CLI Example:
        salt '*' hello.world
    """
    return 'Hello world'

模塊添加完成后,需要把模塊推送到所有minion上,命令如下:

[root@localhost _modules]# salt "*" saltutil.sync_modules
minion-one:
    - modules.hello
minion-two:
    - modules.hello

返回結(jié)果是minion接收到的新模塊名,接下來我們就可以測試模塊的執(zhí)行效果了:

[root@localhost _modules]# salt "*" hello.world
minion-two:
    Hello world
minion-one:
    Hello world

ok,這樣我們就成功的完成了第一個模塊的編寫執(zhí)行,雖然這個例子很簡單,但是python語言能提供的強大功能都可以編寫成Salt模塊,并通過遠程執(zhí)行的方式進行調(diào)用。

實戰(zhàn)編寫一個完整模塊

作為系統(tǒng)管理員可能經(jīng)常需要查出當(dāng)前系統(tǒng)最占用內(nèi)存的程序以及當(dāng)前系統(tǒng)中最占用CPU的程序,下面我們就把這個常用的功能寫成一個模塊,讓它使用起來更加便捷。

vi /srv/salt/_modules/prank.py
# -*- coding: utf-8 -*-
'''
The top nth processes which take up CPU and memory space usage are available through this module, additional;
The module can get the system load information.
'''
# Import python libs
import os

# Import salt libs
import salt.utils

def cpu(n):
    '''
    Return the top nth processes which take up the CPU usage for this minion
    CLI Example:
        salt '*' prank.cpu <n>
    '''
    cmd = "ps axu|sort -k3 -nr|head -n%s" % str(n)
    output = __salt__['cmd.run_stdout'](cmd)
    res = []
    for line in output.splitlines():
        res.append(lines)
    return res

def mem(n):
    '''
    Return the top nth processes which take up the memory usage for this minion
    CLI Example:
        salt '*' prank.mem <n>
    '''
    cmd = "ps axu|sort -k4 -nr|head -n%s" % str(n)
    output = __salt__['cmd.run_stdout'](cmd)
    res = []
    for line in output.splitlines():
        res.append(lines)
    return res

def load():
    '''
    Return the load averages for this minion
    CLI Example:
    .. code-block:: bash
        salt '*' prank.loadavg
    '''
    load_avg = os.getloadavg()
    return {'1-min': load_avg[0],
           '5-min': load_avg[1],
           '15-min': load_avg[2]}

這個模塊中cpu(n)mem(n)分別調(diào)用了__salt__['cmd.run_stdout']來執(zhí)行命令,最后將命令返回結(jié)果傳回給master顯示,load()函數(shù)用系統(tǒng)的os模塊收集了load情況。

模塊編寫完成后執(zhí)行模塊同步命令:

[root@localhost _modules]# salt "*" saltutil.sync_modules
minion-one:
    - modules.prank
minion-two:
    - modules.prank

同步完成后就可以使用命令查看最占CPU或者最占內(nèi)存的程序了

salt "*" prank.cpu 4
salt "*" prank.mem 4
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,412評論 6 532
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 98,514評論 3 416
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事?!?“怎么了?”我有些...
    開封第一講書人閱讀 176,373評論 0 374
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經(jīng)常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 62,975評論 1 312
  • 正文 為了忘掉前任,我火速辦了婚禮,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當(dāng)我...
    茶點故事閱讀 71,743評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 55,199評論 1 324
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,262評論 3 441
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 42,414評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 48,951評論 1 336
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 40,780評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,983評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,527評論 5 359
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 44,218評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,649評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,889評論 1 286
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,673評論 3 391
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 47,967評論 2 374

推薦閱讀更多精彩內(nèi)容