python的_thread模块

2022-10-31 09:32:40 浏览数 (2)

在python中,启用线程有两种方式,一种是利用_thread模块,另一种是用threading模块。一般来说,不建议直接使用_thread模块。但是某些简单的场合也是可以使用的,因为_thread模块的使用方法非常非常的简单。

_thread模块的核心是_thread.start_new_thread方法

_thread.start_new_thread(function, args, [,kwargs])

启动一个新线程并返回其标识符,线程使用args作为执行函数。这个args必须是元组。可选kwargs参数指定关键字参数的字典。当函数返回时,线程将以静默方式退出。当函数以未处理的异常终止时,将打印堆栈跟踪,然后线程退出(其他线程继续运行)

举个例子

代码语言:javascript复制
import time
import datetime
import _thread

date_time_format = '%H:%M:%S'


def get_time_str():
    now = datetime.datetime.now()
    return datetime.datetime.strftime(now, date_time_format)


def thread_function(thread_id):
    print('Thread %dt start at %s' % (thread_id, get_time_str()))
    print('Thread %dt sleeping' % thread_id)
    time.sleep(4)
    if thread_id==3:
        raise Exception('xxx')
    print('Thread %dt finish at %s' % (thread_id, get_time_str()))


def main():
    print('Main thread start at %s' % get_time_str())
    for i in range(5):
        _thread.start_new_thread(thread_function, (i, ))
        time.sleep(1)
    time.sleep(6)
    print('Main thread finish at %s' % get_time_str())


if __name__ == '__main__':
    main()

从代码之中我们可以看到,我在thread_id==3时,抛出一个异常。这时候,第三个线程就自动终止了,并且会在屏幕上打印异常信息。

从执行结果可以看出,_thread模块的start_new_thread方法提供了一种比较简单的多线程机制。单个线程在执行时,别的线程也在“同步”地被执行。虽然从上面的执行结果可以看出执行结果是“顺序”的,但是,也肯出现两行结果相重叠的情况。不信可以试试删掉那个sleep(1)。这是因为线程之间的调度是很难预知的。

那么,我们为什么要在主线程之中加sleep(6)呢?这个目的是为了让主线程不要执行完就立即退出。主线程一旦结束,其他线程无论是否执行完,都会强制退出。

但是,问题来了。

我们在真实情况下,怎么知道线程会在什么时候结束呢?主线程过早或者过晚退出都不是我们所希望的。这时候我们就需要用到线程锁,主线程可以在其他线程执行完之后立即退出。

_thread.allocate_lock方法返回一个Lock对象。Lock对象有三个常见的方法:acquire,release,locked

acquire方法用来获取一个线程锁。release方法用来释放线程锁。locked方法用于获取一个Lock对象是否被锁定。

来举个例子吧,还是参照上面那个例子,做一点点改进。

代码语言:javascript复制
import time
import datetime
import _thread

date_time_format = '%H:%M:%S'


def get_time_str():
    now = datetime.datetime.now()
    return datetime.datetime.strftime(now, date_time_format)


def thread_function(thread_id, lock):
    print('Thread %dt start at %s' % (thread_id, get_time_str()))
    print('Thread %dt sleeping' % thread_id)
    time.sleep(4)
    print('Thread %dt finish at %s' % (thread_id, get_time_str()))
    lock.release()

def main():
    print('Main thread start at %s' % get_time_str())
    locks = []
    for i in range(5):
        lock = _thread.allocate_lock()
        lock.acquire()
        locks.append(lock)
    for i in range(5):
        _thread.start_new_thread(thread_function, (i, locks[i]))
        time.sleep(1)
    for i in range(5):
        while locks[i].locked():
            time.sleep(1)
    print('Main thread finish at %s' % get_time_str())


if __name__ == '__main__':
    main()

从运行的结果可以看出,使用线程锁可以避免主线程过早或过晚退出而产生不可预期的结果。

0 人点赞