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

3	import pika

4

5	connection = pika.BlockingConnection(pika.ConnectionParameters(

6	'localhost'))

7	channel = connection.channel()

8

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

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

11

12

#定义三个路由键

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

14

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

16	for 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

23	connection.close()

receive.py代码分析

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

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

查看源代码

打印 帮助

1	#!/usr/bin/env python

2	#coding=utf8

3	import pika, sys

4

5	connection = pika.BlockingConnection(pika.ConnectionParameters(

6	'localhost'))

7	channel = connection.channel()

8

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

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

11

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

13	routings = sys.argv[1:]

14	if not routings:

15	routings = ['info']

16

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

18	result = channel.queue_declare(exclusive=True)

19	queue_name = result.method.queue

20	for routing in routings:

21	channel.queue_bind(exchange='messages',

22	queue=queue_name,

23	routing_key=routing)

24

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

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

27

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

29

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

31	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