跳至主要内容

RabbitMQ 教程 - 路由

路由​

(使用 Pika Python 客户端)​

信息

先决条件​

本教程假设 RabbitMQ 已安装并在 localhost 上的标准端口 (5672) 上运行。如果您使用不同的主机、端口或凭据，则连接设置需要进行调整。

获取帮助​

如果您在学习本教程时遇到问题，可以通过GitHub 讨论或RabbitMQ 社区 Discord联系我们。

先决条件​

与其他 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，了解如何根据模式侦听消息。

© . All rights reserved.