python多进程编程-多进程编程中的异常处理（二）

原创

玖叁叁

发布于 2023-04-20 13:58:10

94500

代码可运行

文章被收录于专栏：玖叁叁玖叁叁

运行总次数：0

代码可运行

进程池中的异常处理

除了在进程中添加异常处理机制，我们还可以在进程池中添加异常处理机制，以更好地管理进程池中的异常。在 Python 的 multiprocessing 模块中，可以使用 Pool 类提供的 apply_async 方法来提交任务，该方法还支持传递一个回调函数，用于处理任务执行的结果和异常。

import multiprocessing
import time

def worker():
    for i in range(5):
        print(f"Process {multiprocessing.current_process().pid} is running")
        time.sleep(1)
    raise Exception("Process error")

def handle_result(result):
    if isinstance(result, Exception):
        print(f"Pool is raising {result}")

if __name__ == '__main__':
    pool = multiprocessing.Pool(2)
    try:
        pool.apply_async(worker, callback=handle_result)
        pool.apply_async(worker, callback=handle_result)
        pool.close()
        pool.join()
    except Exception as e:
        print(f"Main process is raising {e}")

在上面的程序中，我们定义了一个 handle_result 函数，用于处理任务的结果和异常。在 Pool 类的 apply_async 方法中，我们通过 callback 参数将这个函数传递给了进程池。当任务执行完成时，这个函数会被自动调用，并传递任务的结果或异常对象作为参数。

需要注意的是，在 handle_result 函数中，我们首先判断任务的结果类型是否为异常对象。如果是异常对象，那么就说明任务执行过程中发生了异常，我们需要将这个异常对象打印出来，并及时处理。

原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。

如有侵权，请联系 cloudcommunity@tencent.com 删除。

python

原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。

如有侵权，请联系 cloudcommunity@tencent.com 删除。

python

登录后参与评论

0 条评论

热度