python消息队列Queue

#coding=utf-8
from multiprocessing import Queue 

q = Queue(3)#初始化一个Queue对象，最多可接收三条put消息
q.put('message-1')
q.put('message-2')
print(q.full())#False,是否满了
q.put('message-3')
print(q.full())#True

#因为消息队列已满，下面的try都会抛出异常，第一个try会等待2秒后再抛出异常，第二个try会立即抛出异常
try:
	q.put('message-4',True,2)
except:
	print('except1,消息队列已满，现有消息数量：%s'%q.qsize())

try:
	q.put_nowait('message-4')
except:
	print('except2,消息队列已满，现有消息数量：%s'%q.qsize())

#判断队列是否已满
if not q.full():
	q.put_nowait('message-4')

#读取消息时，先判断消息队列是否为空，在读取
if not q.empty():
	for i in range(q.qsize()):
		print(q.get())#q.get会阻塞，q.get_nowait()不阻塞，但会抛异常 					

from multiprocessing import Process,Queue
import os,time,random 

#写数据
def write(q):
	for value in ['A','B','C']:
		print('Put %s to queue...'%value)
		q.put(value)
		time.sleep(random.random())

#读数据
def read(q):
	while True:
		if not q.empty():
			value = q.get(True)
			print('Get %s from queue...'%value)
			time.sleep(random.random())
		else:
			break

if __name__ == '__main__':
	print('start...')
	q = Queue()
	#父进程的queue传递给子进程
	pw = Process(target=write,args=(q,))
	pr = Process(target=read,args=(q,))				
	#写进程
	pw.start()
	pw.join()
	#读进程
	pr.start()
	pr.join()
	print('done...')

from multiprocessing import Manager,Pool
import os,time,random 

#写数据
def writer(q):
	print('writer启动(%s),父进程为(%s)'%(os.getpid(),os.getppid()))
	for i in 'chaoge':
		q.put(i)

#读数据
def reader(q):
	print('reader启动(%s),父进程为(%s)'%(os.getpid(),os.getppid()))
	for i in range(q.qsize()):
		print('reader 从Queue获取到消息:%s'%q.get())


if __name__ == '__main__':
	print('(%s) start'%os.getpid())
	q = Manager().Queue()#使用Manager中的Queue来初始化
	po=Pool()
	#使用阻塞模式创建进程，这样就不需要再reader中使用死循环了，可以等write执行完成后，再用reader
	po.apply(writer,(q,))
	po.apply(reader,(q,))
	#写进程
	po.close()
	po.join()
	print('(%s) End'%os.getpid())