1. ホーム
  2. python

[解決済み] スレッドを強制終了させる方法はありますか?

2022-03-20 09:28:26

質問

フラグやセマフォなどを設定・チェックせずに、実行中のスレッドを終了させることは可能でしょうか?

解決方法を教えてください。

Pythonでも、どんな言語でも、スレッドを突然終了させるのは一般的に悪いパターンです。以下のようなケースを考えてみてください。

  • スレッドがクリティカルなリソースを保持しており、適切に終了させる必要がある場合
  • このスレッドは他のスレッドを生成しており、そのスレッドも削除する必要があります。

もし余裕があれば(自分のスレッドを管理している場合)、これを処理する良い方法は、各スレッドが終了する時間かどうかを定期的にチェックするexit_requestフラグを持つことです。

例えば

import threading

class StoppableThread(threading.Thread):
    """Thread class with a stop() method. The thread itself has to check
    regularly for the stopped() condition."""

    def __init__(self,  *args, **kwargs):
        super(StoppableThread, self).__init__(*args, **kwargs)
        self._stop_event = threading.Event()

    def stop(self):
        self._stop_event.set()

    def stopped(self):
        return self._stop_event.is_set()

このコードでは stop() を使い、スレッドが正しく終了するのを待ちます。 join() . スレッドは一定時間ごとに停止フラグをチェックする必要があります。

しかし、本当にスレッドを殺す必要がある場合もあります。例えば、外部ライブラリをラップしていて、そのライブラリが長い間ビジー状態になっており、それを中断させたい場合です。

以下のコードでは、PythonのスレッドでExceptionを発生させることが(いくつかの制限付きで)可能です。

def _async_raise(tid, exctype):
    '''Raises an exception in the threads with id tid'''
    if not inspect.isclass(exctype):
        raise TypeError("Only types can be raised (not instances)")
    res = ctypes.pythonapi.PyThreadState_SetAsyncExc(ctypes.c_long(tid),
                                                     ctypes.py_object(exctype))
    if res == 0:
        raise ValueError("invalid thread id")
    elif res != 1:
        # "if it returns a number greater than one, you're in trouble,
        # and you should call it again with exc=NULL to revert the effect"
        ctypes.pythonapi.PyThreadState_SetAsyncExc(ctypes.c_long(tid), None)
        raise SystemError("PyThreadState_SetAsyncExc failed")

class ThreadWithExc(threading.Thread):
    '''A thread class that supports raising an exception in the thread from
       another thread.
    '''
    def _get_my_tid(self):
        """determines this (self's) thread id

        CAREFUL: this function is executed in the context of the caller
        thread, to get the identity of the thread represented by this
        instance.
        """
        if not self.isAlive():
            raise threading.ThreadError("the thread is not active")

        # do we have it cached?
        if hasattr(self, "_thread_id"):
            return self._thread_id

        # no, look for it in the _active dict
        for tid, tobj in threading._active.items():
            if tobj is self:
                self._thread_id = tid
                return tid

        # TODO: in python 2.6, there's a simpler way to do: self.ident

        raise AssertionError("could not determine the thread's id")

    def raiseExc(self, exctype):
        """Raises the given exception type in the context of this thread.

        If the thread is busy in a system call (time.sleep(),
        socket.accept(), ...), the exception is simply ignored.

        If you are sure that your exception should terminate the thread,
        one way to ensure that it works is:

            t = ThreadWithExc( ... )
            ...
            t.raiseExc( SomeException )
            while t.isAlive():
                time.sleep( 0.1 )
                t.raiseExc( SomeException )

        If the exception is to be caught by the thread, you need a way to
        check that your thread has caught it.

        CAREFUL: this function is executed in the context of the
        caller thread, to raise an exception in the context of the
        thread represented by this instance.
        """
        _async_raise( self._get_my_tid(), exctype )

(ベースは キラブルスレッド Tomer Filiba著。の戻り値についての引用です。 PyThreadState_SetAsyncExc のものと思われる。 古いバージョンのPython .)

ドキュメントにあるように、これは特効薬ではありません。なぜなら、スレッドが Python インタープリタの外でビジー状態になっていると、中断をキャッチできないからです。

このコードの良い使用パターンは、スレッドが特定の例外をキャッチし、クリーンアップを実行することです。そうすれば、タスクを中断しても、適切なクリーンアップを行うことができます。