跳至主内容

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)

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

直连交换机

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

我们使用的是fanout交换,这并没有给我们太大的灵活性——它只能进行无差别的广播。

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

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

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

在这种设置下,发布到具有路由 key orange 的交换机的消息将被路由到队列 Q1。具有路由 key blackgreen 的消息将发送到 Q2。所有其他消息将被丢弃。

多个绑定

使用相同的绑定 key 绑定多个队列是完全合法的。在我们的示例中,我们可以添加一个 XQ1 之间的绑定,绑定 key 为 black。在这种情况下,direct 交换机将表现得像 fanout,并将消息广播给所有匹配的队列。路由 key 为 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
}

如果您只想将“警告”和“错误”(而不是“信息”)日志消息保存到文件,只需打开一个终端并输入:

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,了解如何根据模式监听消息。

© . This site is unofficial and not affiliated with VMware.