python线程池回调函数_python多线程队列处理任务的问题

import time

import Queue

import threading

import traceback

class Worker(threading.Thread):

def __init__(self, work_manager):

threading.Thread.__init__(self)

self.work_queue = work_manager.work_queue

self.done_queue = work_manager.done_queue

self.start()

def run(self):

while True:

try:

do, arg, kwargs = self.work_queue.get(block=True)

result = do(*arg, **kwargs)

self.done_queue.put((do, arg, result))

except Exception, e:

print traceback.format_exc()

print self.getName(), "end", str(e)

break

class WorkManager(object):

def __init__(self, thread_num=2):

self.work_queue = Queue.Queue() # 队列对象

self.done_queue = Queue.Queue() # 执行完成的任务

self.threads = []

self._init_thread_pool(thread_num)

def _init_thread_pool(self, thread_num):

"""初始化线程"""

for i in range(thread_num):

worker = Worker(self)

self.threads.append(worker)

def add_job(self, job, arg=[], kwargs={}):

"""初始化工作队列"""

self.work_queue.put((job, arg, kwargs))

work_manager = WorkManager()

work_manager.add_job(sum, [[1, 2, 3]])

work_manager.add_job(max, [[2, 3, 4]])

work_manager.add_job(len, ['Hello World'])

time.sleep(1)

while not work_manager.done_queue.empty():

print work_manager.done_queue.get(False)

不需要do_job方法，通过work_manager.add_job来发布任务给线程池

线程中Queue.get需要block

task_done方法是配合Queue.join使用的，这里应该不需要

任务执行成功后把执行结果放在done_queue中

方法的注释应该写在函数名的下面。