Python中使用threading.Event協(xié)調(diào)線程的運行詳解

更新時間：2020年05月02日 10:16:55 作者：tinyid

這篇文章主要介紹了Python中使用threading.Event協(xié)調(diào)線程的運行詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

threading.Event機制類似于一個線程向其它多個線程發(fā)號施令的模式，其它線程都會持有一個threading.Event的對象，這些線程都會等待這個事件的“發(fā)生”，如果此事件一直不發(fā)生，那么這些線程將會阻塞，直至事件的“發(fā)生”。

對此，我們可以考慮一種應用場景（僅僅作為說明），例如，我們有多個線程從Redis隊列中讀取數(shù)據(jù)來處理，這些線程都要嘗試去連接Redis的服務，一般情況下，如果Redis連接不成功，在各個線程的代碼中，都會去嘗試重新連接。

如果我們想要在啟動時確保Redis服務正常，才讓那些工作線程去連接Redis服務器，那么我們就可以采用threading.Event機制來協(xié)調(diào)各個工作線程的連接操作：

主線程中會去嘗試連接Redis服務，如果正常的話，觸發(fā)事件，各工作線程會嘗試連接Redis服務。

為此，我們可以寫下如下的程序：

import threading
import time
import logging
 
logging.basicConfig(level=logging.DEBUG, format='(%(threadName)-10s) %(message)s',)
 
def worker(event):
  logging.debug('Waiting for redis ready...')
  event.wait()
  logging.debug('redis ready, and connect to redis server and do some work [%s]', time.ctime())
  time.sleep(1)
 
readis_ready = threading.Event()
t1 = threading.Thread(target=worker, args=(readis_ready,), name='t1')
t1.start()
 
t2 = threading.Thread(target=worker, args=(readis_ready,), name='t2')
t2.start()
 
logging.debug('first of all, check redis server, make sure it is OK, and then trigger the redis ready event')
time.sleep(3) # simulate the check progress 
readis_ready.set()

運行這個程序：

(t1    ) Waiting for redis ready...
(t2    ) Waiting for redis ready...
(MainThread) first of all, check redis server, make sure it is OK, and then trigger the redis ready event
(t2    ) redis ready, and connect to redis server and do some work [Wed Nov 5 12:45:03 2014]
(t1    ) redis ready, and connect to redis server and do some work [Wed Nov 5 12:45:03 2014]

t1和t2線程開始的時候都阻塞在等待redis服務器啟動的地方，一旦主線程確定了redis服務器已經(jīng)正常啟動，那么會觸發(fā)redis_ready事件，各個工作線程就會去連接redis去做相應的工作。

threading.Event的wait方法還接受一個超時參數(shù)，默認情況下如果事件一直沒有發(fā)生，wait方法會一直阻塞下去，而加入這個超時參數(shù)之后，如果阻塞時間超過這個參數(shù)設定的值之后，wait方法會返回。

對應于上面的應用場景，如果Redis服務器一致沒有啟動，我們希望子線程能夠打印一些日志來不斷地提醒我們當前沒有一個可以連接的Redis服務，我們就可以通過設置這個超時參數(shù)來達成這樣的目的：

import threading
import time
import logging
 
logging.basicConfig(level=logging.DEBUG, format='(%(threadName)-10s) %(message)s',)
 
def worker(event):
  while not event.is_set():
    logging.debug('Waiting for redis ready...')
    event.wait(1)
  logging.debug('redis ready, and connect to redis server and do some work [%s]', time.ctime())
  time.sleep(1)
 
readis_ready = threading.Event()
t1 = threading.Thread(target=worker, args=(readis_ready,), name='t1')
t1.start()
 
t2 = threading.Thread(target=worker, args=(readis_ready,), name='t2')
t2.start()
 
logging.debug('first of all, check redis server, make sure it is OK, and then trigger the redis ready event')
time.sleep(3) # simulate the check progress 
readis_ready.set()

與前面的無限阻塞版本唯一的不同就是，我們在工作線程中加入了一個while循環(huán)，直到redis_ready事件觸發(fā)之后才會結(jié)束循環(huán)，wait方法調(diào)用會在1秒的超時后返回，這樣，我們就可以看到各個工作線程在系統(tǒng)啟動的時候等待redis_ready的同時，會記錄一些狀態(tài)信息。

以下是這個程序的運行結(jié)果：

(t1    ) Waiting for redis ready...
(t2    ) Waiting for redis ready...
(MainThread) first of all, check redis server, make sure it is OK, and then trigger the redis ready event
(t2    ) Waiting for redis ready...
(t1    ) Waiting for redis ready...
(t2    ) Waiting for redis ready...
(t1    ) Waiting for redis ready...
(t2    ) redis ready, and connect to redis server and do some work [Wed Nov 5 13:55:46 2014]
(t1    ) redis ready, and connect to redis server and do some work [Wed Nov 5 13:55:46 2014]

這樣，我們就可以在等待Redis服務啟動的同時，看到工作線程里正在等待的情況。

以上這篇Python中使用threading.Event協(xié)調(diào)線程的運行詳解就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: