Python マルチプロセッシング プールでのキーボード割り込みの適切な処理

Python のマルチプロセッシング プールを使用する場合、KeyboardInterrupt イベントの処理は必ずしも簡単ではありません。この記事では、このような割り込みを処理し、プロセスが正常に終了することを確認する方法を検討します。

提供されたコード サンプルは、この課題を示しています。 KeyboardInterrupt の catch ブロックがあるにもかかわらず、Ctrl+C を押しても実行されません。代わりに、プログラムは外部で終了されるまでハングします。

この問題の根本は、threading.Condition.wait() の条件によるブロックに関連する Python のバグにあります。このコンテキストでは、KeyboardInterrupt は送信されません。その結果、プール内で割り込みが処理されなくなります。

この問題に対する 1 つの解決策は、結果を待つときにタイムアウトを指定することです。タイムアウトパラメータを指定したmap_async()メソッドをmap()の代わりに使用できます。このアプローチにより、KeyboardInterrupt を認識して処理できるようになります。

results = pool.map_async(slowly_square, range(40)).get(9999999)

大きなタイムアウト値を設定することにより、 、基本的に、ユーザーによって中断されるまで結果を待ち続けるようにプールに指示します。

これは重要です。この回避策にはいくつかの制限があることに注意してください。すべてのタスクが完了する前にタイムアウトが経過した場合、不完全な結果は破棄されます。したがって、応答性と不完全な結果のリスクのバランスをとる適切なタイムアウト値を選択することが重要です。