之前在开发对比测试工具时,需要同步从数据库拉取数据,拉取数据时间比较长,使用到了多线程。但是由于多线程没有返回值,只能执行,因此在搜索相关资料以后,进行了改进,重新封装了一下。
以下为封装后的以及测试代码:
import threading
import time
class MyThread(threading.Thread):
def __init__(self,func,args=(),kwargs=()):
super(MyThread,self).__init__()
self.func = func
self.args = args
self.kwargs = kwargs
def run(self):
self.result = self.func(*self.args,*self.kwargs)
def get_result(self):
try:
return self.result # 如果子线程不使用join方法,此处可能会报没有self.result的错误
except Exception:
return None
def foo(a,b,c):
time.sleep(1)
return a*2,b*2,c*2
st = time.time()
li = []
for i in range(4):
t = MyThread(foo,args=(i,i+1,i+2))
li.append(t)
t.start()
li[0].join()
li[0].get_result()
et = time.time()
print(et - st)