Python 多处理池中键盘中断的优雅处理

使用 Python 多处理池时，处理键盘中断事件并不总是那么简单。在本文中，我们将探讨如何处理此类中断并确保进程正常退出。

提供的代码示例演示了这一挑战。尽管有一个用于 KeyboardInterrupt 的 catch 块，但按下 control-C 时它不会执行。相反，程序会挂起，直到外部终止。

此问题的根源在于与 threading.Condition.wait() 中的条件阻塞相关的 Python 错误。在这种情况下，键盘中断永远不会被发送。结果，它阻止了池内处理中断。

解决此问题的一个方法是在等待结果时指定超时。可以使用带有超时参数的map_async()方法来代替map()。这种方法允许键盘中断被识别和处理：

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

通过设置一个大的超时值，我们本质上是告诉池继续等待结果，直到被用户中断。

需要注意的是，此解决方法有一些限制。如果在所有任务完成之前超时，未完成的结果将被丢弃。因此，选择适当的超时值来平衡响应能力和结果不完整的风险至关重要。