zl程序教程

您现在的位置是:首页 >  后端

当前栏目

python多线程获取返回值

Python多线程 获取 返回值
2023-09-11 14:15:15 时间

在使用多线程的时候,通常有时候需要获取多线程操作完的返回值并进行其它操作。

实现思路:

  1. 首先重写threading类,使其满足调用特定的方法获取其返回值;
  2. 调用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多线程获取返回值