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

運行注意：

　　先開啟消費端，用命令行指定接受級別