跳到主要内容

RabbitMQ 教程 - 路由

路由​

（使用 Pika Python 客户端）​

信息

前提条件​

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

在哪里获得帮助​

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

前提条件​

与其他 Python 教程一样，我们将使用 Pika RabbitMQ 客户端 1.0.0 版本。

本教程重点​

在上一教程中，我们构建了一个简单的日志系统。我们能够将日志消息广播给多个接收者。

在本教程中，我们将为其添加一个功能 - 我们将使其可以仅订阅消息的子集。例如，我们将能够仅将关键错误消息定向到日志文件（以节省磁盘空间），同时仍然能够在控制台上打印所有日志消息。

绑定​

在以前的示例中，我们已经创建了绑定。您可能还记得类似这样的代码

channel.queue_bind(exchange=exchange_name,
                   queue=queue_name)

绑定是交换机和队列之间的关系。这可以简单地理解为：队列对来自此交换机的消息感兴趣。

绑定可以采用额外的 routing_key 参数。为了避免与 basic_publish 参数混淆，我们将其称为 binding key。这是我们如何使用密钥创建绑定的方法

channel.queue_bind(exchange=exchange_name,
                   queue=queue_name,
                   routing_key='black')

绑定键的含义取决于交换机类型。我们之前使用的 fanout 交换机只是忽略了它的值。

直接交换机​

我们从上一个教程中的日志系统将所有消息广播给所有消费者。我们希望扩展它以允许根据消息的严重性过滤消息。例如，我们可能希望将日志消息写入磁盘的脚本仅接收关键错误，而不要在警告或信息日志消息上浪费磁盘空间。

我们正在使用 fanout 交换机，它没有给我们太多的灵活性 - 它只能进行盲目的广播。

我们将改用 direct 交换机。 direct 交换机背后的路由算法很简单 - 消息会发送到其 binding key 与消息的 routing key 完全匹配的队列。

为了说明这一点，请考虑以下设置

在此设置中，我们可以看到 direct 交换机 X 以及绑定到它的两个队列。第一个队列与绑定键 orange 绑定，第二个队列有两个绑定，一个绑定键为 black，另一个绑定键为 green。

在这种设置中，发布到交换机且路由键为 orange 的消息将路由到队列 Q1。路由键为 black 或 green 的消息将发送到 Q2。所有其他消息将被丢弃。

多个绑定​

使用相同的绑定键绑定多个队列是完全合法的。在我们的示例中，我们可以在 X 和 Q1 之间添加一个绑定，绑定键为 black。在这种情况下， direct 交换机的行为将类似于 fanout ，并将消息广播到所有匹配的队列。路由键为 black 的消息将传递到 Q1 和 Q2。

发送日志​

我们将对我们的日志系统使用此模型。我们将消息发送到 direct 交换机，而不是 fanout。我们将提供日志严重性作为 routing key。这样，接收脚本将能够选择它想要接收的严重性。让我们首先关注发送日志。

像往常一样，我们需要首先创建一个交换机

channel.exchange_declare(exchange='direct_logs',
                         exchange_type='direct')

我们已准备好发送消息

channel.basic_publish(exchange='direct_logs',
                      routing_key=severity,
                      body=message)

为了简化操作，我们将假设“severity”可以是 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 (source) （源代码）

#!/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 (source) （源代码）

#!/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()

如果您只想将“warning”和“error”（而不是“info”）日志消息保存到文件中，只需打开控制台并键入

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.