生产-消费者问题

# -*- coding: UTF-8 -*-
import threading
import time
from random import randint
from Queue import Queue

class MyThread(threading.Thread):

def __init__(self, func, args, name =""):
threading.Thread.__init__(self)
self.name = name
self.func = func
self.args = args

def run(self):
apply(self.func, self.args)

def writeQ(queue):
print "producing object for Q..."
queue.put("XXX", 1)
print "size now", queue.qsize()

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

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

def reader(queue, loops):
for i in range(loops):
readQ(queue)
time.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 = MyThread(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 done"

if __name__ == "__main__":
main()