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()

运行注意：

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