python中斷time.sleep一種更優(yōu)雅的方式:event.wait

更新時(shí)間：2023年11月10日 09:00:10 作者：Rnan-prince

這篇文章主要介紹了python中斷time.sleep一種更優(yōu)雅的方式:event.wait,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

場(chǎng)景描述

實(shí)現(xiàn)一個(gè)功能，啟動(dòng)一個(gè)線(xiàn)程，只要線(xiàn)程不中斷，就一直while true，間斷10s累加某個(gè)數(shù)；

當(dāng)線(xiàn)程線(xiàn)程信號(hào)時(shí)，while停止，線(xiàn)程退出。

1、方法一 time.sleep

import time
from threading import Thread
 
 
class AddTask(Thread):
 
    def __init__(self):
        super().__init__()
        self.sum_num = 0
        self._button = True
 
    def run(self):
        print('Begin to add ...')
        while self._button:
            print('Sleep start...')
            time.sleep(10)
            print('Sleep end...')
            self.sum_num += 10
        print(f"Finish add, sum:{self.sum_num}...")
 
    def stop(self):
        print('Stop task...')
        self._button = False
 
 
def do_main_task():
    print('Do main task...')
    time.sleep(15)
    print('Finish main task...')
    return True
 
 
if __name__ == '__main__':
    add_task = AddTask()
    add_task.start()
    main_task_end = do_main_task()
    if main_task_end:
        add_task.stop()
        time.sleep(1)
        print(f"sum: {add_task.sum_num}")

執(zhí)行結(jié)果：

Begin to add ...Do main task...
Sleep start...
Sleep end...
Sleep start...
Finish main task...
Stop task...
sum: 10
Sleep end...
Finish add，sum: 20...

解釋一下這段代碼的意思。在主線(xiàn)程里面，我調(diào)用do_main_task()觸發(fā)了一個(gè)任務(wù)。這個(gè)任務(wù)執(zhí)行會(huì)久一些（這里設(shè)定15s）。但是這個(gè)任務(wù)完成以后，會(huì)有個(gè)返回值，告訴我完成了。另外創(chuàng)建一個(gè) add_task 子線(xiàn)程，每10秒累加10。

但某些情況下，我不需要等待了，例如用戶(hù)主動(dòng)取消了任務(wù)。這個(gè)時(shí)候，我就想提前結(jié)束這個(gè) add_task 子線(xiàn)程。

通過(guò)執(zhí)行結(jié)果可以看出，當(dāng)執(zhí)行一次+10,等待10s后，又過(guò)了5秒，主線(xiàn)程do_main_task結(jié)束了，這時(shí)add_task線(xiàn)程還在累加的sleep(10)中沒(méi)有退出。主線(xiàn)程執(zhí)行結(jié)果已經(jīng)是sum=10，再過(guò)5秒后add_task線(xiàn)程才結(jié)束。

但是，線(xiàn)程是不能從外面主動(dòng)殺死的，只能讓它自己退出。

2、方法二 event.wait

應(yīng)用threading模塊里面的Event

用法和sleep差不多：

import threading
 
event = threading.Event()
event.wait(5)

上述例子可以這樣實(shí)現(xiàn)：

import time
from threading import Thread
from threading import Event
 
 
class AddTask(Thread):
 
    def __init__(self):
        super().__init__()
        self.sum_num = 0
        self.event = Event()
 
    def run(self):
        print('Begin to add ...')
        while not self.event.is_set():
            print('Sleep start...')
            self.event.wait(10)
            print('Sleep end...')
            self.sum_num += 10
        print(f"Finish add, sum:{self.sum_num}...")
 
    def stop(self):
        print('Stop task...')
        self.event.set()
 
 
def do_main_task():
    print('Do main task...')
    time.sleep(15)
    print('Finish main task...')
    return True
 
 
if __name__ == '__main__':
    add_task = AddTask()
    add_task.start()
    main_task_end = do_main_task()
    if main_task_end:
        add_task.stop()
        time.sleep(1)
        print(f"sum: {add_task.sum_num}")

執(zhí)行結(jié)果：

Do main task...
Begin to add ...
Sleep start...
Sleep end...
Sleep start...
Finish main task...
Stop task...
Sleep end...
Finish add, sum:20...

sum: 20

當(dāng)執(zhí)行event.set()后，子線(xiàn)程里面self.event.is_set()就會(huì)返回 False，于是這個(gè)循環(huán)就不會(huì)繼續(xù)執(zhí)行了。

即使self.event.wait(10)剛剛開(kāi)始阻塞，只要我在主線(xiàn)程中執(zhí)行了event.set()，子線(xiàn)程里面的阻塞立刻就會(huì)結(jié)束。

于是子線(xiàn)程立刻就會(huì)結(jié)束。不需要再白白等待10秒。

并且，event.wait()這個(gè)函數(shù)在底層是使用 C 語(yǔ)言實(shí)現(xiàn)的，不受 GIL 鎖的干擾。