在使用多线程的时候,通常有时候需要获取多线程操作完的返回值并进行其它操作。
实现思路:
- 首先重写threading类,使其满足调用特定的方法获取其返回值;
- 调用get_result()方法,最终将每个线程的列表结果返回
示例代码:
import threading
import time
from queue import Queue
class ThreadBase(threading.Thread):
"""重写多线程,使其能够有返回值"""
def __init__(self, target=None, args=()):
super().__init__()
self.func = target
self.args = args
def run(self):
self.result = self.func(*self.args)
def get_result(self):
try:
return self.result # 如果子线程不使用join方法,此处可能会报没有self.result的错误
except Exception as e:
print(e)
return None
def task_func(num):
ret = []
while queue.qsize() > 0:
data = queue.get()
ret.append(num + '-' + str(data))
time.sleep(0.1)
return ret
if __name__ == '__main__':
results = []
_Thread = []
queue = Queue(maxsize=50)
for i in range(30):
queue.put(i)
for i in range(3):
res = ThreadBase(target=task_func, args=(f"进程{i}", ))
res.start()
_Thread.append(res)
for res in _Thread:
res.join()
results.extend(res.get_result())
print(results)
运行结果:
参考博文:
Python多线程获取返回值