RabbitMQ学习笔记05：Routing
参考资料：RabbitMQ tutorial - Routing — RabbitMQ
前言
在之前的文章中我们构建了一个简单的日志系统，它可以广播消息到多个消费者中。
在这篇文章中，我们打算实现仅订阅消息的子集（即不是所有的消息，仅仅只是一部分消息。注意，这里不是说一条消息的一部分）。例如我们只会把严重的错误消息写到日志文件中，同时我们会把所有的消息都输出到屏幕上。
Bindings
在之前的示例中我们学会了绑定exchange和队列。
channel.queue_bind(exchange=exchange_name,
queue=queue_name)绑定是exchange和队列之间的关系，我们可以简单理解为某个队列对于来自于某个exchange的消息感兴趣。
在绑定的过程当中可以使用额外的routing_key参数。为了避免和basic_publish参数混淆，我们将其称之为binding_key。
其实binding_key就是exchange和队列在绑定的时候使用的routing_key。
下面是我们在绑定的时候创建了一个binding_key。
channel.queue_bind(exchange=exchange_name,
queue=queue_name,
routing_key='black')binding_key的含义取决于exchange的类型，我们之前创建的fanout类型的exchange会忽略这个参数的值。
Direct exchange
想要完成前言中我们提到的需求的话，我们需要使用direct类型的exchange。direct的路由算法也十分简单，发送给direct类型的exchange的消息一般会带有一个routing_key参数，这条消息会被路由到带有相同的binding_key值的队列中。这种匹配是必须准确无误的。
就像这样：
从图中我们可以看到，direct类型的名为x的exchange绑定了2个队列Q1和Q2，在与Q1绑定时的binding_key是orange，在与Q2绑定时共绑定了2次，第一个绑定的binding_key是black，第二个绑定的binding_key是green。
在这种情况下，发给x的消息，如果其routing_key是orange的话，则消息会被发送给队列Q1；如果其routing_key是black或者green的话，则消息会被发送给队列Q2；如果其routing_key是其他值的话，那么这条消息会被丢弃！
Multiple bindings
direct类型的exchange在绑定多个队列的时候可以使用相同的binding_key，这是被允许的。这种情况下的路由模式就类似于fanout即消息会被复制并广播到多个匹配的队列中去，只不过区别在于fanout不会丢弃消息，而direct的话，如果routing_key和binding_key没有匹配的话，消息就会被丢弃。
Emitting logs
我们将会采用这个新的模型来作为日志系统。日志的严重性我们用routing_key来表示。
让我们先聚焦提供日志的程序。首先我们声明一个exchange。
channel.exchange_declare(exchange='direct_logs',
exchange_type='direct')向该exchange发送消息。
channel.basic_publish(exchange='direct_logs',
routing_key=severity,
body=message)我们假设severity的取值为
- info
- warning
- error
Subscribing
接收消息的消费者程序和之前的差不多，区别在于我们会根据我们的兴趣和需要做绑定。
result = channel.queue_declare(queue='', exclusive=True)
queue_name = result.method.queue
for severity in severities:
channel.queue_bind(exchange='direct_logs',
queue=queue_name,
routing_key=severity)
Putting it all together
emit_log_direct.py
#!/usr/bin/env python
import pika
import sys
connection = pika.BlockingConnection(
pika.ConnectionParameters(host='localhost'))
channel = connection.channel()
channel.exchange_declare(exchange='direct_logs', exchange_type='direct')
severity = sys.argv[1] if len(sys.argv) > 2 else 'info'
message = ' '.join(sys.argv[2:]) or 'Hello World!'
channel.basic_publish(
exchange='direct_logs', routing_key=severity, body=message)
print(" [x] Sent %r:%r" % (severity, message))
connection.close()receive_logs_direct.py
#!/usr/bin/env python
import pika, sys, os
def main():
connection = pika.BlockingConnection(
pika.ConnectionParameters(host='localhost'))
channel = connection.channel()
channel.exchange_declare(exchange='direct_logs', exchange_type='direct')
result = channel.queue_declare(queue='', exclusive=True)
queue_name = result.method.queue
severities = sys.argv[1:]
if not severities:
sys.stderr.write("Usage: %s [info] [warning] [error]\n" % sys.argv[0])
sys.exit(1)
for severity in severities:
channel.queue_bind(
exchange='direct_logs', queue=queue_name, routing_key=severity)
print(' [*] Waiting for logs. To exit press CTRL+C')
def callback(ch, method, properties, body):
print(" [x] %r:%r" % (method.routing_key, body.decode()))
channel.basic_consume(
queue=queue_name, on_message_callback=callback, auto_ack=True)
channel.start_consuming()
if __name__ == '__main__':
try:
main()
except KeyboardInterrupt:
print('Interrupted')
try:
sys.exit(0)
except SystemExit:
os._exit(0)按照官方的测试的话，其实和上一篇博文一样，都会遇到文件写入未成功的问题。因此我们测试时，只能都直接输出到控制台了，这种测试方式不影响我们希望测试的功能。
第一个终端，启动一个仅接收error的消息的消费者进程。
python receive_logs_direct.py error第二个终端，启动一个仅接收info或者warning的消息的消费者进程。
python receive_logs_direct.py info warning error第三个终端用于运行生产者程序，每次使用不同的routing_key。
[root@rabbitmq-01 code]# python emit_log_direct.py error "Run. Run. Or it will explode."
[x] Sent 'error':'Run. Run. Or it will explode.'
[root@rabbitmq-01 code]# python emit_log_direct.py info "Run. Run. Or it will explode."
[x] Sent 'info':'Run. Run. Or it will explode.'
[root@rabbitmq-01 code]# python emit_log_direct.py warning "Run. Run. Or it will explode."
[x] Sent 'warning':'Run. Run. Or it will explode.'
[root@rabbitmq-01 code]# python emit_log_direct.py Disater "Run. Run. Or it will explode."
[x] Sent 'Disater':'Run. Run. Or it will explode.'4条消息发好之后，我们来看下两个终端的结果。
# 第一个终端
[root@rabbitmq-01 code]# python receive_logs_direct.py error
[*] Waiting for logs. To exit press CTRL+C
[x] 'error':'Run. Run. Or it will explode.'
^CInterrupted
# 第二个终端
[root@rabbitmq-01 code]# python receive_logs_direct.py info warning error
[*] Waiting for logs. To exit press CTRL+C
[x] 'error':'Run. Run. Or it will explode.'
[x] 'info':'Run. Run. Or it will explode.'
[x] 'warning':'Run. Run. Or it will explode.'
^CInterrupted我们得到以下信息：
- error级别的信息在两个终端上都显示了。因为两个终端对应的队列的
binding_key都包含error。 - info和warning级别的信息只会在第二个终端上显示。因为只有第二个终端对应的队列的
binding_key包含info和warning。 - Disater级别的信息没有被显示。因为两个队列的
binding_key都没有包含Disater，不匹配消息就会被丢弃，这是和fanout的区别。
总结
这篇博文介绍了direct类型的exchange，使得消息可以根据routing_key来路由到不同的队列中。
