如何在不修改原函数的前提下获取multiprocessing调用的函数返回字符串值

核心结论 #

不需要修改原函数即可实现多进程调用时的返回值获取，常用两种落地方式，都不会影响原函数在其他场景的正常使用。

方案1：使用multiprocessing.Pool（最简实现） #

multiprocessing内置的进程池模块原生支持返回值收集，不需要手动维护队列，对原有代码侵入性为0。

代码示例 #

from multiprocessing import Pool

# 你的原有业务函数，完全不需要做任何修改
def your_original_func(param: str) -> str:
    # 这里是原函数的原有逻辑
    return f"处理完成，入参为{param}，这是返回的可变长字符串"

if __name__ == "__main__":
    # 初始化进程池，可根据需求调整进程数量
    with Pool(processes=4) as pool:
        # 单次提交单个任务
        single_result = pool.apply_async(your_original_func, args=("测试入参",)).get()
        print(single_result)

        # 批量提交多个任务
        batch_params = ["入参1", "入参2", "入参3"]
        batch_results = pool.map(your_original_func, batch_params)
        print(batch_results)

方案2：使用Process+包装函数（适合自定义进程场景） #

如果你的场景不适合用进程池，需要直接创建Process实例运行任务，可以通过一个通用包装函数处理队列逻辑，完全不侵入原函数。

代码示例 #

from multiprocessing import Process, Queue

# 通用包装函数，可复用在任意原函数的多进程调用场景
def func_wrapper(queue: Queue, original_func, *args, **kwargs):
    result = original_func(*args, **kwargs)
    queue.put(result)

# 你的原有业务函数，完全不需要做任何修改
def your_original_func(param: str) -> str:
    return f"处理完成，入参为{param}，这是返回的可变长字符串"

if __name__ == "__main__":
    res_queue = Queue()
    # 传入包装函数、队列、原函数及原函数的参数即可
    p = Process(target=func_wrapper, args=(res_queue, your_original_func, "测试入参"))
    p.start()
    p.join()
    # 从队列中取出原函数的返回值
    result = res_queue.get()
    print(result)

注意事项 #

• 两种方案都支持任意长度的字符串返回值，只要返回内容可被pickle序列化即可
• 原函数不需要做任何调整，可正常在非多进程场景调用

内容的提问来源于stack exchange，提问作者swiss_knight