日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學無先后,達者為師

網站首頁 編程語言 正文

Python 如何獲取線程的返回值

作者:別出BUG求求了 更新時間: 2023-11-17 編程語言

線程,按正常的思路,我們可以控制它何時開始,卻無法控制它何時結束,那么如何獲取線程的返回值呢?

一、使用全局變量的列表,來保存返回值

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

原文鏈接:https://blog.csdn.net/weixin_39589455/article/details/126954318

  • 上一篇:沒有了
  • 下一篇:沒有了
欄目分類
最近更新