天天看点

python生产者消费者写线程池_python中生产者消费者线程问题

我的方法:

添加一个 flag 标识。

先看结果吧:

python生产者消费者写线程池_python中生产者消费者线程问题

更多废话也不多说了,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的存在,其实多线程有的时候并不是最好的选择,具体什么时候使用,网上也说的很多了,楼主也可以结合自己的业务情况舍取多线程模块。