生产消费者模式

#!/usr/bin/env python

# -*- coding: utf-8 -*-

import threading
from time import ctime

from random import randint
from time import sleep
from Queue import Queue

class CThread(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 '%s starting at %s.' % (self.name, ctime())
        self.res = apply(self.func, self.args)
        print '%s finished at %s.' % (self.name, ctime())
        
def writeQ(queue):
    print 'producing object for Q ...'
    queue.put('item', 1)
    print 'queue size now %d' % queue.qsize()
    
def readQ(queue):
    val = queue.get(1)
    print 'consumed object from Q ... size now %d.' % queue.qsize()
    
def writer(queue, loops):
    for i in range(loops):
        writeQ(queue)
        sleep(randint(1, 3))
        
def reader(queue, loops):
    for i in range(loops):
        readQ(queue)
        sleep(randint(2, 5))
        
def main():
    funcs = [writer, reader]
    nfuncs = range(len(funcs))
    
    nloops = randint(2, 5)
    q = Queue(32)
    
    threads = []
    for i in nfuncs:
        t = CThread(funcs[i], (q, nloops), funcs[i].__name__)
        threads.append(t)
        
    for i in nfuncs:
        threads[i].start()
        
    for i in nfuncs:
        threads[i].join()
        
    print "all task DONE"
    
if __name__ == '__main__':
    main()