python操作rabbitmq实现消息过滤接收

目标：

　　

代码实现(direct_product.py)

 # __author__ = 'STEVEN'
 import pika,sys
 #开启socket
 connection = pika.BlockingConnection(pika.ConnectionParameters('localhost'))
 #开启一个通道
 channel = connection.channel()
 #这里不用再创建队列
 channel.exchange_declare(exchange='d_logs',exchange_type='direct')
 #通过命令行运行本代码，指定要发送的消息级别，如果不指定则为info
 severity = sys.argv[1] if len(sys.argv)>1 else 'info'
 #通过命令行输入消息级别后的参数
 mes = ''.join(sys.argv[2:]) or 'hello world'
 #添加了消息级别，即将消息发布到指定了级别的消息队列
 channel.basic_publish(exchange='d_logs',routing_key=severity,body=mes)
 print('[x] send the mes%s to queue'%mes)
 connection.close()

代码实现(direct_consumer.py)

 # __author__ = 'STEVEN'
 import pika,sys
 #建立socket
 connection = pika.BlockingConnection(pika.ConnectionParameters(host='localhost'))
 #开启通道
 channel = connection.channel()
 #通道先声明exchange类型
 channel.exchange_declare(exchange='d_logs',exchange_type='direct')
 #声明queue
 result = channel.queue_declare(exclusive=True)
 #获取queue_name
 queue_name = result.method.queue
 #从sys中获取消息级别等信息
 severitis = sys.argv[1:]
 if not severitis:
     sys.stderr.write('usage:%s [info] [error] [warning]'%sys.argv[0])
     sys.exit(1)
 #遍历绑定各个exchange和queue以及消息级别
 for severity in severitis:
     #绑定queue
     channel.queue_bind(exchange='d_logs',routing_key=severity,queue=queue_name)
 #回调函数
 def callback(ch,method,properties,body):
     print('[x] receive mess :%s'%body.decode())
 #指定消费相关参数
 channel.basic_consume(callback,queue=queue_name,no_ack=True)
 print('[*] is waiting for the message')
 #开启消费
 channel.start_consuming()

运行注意：

　　先开启消费端，用命令行指定接受级别