Python中的threading.Thread
是一个常用的多线程编程工具,可以通过继承threading.Thread
类来创建自定义的线程类,并实现多线程编程。
在实际开发中,我们有时需要从多线程中获取返回值。本文将介绍如何通过继承threading.Thread
类,来创建具有返回值的自定义线程类,并分别给出两个示例。
一、使用queue
模块来传递返回值
首先,我们来介绍一种使用queue
模块来传递返回值的方法。具体的实现步骤如下:
-
导入
threading
和queue
模块。 -
创建自定义线程类,继承
threading.Thread
。 -
在自定义线程类的
__init__
函数中,将返回值队列queue.Queue()
传递进来,并赋值给实例属性。 -
在自定义线程类的
run
函数中,执行线程任务,并将结果通过put()
方法放入队列中。 -
在主线程中创建线程实例,并启动线程。在调用线程的
join()
方法等待线程执行完毕后,使用get()
方法从队列中取出线程的返回值。
示例代码如下:
import threading
import queue
class MyThread(threading.Thread):
def __init__(self, arg, result_queue):
super().__init__()
self.arg = arg
self.result_queue = result_queue
def run(self):
# 线程任务
result = self.arg * 2
self.result_queue.put(result)
if __name__ == '__main__':
result_queue = queue.Queue()
t = MyThread(3, result_queue)
t.start()
t.join()
result = result_queue.get()
print(result)
在本示例中,自定义的线程类MyThread
有两个参数:arg
和result_queue
。其中,arg
为传入的线程参数,result_queue
为返回值队列。
在run
函数中,先执行了线程任务,即将传入参数arg
乘以2得到结果。然后,通过put()
方法将结果result
放入result_queue
队列中。
在主线程中,首先创建了一个result_queue
对象作为返回值队列。然后,创建了MyThread
类的实例t
,并传入传递给线程的参数3
和返回值队列result_queue
,启动线程,等待线程执行完毕,使用get()
方法从返回值队列中获取结果。
实际运行结果:
6
二、使用concurrent.futures
模块来实现有返回值的子类实例
除了使用queue
模块传递返回值,还可以使用concurrent.futures
模块来实现打包进程和线程的高级接口,通过Future
对象获取返回值。
具体的实现步骤如下:
-
导入
concurrent.futures
模块。 -
创建自定义线程类,继承
threading.Thread
。 -
在自定义线程类中实现一个方法,用来执行线程任务。待调用此方法时,直接返回值。
-
在主线程中通过
concurrent.futures
模块的ThreadPoolExecutor
类创建线程池对象,调用submit
方法来提交线程任务。 -
通过
Future
的result
方法获取线程的返回值。
示例代码如下:
import concurrent.futures
import threading
class MyThread(threading.Thread):
def __init__(self, arg):
super().__init__()
self.arg = arg
def task(self):
# 线程任务
result = self.arg * 2
return result
if __name__ == '__main__':
t = MyThread(3)
with concurrent.futures.ThreadPoolExecutor() as executor:
future = executor.submit(t.task)
result = future.result()
print(result)
在本示例中,自定义的线程类MyThread
仅有一个参数:arg
。在task
方法中,执行了线程任务,即将传入参数arg
乘以2得到结果,并直接返回结果。
在主线程中,使用concurrent.futures
模块的ThreadPoolExecutor
类创建线程池对象,并通过submit
方法提交线程任务,返回值将会通过Future
对象获取,最后使用result
方法获取线程返回值。
实际运行结果:
6
两个示例说明了如何通过继承threading.Thread
类,来创建具有返回值的自定义线程类,并分别使用queue
模块和concurrent.futures
模块来获取线程返回值。可以根据实际需要选择合适的方法来实现自己的多线程编程需求。