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

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

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

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

send.py代码分析

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

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

1#!/usr/bin/env python

2#coding=utf8

3import pika

4 

5connection = pika.BlockingConnection(pika.ConnectionParameters(

6               'localhost'))

7channel = connection.channel()

8 

9#定义交换机，设置类型为direct

10channel.exchange_declare(exchange='messages', type='direct')

11 

12#定义三个路由键

13routings = ['info', 'warning', 'error']

14 

15#将消息依次发送到交换机，并设置路由键

16for routing in routings:

17    message = '%s message.' % routing

18    channel.basic_publish(exchange='messages',

19                          routing_key=routing,

20                          body=message)

21    print message

22 

23connection.close()

receive.py代码分析

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

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

查看源代码

打印帮助

1#!/usr/bin/env python

2#coding=utf8

3import pika, sys

4 

5connection = pika.BlockingConnection(pika.ConnectionParameters(

6               'localhost'))

7channel = connection.channel()

8 

9#定义交换机，设置类型为direct

10channel.exchange_declare(exchange='messages', type='direct')

11 

12#从命令行获取路由键参数，如果没有，则设置为info

13routings = sys.argv[1:]

14if not routings:

15    routings = ['info']

16 

17#生成临时队列，并绑定到交换机上，设置路由键

18result = channel.queue_declare(exclusive=True)

19queue_name = result.method.queue

20for routing in routings:

21    channel.queue_bind(exchange='messages',

22                       queue=queue_name,

23                       routing_key=routing)

24 

25def callback(ch, method, properties, body):

26    print " [x] Received %r" % (body,)

27 

28channel.basic_consume(callback, queue=queue_name, no_ack=True)

29 

30print ' [*] Waiting for messages. To exit press CTRL+C'

31channel.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