我的方法:
添加一个 flag 标识。
先看结果吧:
更多废话也不多说了,show u the code
#!/usr/bin/python
# coding=utf-8
import threading
import time
class Producer(threading.Thread):
def __init__(self, queue, flag):
super(Producer, self).__init__()
self.queue = queue
self.flag = flag
def run(self):
while True:
length = max(self.queue) + 1
print "============================= producer queue", self.queue
self.queue.append(length)
print 'flag length=', len(self.flag)
if len(self.flag) == 0:
print "producer 我也结束了"
break
time.sleep(2)
class Consumer(threading.Thread):
def __init__(self, queue, flag):
super(Consumer, self).__init__()
self.queue = queue
self.flag = flag
def run(self):
while True:
try:
length = len(self.queue)
print "consumer queue", self.queue
if length > 5:
self.flag.pop() # 注意我是flag
raise ValueError('over')
self.queue.pop(0)
except ValueError as e:
# 通知生产者结束
# 如何实现?
print "consumer 我结束了", e
break
# raise(e)
time.sleep(4)
queue = [1, 2, 3]
flag = [0] # 表示正常
Consumer(queue, flag).start()
time.sleep(1)
Producer(queue, flag).start()
最后说说python的多线程,由于GIL的存在,其实多线程有的时候并不是最好的选择,具体什么时候使用,网上也说的很多了,楼主也可以结合自己的业务情况舍取多线程模块。