python kill 线程

python kill 线程

import ctypes
import threading


class Thread(threading.Thread):
    def __init__(self, *params, **known):
        super(Thread, self).__init__(*params, **known)
        parent_thread = threading.current_thread()
        self.is_killed = False
        self.child_threads = []
        if hasattr(parent_thread, 'child_threads'):
            parent_thread.child_threads.append(self)

    def _raise_exc(self, exc_obj):
        if not self.is_alive():
            return

        res = ctypes.pythonapi.PyThreadState_SetAsyncExc(
            ctypes.c_long(self.ident), ctypes.py_object(exc_obj))
        if res == 0:
            raise RuntimeError("Not existent thread id.")
        elif res > 1:
            ctypes.pythonapi.PyThreadState_SetAsyncExc(self.ident, None)
            raise SystemError("PyThreadState_SetAsyncExc failed.")

    def kill(self):
        if hasattr(self, 'child_threads'):
            for child_thread in self.child_threads:
                if child_thread.is_alive():
                    child_thread.kill()
        self._raise_exc(SystemExit)
        self.is_killed = True

th = Thread(target=func)
th.start()
th.kill()

使用该方法就可以启动杀线程的操作，但是！！！这里其实并没有真正杀掉。

如果上面的func里面包含有类似sleep的等待函数，并且在执行kill的时候恰好就在sleep，那么程序是不会立即被杀掉，而是会等待sleep结束才会杀掉，如果要等待程序完全stop，那么需要在kill结束后跟一个join

def func():
    time.sleep(11)

th = Thread(target=func)
th.start()
time.sleep(1)
th.kill()
t1 = time.time()
th.join()
t2 = time.time()
print(t2 - t1)

这里先创建一个线程，启动，等待1s让线程中的函数进入sleep状态，然后启动杀线程函数，由于线程中的程序在sleep，所以并不会直接结束，而是等待sleep结束，join会消耗10s钟的时间