快捷導(dǎo)航

Python操作RabbitMQ服務(wù)器實(shí)現(xiàn)消息隊(duì)列的路由功能

更新時(shí)間：2016年06月29日 18:59:57 作者：陳杰斌

RabbitMQ是一個(gè)消息隊(duì)列服務(wù)器,這里我們針對(duì)Python+Pika+RabbitMQ的服務(wù)器端環(huán)境,來看一下如何使用Python操作RabbitMQ服務(wù)器實(shí)現(xiàn)消息隊(duì)列的路由功能

Python使用Pika庫(kù)（安裝：sudo pip install pika）可以操作RabbitMQ消息隊(duì)列服務(wù)器（安裝：sudo apt-get install rabbitmq-server），這里我們來看一下MQ相關(guān)的路由功能。

路由鍵的實(shí)現(xiàn)

比如有一個(gè)需要給所有接收端發(fā)送消息的場(chǎng)景，但是如果需要自由定制，有的消息發(fā)給其中一些接收端，有些消息發(fā)送給另外一些接收端，要怎么辦呢？這種情況下就要用到路由鍵了。

路由鍵的工作原理：每個(gè)接收端的消息隊(duì)列在綁定交換機(jī)的時(shí)候，可以設(shè)定相應(yīng)的路由鍵。發(fā)送端通過交換機(jī)發(fā)送信息時(shí)，可以指明路由鍵，交換機(jī)會(huì)根據(jù)路由鍵把消息發(fā)送到相應(yīng)的消息隊(duì)列，這樣接收端就能接收到消息了。

這邊繼上一篇，還是用send.py和receive.py來模擬實(shí)現(xiàn)路由鍵的功能。send.py表示發(fā)送端，receive.py表示接收端。實(shí)例的功能就是將info、warning、error三種級(jí)別的信息發(fā)送到不同的接收端。

send.py代碼分析

#!/usr/bin/env python
#coding=utf8
import pika
 
connection = pika.BlockingConnection(pika.ConnectionParameters(
        'localhost'))
channel = connection.channel()
 
#定義交換機(jī)，設(shè)置類型為direct
channel.exchange_declare(exchange='messages', type='direct')
 
#定義三個(gè)路由鍵
routings = ['info', 'warning', 'error']
 
#將消息依次發(fā)送到交換機(jī)，并設(shè)置路由鍵
for routing in routings:
  message = '%s message.' % routing
  channel.basic_publish(exchange='messages',
             routing_key=routing,
             body=message)
  print message
 
connection.close()

receive.py代碼分析

#!/usr/bin/env python
#coding=utf8
import pika, sys
 
connection = pika.BlockingConnection(pika.ConnectionParameters(
        'localhost'))
channel = connection.channel()
 
#定義交換機(jī)，設(shè)置類型為direct
channel.exchange_declare(exchange='messages', type='direct')
 
#從命令行獲取路由鍵參數(shù)，如果沒有，則設(shè)置為info
routings = sys.argv[1:]
if not routings:
  routings = ['info']
 
#生成臨時(shí)隊(duì)列，并綁定到交換機(jī)上，設(shè)置路由鍵
result = channel.queue_declare(exclusive=True)
queue_name = result.method.queue
for routing in routings:
  channel.queue_bind(exchange='messages',
            queue=queue_name,
            routing_key=routing)
 
def callback(ch, method, properties, body):
  print " [x] Received %r" % (body,)
 
channel.basic_consume(callback, queue=queue_name, no_ack=True)
 
print ' [*] Waiting for messages. To exit press CTRL+C'
channel.start_consuming()

打開兩個(gè)終端，一個(gè)運(yùn)行代碼python receive.py info warning，表示只接收info和warning的消息。另外一個(gè)終端運(yùn)行send.py，可以觀察到接收終端只接收到了info和warning的消息。如果打開多個(gè)終端運(yùn)行receive.py，并傳入不同的路由鍵參數(shù)，可以看到更明顯的效果。

當(dāng)接收端正在運(yùn)行時(shí)，可以使用rabbitmqctl list_bindings來查看綁定情況。

路由鍵模糊匹配
路由鍵模糊匹配，就是可以使用正則表達(dá)式，和常用的正則表示式不同，這里的話“#”表示所有、全部的意思；“*”只匹配到一個(gè)詞?？赐晔纠湍苊靼琢恕?/p>

這邊繼上面的例子，還是用send.py和receive.py來實(shí)現(xiàn)路由鍵模糊匹配的功能。send.py表示發(fā)送端，receive.py表示接收端。實(shí)例的功能大概是這樣：比如你有個(gè)知心好朋友，不管開心、傷心、工作上的還是生活上的事情都可以和她說；還有一些朋友可以分享開心的事情；還有一些朋友，你可以把不開心的事情和她說。

send.py代碼分析

因?yàn)橐M(jìn)行路由鍵模糊匹配，所以交換機(jī)的類型要設(shè)置為topic，設(shè)置為topic，就可以使用#，*的匹配符號(hào)了。

#!/usr/bin/env python
#coding=utf8
import pika
 
connection = pika.BlockingConnection(pika.ConnectionParameters(
        'localhost'))
channel = connection.channel()
 
#定義交換機(jī)，設(shè)置類型為topic
channel.exchange_declare(exchange='messages', type='topic')
 
#定義路由鍵
routings = ['happy.work', 'happy.life', 'sad.work', 'sad.life']
 
#將消息依次發(fā)送到交換機(jī)，并設(shè)定路由鍵
for routing in routings:
  message = '%s message.' % routing
  channel.basic_publish(exchange='messages',
             routing_key=routing,
             body=message)
  print message
 
connection.close()

上例中定義了四種類型的消息，容易理解，就不解釋了，然后依次發(fā)送出去。

receive.py代碼分析

同樣，交換機(jī)的類型要設(shè)定為topic就可以了。從命令行接收參數(shù)的功能稍微調(diào)整了一下，就是沒有參數(shù)時(shí)報(bào)錯(cuò)退出。

#!/usr/bin/env python
#coding=utf8
import pika, sys
 
connection = pika.BlockingConnection(pika.ConnectionParameters(
        'localhost'))
channel = connection.channel()
 
#定義交換機(jī)，設(shè)置類型為topic
channel.exchange_declare(exchange='messages', type='topic')
 
#從命令行獲取路由參數(shù)，如果沒有，則報(bào)錯(cuò)退出
routings = sys.argv[1:]
if not routings:
  print >> sys.stderr, "Usage: %s [routing_key]..." % (sys.argv[0],)
  exit()
 
#生成臨時(shí)隊(duì)列，并綁定到交換機(jī)上，設(shè)置路由鍵
result = channel.queue_declare(exclusive=True)
queue_name = result.method.queue
for routing in routings:
  channel.queue_bind(exchange='messages',
            queue=queue_name,
            routing_key=routing)
 
def callback(ch, method, properties, body):
  print " [x] Received %r" % (body,)
 
channel.basic_consume(callback, queue=queue_name, no_ack=True)
 
print ' [*] Waiting for messages. To exit press CTRL+C'
channel.start_consuming()

打開四個(gè)終端，一個(gè)運(yùn)行如下，表示任何事情都可以和她說：