如何用 Python 實(shí)現(xiàn)優(yōu)先級(jí)調(diào)度器？

作者：kingname

來源：未聞Code

Python 自帶一個(gè)調(diào)度器模塊sched，它能為你實(shí)現(xiàn)優(yōu)先級(jí)隊(duì)列/延遲隊(duì)列和定時(shí)隊(duì)列。

這個(gè)模塊的使用非常簡(jiǎn)單，首先以延遲隊(duì)列為例：

import sched

def do_work(name):
    print(f'你好：{name}')

sch = sched.scheduler()
sch.enter(5, 1, do_work, argument=('kingname', ))
sch.run()

代碼運(yùn)行以后，會(huì)卡在sch.run()這里，5秒鐘以后執(zhí)行do_work('kingname')，運(yùn)行效果如下圖所示：

其中，sch.enter()的第一個(gè)參數(shù)為延遲的時(shí)間，單位為秒，第二個(gè)參數(shù)為優(yōu)先級(jí)，數(shù)字越小優(yōu)先級(jí)越高。當(dāng)兩個(gè)任務(wù)同時(shí)要執(zhí)行時(shí)，優(yōu)先級(jí)高的先執(zhí)行。但需要注意的是，如果你這樣寫：

import sched

def do_work(name):
    print(f'你好：{name}')

sch = sched.scheduler()
sch.enter(5, 2, do_work, argument=('產(chǎn)品經(jīng)理', ))
sch.enter(5, 1, do_work, argument=('kingname', ))
sch.run()

那么先打印出來的是你好：產(chǎn)品經(jīng)理，如下圖所示：

為什么這里優(yōu)先級(jí)失效了？1的優(yōu)先級(jí)大于2，應(yīng)該先運(yùn)行下面的才對(duì)啊。

這是由于，只有當(dāng)兩個(gè)任務(wù)同時(shí)運(yùn)行的時(shí)候，才會(huì)去檢查優(yōu)先級(jí)。如果兩個(gè)任務(wù)觸發(fā)的時(shí)間一前一后，那么還輪不到比較優(yōu)先級(jí)。

由于延遲隊(duì)列的延遲是相對(duì)于當(dāng)前運(yùn)行這一行代碼的時(shí)間來計(jì)算的，后一行代碼比前一行代碼晚了幾毫秒，所以實(shí)際上產(chǎn)品經(jīng)理這一行會(huì)先到時(shí)間，所以就會(huì)先運(yùn)行。

為了使用絕對(duì)的精確時(shí)間，我們可以使用另外一個(gè)方法：

import sched
import time
import datetime

def do_work(name):
    print(f'你好：{name}')

sch = sched.scheduler(time.time, time.sleep)
start_time = datetime.datetime.now() + datetime.timedelta(seconds=10)
start_time_ts = start_time.timestamp()
sch.enterabs(start_time_ts, 2, do_work, argument=('產(chǎn)品經(jīng)理', ))
sch.enterabs(start_time_ts, 1, do_work, argument=('kingname', ))
sch.run()

運(yùn)行效果如下圖所示：

sch.enterabs()的第一個(gè)參數(shù)是任務(wù)開始時(shí)間的時(shí)間戳，這是一個(gè)絕對(duì)時(shí)間，這個(gè)時(shí)間可以使用datetime模塊來生成，或者其他你熟悉的方式。后面的參數(shù)和sch.enter()完全一樣。

如果你要運(yùn)行的函數(shù)帶有多個(gè)參數(shù)或者默認(rèn)參數(shù)，那么可以使用下面的方式傳入?yún)?shù)：

import sched
import time
import datetime

def do_work(name, place, work='寫代碼'):
    print(f'你好：{name}，你在：{place}{work}')

sch = sched.scheduler(time.time, time.sleep)
start_time = datetime.datetime.now() + datetime.timedelta(seconds=10)
start_time_ts = start_time.timestamp()
sch.enter(5, 2, do_work, argument=('產(chǎn)品經(jīng)理', '杭州'), kwargs={'work': '寫需求文檔'})
sch.enterabs(start_time_ts, 1, do_work, argument=('kingname', '產(chǎn)品經(jīng)理旁邊'), kwargs={'work': '看著她'})
sch.run()

argument參數(shù)對(duì)應(yīng)的元組存放普通參數(shù)，kwargs對(duì)應(yīng)的字典存放帶參數(shù)名的參數(shù)。