快捷導(dǎo)航

詳解Python獲取線程返回值的三種方式

更新時間：2022年07月06日 09:05:24 作者：somenzz

提到線程，你的大腦應(yīng)該有這樣的印象：我們可以控制它何時開始，卻無法控制它何時結(jié)束，那么如何獲取線程的返回值呢？今天就和大家分享一下一些做法

方法一

使用全局變量的列表，來保存返回值

ret_values?=?[]

def?thread_func(*args):
????...
????value?=?...
????ret_values.append(value)

選擇列表的一個原因是：列表的 append() 方法是線程安全的，CPython 中，GIL 防止對它們的并發(fā)訪問。如果你使用自定義的數(shù)據(jù)結(jié)構(gòu)，在并發(fā)修改數(shù)據(jù)的地方需要加線程鎖。

如果事先知道有多少個線程，可以定義一個固定長度的列表，然后根據(jù)索引來存放返回值，比如：

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 方法，返回線程函數(shù)的返回值

默認的 thread.join() 方法只是等待線程函數(shù)結(jié)束，沒有返回值，我們可以在此處返回函數(shù)的運行結(jié)果，代碼如下：

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。

這樣當我們調(diào)用 thread.join() 等待線程結(jié)束的時候，也就得到了線程的返回值。

方法三

使用標準庫 concurrent.futures

我覺得前兩種方式實在太低級了，Python 的標準庫 concurrent.futures 提供更高級的線程操作，可以直接獲取線程的返回值，相當優(yōu)雅，代碼如下：

import?concurrent.futures


def?foo(bar):
????return?bar


with?concurrent.futures.ThreadPoolExecutor(max_workers=10)?as?executor:
????to_do?=?[]
????for?i?in?range(10):??#?模擬多個任務(wù)
????????future?=?executor.submit(foo,?f"hello?world!?{i}")
????????to_do.append(future)

????for?future?in?concurrent.futures.as_completed(to_do):??#?并發(fā)執(zhí)行
????????print(future.result())

某次運行的結(jié)果如下：