python中停止线程的方法代码举例

原创

ithorizon 9个月前 (07-01) 阅读数 166 #Python

1 threading.Event()方法

一种常见的方法是使用标志位来通知线程应该停止。线程可以定期检查这个标志位，如果它被设置为停止，那么线程就终止其执行。下面是一个明了的例子：

import threading  
import time  
  
class MyThread(threading.Thread):  
    def __init__(self):  
        super(MyThread, self).__init__()  
        self.stop_event = threading.Event()  
  
    def run(self):  
    	# 清空设置
        self.stop_event.clear()
        while not self.stop_event.is_set():  
            print("Thread is running...")  
            time.sleep(1)  
  
    def stop(self):  
        self.stop_event.set()  
  
# 创建线程  
thread = MyThread()  
thread.start()  
  
# 在某个时间点，停止线程  
time.sleep(5)  
thread.stop()

需要注意的是，这只是一种优雅的停止线程的方法，它依靠于线程在run方法中定期检查stop_event。如果线程没有这样的检查，或者它正在执行一个无法被中断的阻塞操作（例如IO操作），那么这种方法大概无法立即停止线程。

2 子线程抛出异常，立刻停止

在Python中，使用ctypes和PyThreadState_SetAsyncExc函数来在子线程中异步抛出一个异常是一种相对底层的做法，它可以直接在子线程的上下文中触发一个异常。然而，这种做法需要谨慎使用，出于它大概会致使线程状态不稳定或未定义的行为，特别是如果线程没有正确地处理异常。

import threading
import time
import ctypes
import inspect

def do_some_task():
    while True:
        time.sleep(1)
        print("子线程！1")
        time.sleep(1)
        print("子线程！2")
        time.sleep(1)
        print("子线程！3")
        time.sleep(1)
        print("子线程！4")
        time.sleep(1)
        print("子线程！5")

def async_raise(thread_id, exctype):
    """
    通过C语言的库抛出异常
    :param thread_id:
    :param exctype:
    :return:
    """
    # 在子线程内部抛出一个异常终止线程
    thread_id = ctypes.c_long(thread_id)
    if not inspect.isclass(exctype):
        exctype = type(exctype)
    res = ctypes.pythonapi.PyThreadState_SetAsyncExc(thread_id, ctypes.py_object(exctype))
    if res == 0:
        raise ValueError("线程id违法")
    elif res != 1:
        ctypes.pythonapi.PyThreadState_SetAsyncExc(thread_id, None)
        raise SystemError("异常抛出挫败")

def stop_thread_now(thread):
    # 终止线程
    async_raise(thread.ident, SystemExit)

if __name__ == "__main__":
    # 可以在子线程任何时候随时终止子线程
    sub_thread = threading.Thread(target=do_some_task,name="sub_thread")
    sub_thread.start()
    print(sub_thread.is_alive())
    time.sleep(7)
    stop_thread_now(sub_thread)
    time.sleep(1)
    print(sub_thread.is_alive())