一日一技：Python多线程的事件监控

摄影：产品经理沙拉

设想这样一个场景：

你创建了10个子线程，每个子线程分别爬一个网站，一开始所有子线程都是阻塞等待。一旦某个事件发生：例如有人在网页上点了一个按钮，或者某人在命令行输入了一个命令，10个爬虫同时开始工作。

肯定有人会想到用Redis来实现这个开关：所有子线程全部监控Redis中名为start_crawl的字符串，如果这个字符串不存在，或者为0，那么就等待1秒钟，再继续检查。如果这个字符串为1，那么就开始运行。

代码片段可以简写为：

import time
import redis
client = redis.Redis()
while client.get('start_crawl') != 1:
    print('继续等待')
    time.sleep(1)

这样做确实可以达到目的，不过每一个子线程都会频繁检查Redis。

实际上，在Python的多线程中，有一个Event模块，天然就是用来实现这个目的的。

Event是一个能在多线程中共用的对象，一开始它包含一个为False的信号标志，一旦在任一一个线程里面把这个标记改为True，那么所有的线程都会看到这个标记变成了True。

我们通过一段代码来说明它的使用方法：

import threading
import time
class spider(threading.Thread):
    def __init__(self, n, event):
        super().__init__()
        self.n = n
        self.event = event
    def run(self):
        print(f'第{self.n}号爬虫已就位！')
        self.event.wait()
        print(f'信号标记变为True！！第{self.n}号爬虫开始运行')
eve = threading.Event()
for num in range(10):
    crawler = spider(num, eve)
    crawler.start()
input('按下回车键，启动所有爬虫！')
eve.set()
time.sleep(10)

运行效果如下图所示：

在这段代码中，线程spider在运行以后，会运行到self.event.wait()这一行，然后10个子线程会全部阻塞在这里。而这里的self.event，就是主线程中eve = threading.Event()生成的对象传入进去的。

在主线程里面，当执行了eve.set()后，所有子线程的阻塞会被同时解除，于是子线程就可以继续运行了。