python-线程池

#!/usr/bin/env python
# -*- coding:utf-8 -*-
import queue
import threading
import contextlib
import time

StopEvent = object()  #定义标记的意义在于任务结束后退出的标记 


class ThreadPool(object):

    def __init__(self, max_num):
        self.q = queue.Queue() #定义队列无限大
        self.max_num = max_num  #定义最大值

        self.terminal = False  #定义为false
        self.generate_list = [] #多少个进程正在执行
        self.free_list = []   #定义空闲列表--空闲线程  初始化各种属性

    def run(self, func, args, callback=None):
        """
        线程池执行一个任务
        :param func: 任务函数
        :param args: 任务函数所需参数
        :param callback: 任务执行失败或成功后执行的回调函数，回调函数有两个参数1、任务函数执行状态；2、任务函数返回值（默认为None，即：不执行回调函数）
        :return: 如果线程池已经终止，则返回True否则None
        """

        if len(self.free_list) == 0 and len(self.generate_list) < self.max_num:
            self.generate_thread()
        w = (func, args, callback,)
        self.q.put(w)

    def generate_thread(self):
        """
        创建一个线程
        """
        t = threading.Thread(target=self.call)#并执行call方法
        t.start()

    def call(self):
        """
        循环去获取任务函数并执行任务函数
        """
        current_thread = threading.currentThread #拿到当前线程
        self.generate_list.append(current_thread) #添加到正在使用线程队列

        event = self.q.get() #--这里是一个获取到的元组w=()....
        while event != StopEvent:

            func, arguments, callback = event # w= (func, args, callback)
            try:
                result = func(*arguments)  #执行任务  ret = aaa() --def aaa():   return 1
                status = True      
            except Exception as e: #如果我这个任务报错
                status = False
                result = e

            if callback is not None:  #这是一个返回值
                try:
                    callback(status, result)
                except Exception as e:
                    pass

            self.free_list.append(current_thread)  #---记得看上下文代码，执行完任务，把这个线程放到空闲队列里面
            event = self.q.get()#当前的状态应该是没任务，线程等待任务，不结束
            self.free_list.remove(current_thread)  #获取任务移除休息线程
        else:
            if event = StopEvent:
            self.generate_list.remove(current_thread) #移除当前正在运行的线程，等他运行完

    def close(self):
        num = len(self.generate_list)
        while num:
            self.q.put(StopEvent)#
            num -= 1


import time

def work(i):
    print(i)

pool = ThreadPool(10)   #定义最大线程为10个，实例化，并初始化
for item in range(50):  #创建了50个任务
    pool.run(func=work, args=(item,))  #执行work函数，args是传参

pool.close() #关闭线程池

理解注释版线程池