跳到主要内容

RabbitMQ 教程 - 路由

路由

(使用 Go RabbitMQ 客户端)

信息

先决条件

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

在哪里获得帮助

如果您在学习本教程时遇到问题,可以通过 GitHub DiscussionsRabbitMQ 社区 Discord 联系我们。

之前的教程中,我们构建了一个简单的日志系统。我们能够将日志消息广播到多个接收器。

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

绑定

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

err = ch.QueueBind(
q.Name, // queue name
"", // routing key
"logs", // exchange
false,
nil)

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

绑定可以接受额外的 routing_key 参数。为了避免与 Channel.Publish 参数混淆,我们将其称为 binding key。这是我们如何使用键创建绑定的方法

err = ch.QueueBind(
q.Name, // queue name
"black", // routing key
"logs", // exchange
false,
nil)

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

直接交换机

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

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

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

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

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

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

多重绑定

使用相同的绑定键绑定多个队列是完全合法的。在我们的示例中,我们可以添加 XQ1 之间绑定键为 black 的绑定。在这种情况下,direct 交换机将像 fanout 一样工作,并将消息广播到所有匹配的队列。路由键为 black 的消息将传递到 Q1Q2

发出日志

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

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

err = ch.ExchangeDeclare(
"logs_direct", // name
"direct", // type
true, // durable
false, // auto-deleted
false, // internal
false, // no-wait
nil, // arguments
)

我们已准备好发送消息

err = ch.ExchangeDeclare(
"logs_direct", // name
"direct", // type
true, // durable
false, // auto-deleted
false, // internal
false, // no-wait
nil, // arguments
)
failOnError(err, "Failed to declare an exchange")

ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()

body := bodyFrom(os.Args)
err = ch.PublishWithContext(ctx,
"logs_direct", // exchange
severityFrom(os.Args), // routing key
false, // mandatory
false, // immediate
amqp.Publishing{
ContentType: "text/plain",
Body: []byte(body),
})

为了简化操作,我们将假设“严重性”可以是 infowarningerror 之一。

订阅

接收消息的工作方式与之前的教程中一样,但有一个例外 - 我们将为我们感兴趣的每个严重性创建一个新绑定。

q, err := ch.QueueDeclare(
"", // name
false, // durable
false, // delete when unused
true, // exclusive
false, // no-wait
nil, // arguments
)
failOnError(err, "Failed to declare a queue")

if len(os.Args) < 2 {
log.Printf("Usage: %s [info] [warning] [error]", os.Args[0])
os.Exit(0)
}
for _, s := range os.Args[1:] {
log.Printf("Binding queue %s to exchange %s with routing key %s",
q.Name, "logs_direct", s)
err = ch.QueueBind(
q.Name, // queue name
s, // routing key
"logs_direct", // exchange
false,
nil)
failOnError(err, "Failed to bind a queue")
}

将它们放在一起

emit_log_direct.go 脚本的代码

package main

import (
"context"
"log"
"os"
"strings"
"time"

amqp "github.com/rabbitmq/amqp091-go"
)

func failOnError(err error, msg string) {
if err != nil {
log.Panicf("%s: %s", msg, err)
}
}

func main() {
conn, err := amqp.Dial("amqp://guest:guest@localhost:5672/")
failOnError(err, "Failed to connect to RabbitMQ")
defer conn.Close()

ch, err := conn.Channel()
failOnError(err, "Failed to open a channel")
defer ch.Close()

err = ch.ExchangeDeclare(
"logs_direct", // name
"direct", // type
true, // durable
false, // auto-deleted
false, // internal
false, // no-wait
nil, // arguments
)
failOnError(err, "Failed to declare an exchange")

ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()

body := bodyFrom(os.Args)
err = ch.PublishWithContext(ctx,
"logs_direct", // exchange
severityFrom(os.Args), // routing key
false, // mandatory
false, // immediate
amqp.Publishing{
ContentType: "text/plain",
Body: []byte(body),
})
failOnError(err, "Failed to publish a message")

log.Printf(" [x] Sent %s", body)
}

func bodyFrom(args []string) string {
var s string
if (len(args) < 3) || os.Args[2] == "" {
s = "hello"
} else {
s = strings.Join(args[2:], " ")
}
return s
}

func severityFrom(args []string) string {
var s string
if (len(args) < 2) || os.Args[1] == "" {
s = "info"
} else {
s = os.Args[1]
}
return s
}

receive_logs_direct.go 的代码

package main

import (
"log"
"os"

amqp "github.com/rabbitmq/amqp091-go"
)

func failOnError(err error, msg string) {
if err != nil {
log.Panicf("%s: %s", msg, err)
}
}

func main() {
conn, err := amqp.Dial("amqp://guest:guest@localhost:5672/")
failOnError(err, "Failed to connect to RabbitMQ")
defer conn.Close()

ch, err := conn.Channel()
failOnError(err, "Failed to open a channel")
defer ch.Close()

err = ch.ExchangeDeclare(
"logs_direct", // name
"direct", // type
true, // durable
false, // auto-deleted
false, // internal
false, // no-wait
nil, // arguments
)
failOnError(err, "Failed to declare an exchange")

q, err := ch.QueueDeclare(
"", // name
false, // durable
false, // delete when unused
true, // exclusive
false, // no-wait
nil, // arguments
)
failOnError(err, "Failed to declare a queue")

if len(os.Args) < 2 {
log.Printf("Usage: %s [info] [warning] [error]", os.Args[0])
os.Exit(0)
}
for _, s := range os.Args[1:] {
log.Printf("Binding queue %s to exchange %s with routing key %s",
q.Name, "logs_direct", s)
err = ch.QueueBind(
q.Name, // queue name
s, // routing key
"logs_direct", // exchange
false,
nil)
failOnError(err, "Failed to bind a queue")
}

msgs, err := ch.Consume(
q.Name, // queue
"", // consumer
true, // auto ack
false, // exclusive
false, // no local
false, // no wait
nil, // args
)
failOnError(err, "Failed to register a consumer")

var forever chan struct{}

go func() {
for d := range msgs {
log.Printf(" [x] %s", d.Body)
}
}()

log.Printf(" [*] Waiting for logs. To exit press CTRL+C")
<-forever
}

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

go run receive_logs_direct.go warning error &> logs_from_rabbit.log

如果您想在屏幕上查看所有日志消息,请打开一个新终端并执行以下操作

go run receive_logs_direct.go info warning error
# => [*] Waiting for logs. To exit press CTRL+C

并且,例如,要发出 error 日志消息,只需键入

go run emit_log_direct.go error "Run. Run. Or it will explode."
# => [x] Sent 'error':'Run. Run. Or it will explode.'

(完整源代码,请参阅 (emit_log_direct.go 源代码)(receive_logs_direct.go 源代码)

继续阅读教程 5,了解如何根据模式侦听消息。

© . All rights reserved.