python多线程获取返回值类重新封装

之前在开发对比测试工具时,需要同步从数据库拉取数据,拉取数据时间比较长,使用到了多线程。但是由于多线程没有返回值,只能执行,因此在搜索相关资料以后,进行了改进,重新封装了一下。
以下为封装后的以及测试代码:

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)

猜你喜欢

转载自blog.csdn.net/xinxuyang123/article/details/83820325
今日推荐