Python中使用threading.Event協調線程的運行詳解

threading.Event機制類似於一個線程向其它多個線程發號施令的模式，其它線程都會持有一個threading.Event的對象，這些線程都會等待這個事件的“發生”，如果此事件一直不發生，那麼這些線程將會阻塞，直至事件的“發生”。
對此，我們可以考慮一種應用場景（僅僅作為說明），例如，我們有多個線程從Redis隊列中讀取數據來處理，這些線程都要嘗試去連接Redis的服務，一般情況下，如果Redis連接不成功，在各個線程的代碼中，都會去嘗試重新連接。
如果我們想要在啟動時確保Redis服務正常，才讓那些工作線程去連接Redis服務器，那麼我們就可以採用threading.Event機制來協調各個工作線程的連接操作：
主線程中會去嘗試連接Redis服務，如果正常的話，觸發事件，各工作線程會嘗試連接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服務器已經正常啟動，那麼會觸發redis_ready事件，各個工作線程就會去連接redis去做相應的工作。
threading.Event的wait方法還接受一個超時參數，默認情況下如果事件一直沒有發生，wait方法會一直阻塞下去，而加入這個超時參數之後，如果阻塞時間超過這個參數設定的值之後，wait方法會返回。
對應於上面的應用場景，如果Redis服務器一致沒有啟動，我們希望子線程能夠打印一些日誌來不斷地提醒我們當前沒有一個可以連接的Redis服務，我們就可以通過設置這個超時參數來達成這樣的目的：
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循環，直到redis_ready事件觸發之後才會結束循環，wait方法調用會在1秒的超時後返回，這樣，我們就可以看到各個工作線程在系統啟動的時候等待redis_ready的同時，會記錄一些狀態信息。
以下是這個程序的運行結果：
(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服務啟動的同時，看到工作線程里正在等待的情況。