如何从 Python 线程中获取返回值？

python

2024-03-31 01:54:54

如何从 Python 线程中获取返回值

引言

线程在应用程序并发编程中发挥着至关重要的作用。它们允许同时执行多个任务，提高应用程序的性能和响应能力。然而，默认情况下，无法直接从线程中获取返回值。了解如何从线程中检索结果对于有效管理线程并从多线程环境中充分受益至关重要。

获取线程返回值的方法

直接方法

最直接的方法是在目标函数中使用 return 语句，并在主线程中使用 join() 方法等待线程完成。这种方法简单明了，但缺点是它会阻塞调用线程，直到目标函数完成执行。

使用队列

一种更灵活的方法是使用线程安全的队列，例如 Queue。目标函数可以将返回值放入队列中，然后主线程可以从队列中检索它。这种方法是非阻塞的，允许主线程在等待线程返回结果时继续执行其他任务。

使用 concurrent.futures 模块

Python 3.2 引入了 concurrent.futures 模块，它提供了一种高级的方法来管理线程和进程。该模块提供了 Future 对象，它代表异步操作的结果。主线程可以使用 result() 方法从 Future 对象中检索结果。这种方法提供了一个非阻塞的、面向未来的接口，简化了线程管理。

选择合适的方法

选择哪种方法取决于应用程序的特定需求。直接方法对于简单的用例来说很方便，但对于需要非阻塞执行或管理多个线程的情况来说，队列或 concurrent.futures 模块可能是更好的选择。

代码示例

直接方法：

import threading

def foo(bar):
    print('hello {}'.format(bar))
    return 'foo'

thread = threading.Thread(target=foo, args=('world!',))
thread.start()
return_value = thread.join()

print(return_value)  # 输出：foo

使用队列：

import threading
import queue

def foo(bar, q):
    print('hello {}'.format(bar))
    q.put('foo')

q = queue.Queue()
thread = threading.Thread(target=foo, args=('world!', q))
thread.start()
return_value = q.get()

print(return_value)  # 输出：foo

使用 concurrent.futures 模块：

import concurrent.futures

def foo(bar):
    print('hello {}'.format(bar))
    return 'foo'

with concurrent.futures.ThreadPoolExecutor() as executor:
    future = executor.submit(foo, 'world!')
    return_value = future.result()

print(return_value)  # 输出：foo