python使用rabbitmq实例四，路由键（4）

最新推荐文章于 2025-02-27 01:06:39 发布

转载最新推荐文章于 2025-02-27 01:06:39 发布 · 952 阅读

python 专栏收录该内容

68 篇文章

订阅专栏

第三篇说明了关于交换机的使用，已经能实现给所有接收端发送消息，但是如果需要自由定制，有的消息发给其中一些接收端，有些消息发送给另外一些接收端，要怎么办呢？这种情况下就要用到路由键了。

路由键的工作原理：每个接收端的消息队列在绑定交换机的时候，可以设定相应的路由键。发送端通过交换机发送信息时，可以指明路由键，交换机会根据路由键把消息发送到相应的消息队列，这样接收端就能接收到消息了。

这边继上一篇，还是用send.py和receive.py来模拟实现路由键的功能。send.py表示发送端，receive.py表示接收端。实例的功能就是将info、warning、error三种级别的信息发送到不同的接收端。

send.py代码分析

和上一篇相比，改动点主要在两个方面：

设定交换机的类型（type）为direct。上一篇是设置为fanout，表示广播的意思，会将消息发送到所有接收端，这里设置为direct表示要根据设定的路由键来发送消息。
发送信息时设置发送的路由键。

 
#!/usr/bin/env python
 
#coding=utf8
 
import pika
 
 
 
connection = pika.BlockingConnection(pika.ConnectionParameters(
 
               'localhost'))
 
channel = connection.channel()
 
 
 
#定义交换机，设置类型为direct
 
channel.exchange_declare(exchange='messages', type='direct')
 
 
 
#定义三个路由键
 
routings = ['info', 'warning', 'error']
 
 
 
#将消息依次发送到交换机，并设置路由键
 
for routing in routings:
 
    message = '%s message.' % routing
 
    channel.basic_publish(exchange='messages',
 
                          routing_key=routing,
 
                          body=message)
 
    print message
 
 
 
connection.close()

receive.py代码分析

和第三篇相比，改动点主要在三个方面：

设定交换机的类型（type）为direct。
增加命令行获取参数功能，参数即为路由键。
将队列绑定到交换机上时，设定路由键。

 
#!/usr/bin/env python
 
#coding=utf8
 
import pika, sys
 
 
 
connection = pika.BlockingConnection(pika.ConnectionParameters(
 
               'localhost'))
 
channel = connection.channel()
 
 
 
#定义交换机，设置类型为direct
 
channel.exchange_declare(exchange='messages', type='direct')
 
 
 
#从命令行获取路由键参数，如果没有，则设置为info
 
routings = sys.argv[1:]
 
if not routings:
 
    routings = ['info']
 
 
 
#生成临时队列，并绑定到交换机上，设置路由键
 
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()

打开两个终端，一个运行代码python receive.py info warning，表示只接收info和warning的消息。另外一个终端运行send.py，可以观察到接收终端只接收到了info和warning的消息。如果打开多个终端运行receive.py，并传入不同的路由键参数，可以看到更明显的效果。

当接收端正在运行时，可以使用rabbitmqctl list_bindings来查看绑定情况。

参考文章：http://www.rabbitmq.com/tutorials/tutorial-four-python.html