Python多线程join()的实现细节

2016-04-28 Li Shuai 更多博文 » 博客 » GitHub »

Python 技术

原文链接 https://cyrusin.github.io/2016/04/28/python-join-implementation/
注:以下为加速网络访问所做的原文缓存,经过重新格式化,可能存在格式方面的问题,或偶有遗漏信息,请以原文为准。


写Python多线程都知道当前线程调用a.join()后,会阻塞直到线程a运行结束,看了一下threading模块的源码, 了解了一下实现的原理。

每一个新开启的线程内部都维护着一个Condition类型的条件变量,对线程a进行join(),其实是wait()在线程a 内部的条件变量上,当线程a执行结束时,会通过notify_all()通知所有join()的线程,则阻塞的线程被唤醒,恢复执行。

以下是源码:

self.__block = Condition(Lock()) #线程内部维护的Contition变量
def __stop(self):
    if not hasattr(self, '_Thread__block'):
        return
    self.__block.acquire()
    self.__stopped = True
    self.__block.notify_all() #唤醒所有join()的线程
    self.__block.release()
def join(self, timeout=None): #可以添加timeout
    ...
    self.__block.acquire()
    try:
        if timeout is None:
            while not self.__stopped:
                self.__block.wait() #等待线程运行结束
    else:
        deadline = _time() + timeout
        while not self.__stopped:
            delay = deadline - _time()
            if delay <= 0:
                break
            self.__block.wait(delay)
    finally:
        self.__block.release()

由于是Condition,所以线程可以被join()多次,最终都会由notify_all()唤醒。