【Python】实现线程池

修改【立即终止符self.isterminal】为True；
但是该方法不能终止以下类型的线程，即任务已经全部完成，但是还有一些线程处于68行代码处的阻塞状态，因此还需要向待处理任务队列中put【任务完成符StopTask】，直到self.generate_list为空，即所有线程均被销毁。
最后将任务队列中多余的StopTask清空，但是python3中的Queue已经没有清空clear()的方法的了.

二.线程池代码块

#!/usr/bin/env python
# -*- coding:utf-8 -*-

import queue
import threading
import time

StopTask = object()


class threadpool():
    def __init__(self, max_num=5):
        self.task = queue.Queue()
        self.max_num = max_num

        self.generate_list = []
        self.free_list = []

        self.isterminal = False

    def run(self, func, args, callback=None):
        """
        线程池启动函数
        :param func: 函数名称
        :param args: 函数参数
        :param callback: 回调函数
        :return:
        """
        task = (func, args, callback)
        self.task.put(task)

        if len(self.free_list) == 0 and len(self.generate_list) < self.max_num:
            self.generate_thread()

    def generate_thread(self):
        """创建线程"""
        t = threading.Thread(target=self.call)
        t.start()

    def call(self):
        """执行线程"""
        current_thread = threading.currentThread
        self.generate_list.append(current_thread)

        get_task = self.task.get()
        while get_task != StopTask:
            # 执行任务
            func, args, callback = get_task
            try:
                ret = func(*args)
                status = True
            except Exception as e:
                ret = e
                status = False
            # 执行回调函数
            if callback:
                try:
                    callback(status, ret)
                except Exception:
                    pass
            # 处理任务or立即终止
            if self.isterminal:
                # 立即终止
                get_task = StopTask
            else:
                # 空闲处理
                self.free_list.append(current_thread)
                get_task = self.task.get()
                self.free_list.remove(current_thread)
        else:
            # 销毁线程
            self.generate_list.remove(current_thread)

    def close(self):
        """
        等待任务全部完成后，清空所有线程
        :return:
        """
        num = len(self.generate_list)
        while num:
            self.task.put(StopTask)
            num -= 1

    def terminal(self):
        """
        立即终止所有的线程
        :return:
        """
        self.isterminal = True

        while self.generate_list:
            self.task.put(StopTask)

        self.task.empty()

三.线程池使用代码：

def func(i):
    """
    待处理任务
    :param i:
    :return:
    """
    time.sleep(0.5)
    print(i)


def call(status, ret):
    """
    回调函数
    :param status: True:任务运行无误；False：报异常
    :param ret: 任务运行返回值/异常信息
    :return:
    """
    pass


if __name__ == '__main__':
    #创建线程池
    pool = threadpool(5)
    #执行任务
    for i in range(20):
        pool.run(func, args=(i,), callback=call)
    #终止任务
    pool.close()