RabbitMQ 教程 - 路由
路由
(使用 Pika Python 客户端)
先决条件
与其他 Python 教程一样,我们将使用Pika RabbitMQ 客户端版本 1.0.0。
本教程的重点
在上一篇教程中,我们构建了一个简单的日志系统。我们能够将日志消息广播到许多接收方。
在本教程中,我们将向其中添加一项功能 - 我们将使其能够仅订阅消息的子集。例如,我们将能够仅将严重错误消息定向到日志文件(以节省磁盘空间),同时仍然能够在控制台上打印所有日志消息。
绑定
在之前的示例中,我们已经创建了绑定。您可能还记得类似的代码
channel.queue_bind(exchange=exchange_name,
queue=queue_name)
绑定是交换机和队列之间的一种关系。这可以简单地理解为:队列对来自此交换机的消息感兴趣。
绑定可以带有一个额外的 routing_key
参数。为了避免与 basic_publish
参数混淆,我们将称之为 绑定键
。以下是如何使用键创建绑定
channel.queue_bind(exchange=exchange_name,
queue=queue_name,
routing_key='black')
绑定键的含义取决于交换机类型。我们之前使用的 fanout
交换机只是忽略了它的值。
直接交换机
我们之前教程中的日志系统将所有消息广播到所有消费者。我们希望扩展它以允许根据消息的严重性过滤消息。例如,我们可能希望将日志消息写入磁盘的脚本仅接收严重错误,而不是浪费磁盘空间来存储警告或信息日志消息。
我们使用的是 fanout
交换机,它没有给我们太多灵活性 - 它只能进行无脑广播。
我们将改用 direct
交换机。direct
交换机背后的路由算法很简单 - 消息将发送到其 绑定键
与消息的 路由键
完全匹配的队列。
为了说明这一点,请考虑以下设置
在此设置中,我们可以看到 direct
交换机 X
以及与其绑定的两个队列。第一个队列使用绑定键 orange
绑定,第二个队列有两个绑定,一个使用绑定键 black
,另一个使用 green
。
在这种设置中,发布到交换机且路由键为 orange
的消息将被路由到队列 Q1
。路由键为 black
或 green
的消息将发送到 Q2
。所有其他消息将被丢弃。
多个绑定
使用相同的绑定键绑定多个队列是完全合法的。在我们的示例中,我们可以添加 X
和 Q1
之间的绑定,并使用绑定键 black
。在这种情况下,direct
交换机将表现得像 fanout
并将消息广播到所有匹配的队列。路由键为 black
的消息将同时传递到 Q1
和 Q2
。
发出日志
我们将使用此模型来构建我们的日志系统。我们将消息发送到 direct
交换机,而不是 fanout
。我们将提供日志严重性作为 路由键
。这样,接收脚本将能够选择其想要接收的严重性。让我们首先关注发出日志。
像往常一样,我们需要先创建一个交换机
channel.exchange_declare(exchange='direct_logs',
exchange_type='direct')
我们准备发送消息了
channel.basic_publish(exchange='direct_logs',
routing_key=severity,
body=message)
为了简化起见,我们将假设“严重性”可以是 info
、warning
或 error
之一。
订阅
接收消息的工作方式与上一篇教程相同,只有一个例外 - 我们将为我们感兴趣的每个严重性创建一个新的绑定。
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)
综合示例
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) > 1 else 'info'
message = ' '.join(sys.argv[2:]) or 'Hello World!'
channel.basic_publish(
exchange='direct_logs', routing_key=severity, body=message)
print(f" [x] Sent {severity}:{message}")
connection.close()
receive_logs_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')
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(f" [x] {method.routing_key}:{body}")
channel.basic_consume(
queue=queue_name, on_message_callback=callback, auto_ack=True)
channel.start_consuming()
如果您只想将“警告”和“错误”(而不是“信息”)日志消息保存到文件,只需打开一个控制台并输入
python receive_logs_direct.py warning error > logs_from_rabbit.log
如果您想在屏幕上查看所有日志消息,请打开一个新的终端并执行
python receive_logs_direct.py info warning error
# => [*] Waiting for logs. To exit press CTRL+C
例如,要发出 error
日志消息,只需输入
python emit_log_direct.py error "Run. Run. Or it will explode."
# => [x] Sent 'error':'Run. Run. Or it will explode.'
继续学习教程 5,了解如何根据模式侦听消息。