python线程实现生产者消费者模型

思路

1、创建一个线程池类，包括线程列表，任务队列，结果队列，创建线程数等属性，方法包括__init__方法，创建线程方法，启动线程方法，停止线程方法，添加任务方法，获取任务结果方法
2、创建一个工作类，继承Thread类，包括任务队列，结果队列，执行参数等属性，方法包括__init__方法，run方法

代码

import threading
import Queue

# 1、创建线程池类
class WorkManage(object):
    def __init__(self, thread_num):
        """
        workQueue：任务队列
        resultQueue：结果队列
        """
        self.workQueue = Queue.Queue()
        self.resultQueue = Queue.Queue()
        self.thread_list = []
        self.thread_num = thread_num

    def _create_work(self):
        """
        创建线程
        """
        for i in range(self.thread_num):
            work = WorkThread(self.workQueue, self.resultQueue)
            self.thread_list.append(work)

    def start(self):
        """
        启动线程
        """
        for work in self.thread_list:
            work.start()

    def stop(self):
        """
        停止线程
        """
        while len(self.thread_list):
            work = self.thread_list.pop()
            work.join()
            if work.isAlive() and not self.workQueue.empty():
                self.thread_list.append(work)

    def add_job(self, callback, *args, **kwgs):
        """
        添加任务
        """
        self.workQueue.put(callback, *args, **kwgs)

    def get_result(self, *args, **kwgs):
        """
        获取任务结果
        """
        return self.resultQueue.get(*args, **kwgs)


# 2、创建线程类
class WorkThread(threading.Thread):
    def __init__(self, workQueue, resultQueue, **kwgs):
        threading.Thread.__init__(**kwgs)
        self.workQueue = workQueue
        self.resultQueue = resultQueue

    def run(self):
        while True:
            try:
                callback, args, kwgs = self.workQueue.get(False)
                res = callback(*args, **kwgs)
                self.resultQueue.put(res)

            except:
                break