多线程装饰器完整实例（控制线程数量及简单使用）

聪少少灬

已于 2022-02-20 02:03:14 修改

阅读量866

点赞数 1

分类专栏： Python 算法数据分析文章标签： python 多线程装饰器

于 2022-02-19 23:56:29 首次发布

本文链接：https://blog.csdn.net/sGDUTBMW/article/details/123025950

版权

Python 同时被 3 个专栏收录

52 篇文章 0 订阅

订阅专栏

数据分析

11 篇文章 0 订阅

订阅专栏

算法

4 篇文章 0 订阅

订阅专栏

以下案例是通用的带参的类装饰器
其他的带参与不带参函数装饰器以及不带参类装饰器案例详见本人博客内其他博文

import threading
import time


class NewThread(object):
    # __init__ : 不再接收被装饰函数，而是接收传入参数.
    # __call__ ：接收被装饰函数，实现装饰逻辑.

    def __init__(self, max_thread=500):
        self.max_thread = max_thread

    def __call__(self, func):  # 接受函数
        from functools import wraps

        @wraps(func)
        def wrapper(*args, **kwargs):
            while True:
                # 获取已启动此函数线程的数量
                func_thread_active_count = len([i for i in threading.enumerate() if i.name == func.__name__])

                if func_thread_active_count <= self.max_thread:
                    # 待执行功能
                    # func(*args, **kwargs)
                    thread = threading.Thread(target=func, args=args, kwargs=kwargs, name=func.__name__)
                    # 把主线程设置为守护线程，主线程执行结束了，就不管子线程是否完成,一并和主线程退出
                    # thread.setDaemon(True)
                    thread.start()
                    # 子线程调用join()方法，使主线程等待子线程运行完毕之后才退出，与thread.setDaemon(True)相反
                    # thread.join()
                    break
                # else:
                #     time.sleep(0.01)

        return wrapper  # 返回函数


@NewThread(200)
def say(something):
    print("say {}!".format(something))
    time.sleep(1)


t1 = time.time()
for i in range(2000):
    say("hello")
t2 = time.time()
print('运行耗时：' + str(round(t2 - t1, 2)) + ' s')  # 50.19s