Python 多线程 生产者消费者模型

my_thread.py

import threading
from time import ctime
class MyThread(threading.Thread):
    def __init__(self, func, args, name=''):
        threading.Thread.__init__(self)
        self.name = name
        self.func = func
        self.args = args

    def get_result(self):
        return self.res

    def run(self):
        print 'starting ', self.name, ' at:',ctime()
        self.res = apply(self.func, self.args)
        print self.name ,'finished at:' ctime()

queue.py:

from random import randint
from time import sleep
from Queue import Queue
from my_thread import MyThread

def write_q(queue):
    print 'producting object for Q...'
    queue.put('xxx',1)
    print 'size now: ',queue.qsize()

def read_q(queue):
    val = queue.get(1)
    print 'consumed object from Q... size now', queue.qsize()

def writer(queue, loops):
    for i in range(loops):
        write_q(queue)
        sleep(randint(1,3))

def reader(queue,loops):
    for i in range(loops):
        read_q(queue)
        sleep(randint(2,5))

funcs = [writer, reader]
nfuncs = range(len(funcs))

def main():
    nloops = randint(2,5)
    q = Queue(32)
    threads = []
    for i in nfuncs:
        t = my_thread.MyThread(funcs[i],(q,nloops),funcs[i].__name__)
        threads.append(t)
    for i in nfuncs:
        threads.start(i)

    for i in nfuncs:
        threads.join()

    print 'all done'