多线程编程之线程结束

327 阅读2分钟

线程管理包括线程创建、线程启动、线程休眠、等待线程结束和线程停止。线程创建、线程启动、线程休眠前文已提到过,这里不再赘述。

1 等待线程结束

等待线程结束使用的是join()方法,当前线程调用t1线程的join()方法时则阻塞当前进程,等待t1线程结束。如果t1线程结束或等待超时,则当前线程回到活动状态继续执行。join()方法语法如下:

join(timeout=None)

参数timeout是设置超时时间,单位是s。如果没有设置timeout,则可以一直等待

当一个线程依赖于另一个线程的运行结果时,就可以调用另一个线程的join()方法等待它运行完成

example:

import threading
import time
​
shared_value = 0
​
​
​
​
def thread_body():
    global shared_value
    print('ThreadA 开始......')
    for _ in range(5):
        print('ThreadA 执行中......')
        shared_value += 1
        time.sleep(1)
    print('ThreadA 结束......')
​
​
def main():
    print('主线程 开始......')
    t1 = threading.Thread(target=thread_body, name='ThreadA')
    t1.start()
    print('主线程 被阻塞......')
    t1.join()
    print(f'value={shared_value}')
    print('主线程 继续执行......')

result:

主线程 开始......
ThreadA 开始......主线程 被阻塞......
​
ThreadA 执行中......
ThreadA 执行中......
ThreadA 执行中......
ThreadA 执行中......
ThreadA 执行中......
ThreadA 结束......
value=5
主线程 继续执行......

这里设置了一个共享变量shared_value,线程主体中对该变量进行了修改。在主函数里面调用了t1的join()方法,就会导致主线程阻塞,等待线程t1执行结束,主线程才会继续执行。从结果来看,主线程缺失被阻塞了,打印出来的shared_value的值已经由0变成5了

2 线程停止

对于有些复杂业务,需要设置一个线程停止变量来决定线程停止。例如,有一个抓取数据的系统,每隔半小时获取执行一次数据抓取任务。数据抓取任务一般在子线程中进行,休眠一段时间再执行。这个子线程中国会有一个死循环。为了能停止子线程,应设置一个线程停止变量

example:

import threading
import time
​
is_running = True
​
​
def thread_body():
    while is_running:
        print('ThreadA 开始......')
        # 数据抓取任务
        print('ThreadA 执行中......')
        time.sleep(1)
        print('ThreadA 结束......')
    print('整个程序执行完成')
​
​
def main():
    print('主线程 开始......')
    t1 = threading.Thread(target=thread_body)
    t1.start()
    command = input('请输入停止指令:')
    if command == 'exit':
        global is_running
        is_running = False
​
​
if __name__ == '__main__':
    main()

result:

主线程 开始......
ThreadA 开始......
ThreadA 执行中......
请输入停止指令:exit
ThreadA 结束......
整个程序执行完成

这里设置了一个线程停止变量is_running,默认值为True.当用户通过键盘在控制台输入exit时,其值变为False,然后程序终止。需要注意的是,控制台输入exit后,还需要按enter键才可以