詳解Python獲取線程返回值的三種方式
提到線程,你的大腦應該有這樣的印象:我們可以控制它何時開始,卻無法控制它何時結束,那麼如何獲取線程的返回值呢?今天就分享一下自己的一些做法。
方法一
使用全局變量的列表,來保存返回值
ret_values = [] def thread_func(*args): ... value = ... ret_values.append(value)
選擇列表的一個原因是:列表的 append() 方法是線程安全的,CPython 中,GIL 防止對它們的並發訪問。如果你使用自定義的數據結構,在並發修改數據的地方需要加線程鎖。
如果事先知道有多少個線程,可以定義一個固定長度的列表,然後根據索引來存放返回值,比如:
from threading import Thread threads = [None] * 10 results = [None] * 10 def foo(bar, result, index): result[index] = f"foo-{index}" for i in range(len(threads)): threads[i] = Thread(target=foo, args=('world!', results, i)) threads[i].start() for i in range(len(threads)): threads[i].join() print (" ".join(results))
方法二
重寫 Thread 的 join 方法,返回線程函數的返回值
默認的 thread.join() 方法隻是等待線程函數結束,沒有返回值,我們可以在此處返回函數的運行結果,代碼如下:
from threading import Thread def foo(arg): return arg class ThreadWithReturnValue(Thread): def run(self): if self._target is not None: self._return = self._target(*self._args, **self._kwargs) def join(self): super().join() return self._return twrv = ThreadWithReturnValue(target=foo, args=("hello world",)) twrv.start() print(twrv.join()) # 此處會打印 hello world。
這樣當我們調用 thread.join() 等待線程結束的時候,也就得到瞭線程的返回值。
方法三
使用標準庫 concurrent.futures
我覺得前兩種方式實在太低級瞭,Python 的標準庫 concurrent.futures 提供更高級的線程操作,可以直接獲取線程的返回值,相當優雅,代碼如下:
import concurrent.futures def foo(bar): return bar with concurrent.futures.ThreadPoolExecutor(max_workers=10) as executor: to_do = [] for i in range(10): # 模擬多個任務 future = executor.submit(foo, f"hello world! {i}") to_do.append(future) for future in concurrent.futures.as_completed(to_do): # 並發執行 print(future.result())
某次運行的結果如下:
hello world! 8
hello world! 3
hello world! 5
hello world! 2
hello world! 9
hello world! 7
hello world! 4
hello world! 0
hello world! 1
hello world! 6
最後的話
本文分享瞭獲取線程返回值的 3 種方法,推薦使用第三種
到此這篇關於詳解Python獲取線程返回值的三種方式的文章就介紹到這瞭,更多相關Python獲取線程返回值內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!
推薦閱讀:
- python使用期物處理並發教程
- 一文詳解Python中多進程和進程池的使用方法
- Python並發編程實例教程之線程的玩法
- Python一些線程的玩法總結
- Python線程池thread pool創建使用及實例代碼分享