跳至主内容

RabbitMQ 教程 - 发布/订阅

发布/订阅

(使用 Go RabbitMQ 客户端)

信息

先决条件

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

哪里寻求帮助

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

上一篇教程中,我们创建了一个工作队列。工作队列的假设是每个任务只交付给一个工作者。在本篇中,我们将做一些完全不同的事情——将一条消息传递给多个消费者。这种模式被称为“发布/订阅”。

为了说明这种模式,我们将构建一个简单的日志系统。它将包含两个程序——第一个程序将发出日志消息,第二个程序将接收并打印它们。

在我们的日志系统中,运行的每个接收者程序实例都将收到消息。这样,我们就可以运行一个接收者并将日志写入磁盘;同时,我们还可以运行另一个接收者并在屏幕上查看日志。

本质上,发布的日志消息将被广播给所有接收者。

交换器

在之前的教程中,我们向队列发送消息并从队列接收消息。现在是时候在 RabbitMQ 中引入完整的消息模型了。

让我们快速回顾一下之前的内容

  • 生产者 是发送消息的用户应用程序。
  • 队列 是存储消息的缓冲区。
  • 消费者 是接收消息的用户应用程序。

RabbitMQ 消息模型的核心思想是生产者从不直接将任何消息发送到队列。实际上,生产者往往甚至不知道消息是否会被传递到任何队列。

相反,生产者只能将消息发送到交换器。交换器是一个非常简单的东西。一方面,它接收来自生产者的消息;另一方面,它将消息推送到队列。交换器必须确切地知道如何处理它收到的消息。应该将其附加到特定队列吗?应该将其附加到多个队列吗?还是应该将其丢弃?这些规则由交换器类型定义。

有几种可用的交换器类型:directtopicheadersfanout。我们将重点关注最后一种——fanout。让我们创建一个此类型的交换器,并将其命名为 logs

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

fanout 交换器非常简单。顾名思义,它只是将收到的所有消息广播到它知道的所有队列。这正是我们日志记录器所需要的。

列出交换器

要列出服务器上的交换器,可以使用一直很有用的 rabbitmqctl

sudo rabbitmqctl list_exchanges

在此列表中,将有一些 amq.* 交换器以及默认(未命名)交换器。这些是默认创建的,但目前不太可能需要使用它们。

默认交换器

在教程的先前部分,我们对交换机一无所知,但仍然能够将消息发送到队列。这是可能的,因为我们使用的是一个默认交换机,它由空字符串 ("") 标识。

回想一下我们之前是如何发布消息的

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

这里我们使用默认的或匿名的交换机:消息会被路由到名为routing_key参数指定的队列(如果存在)。

现在,我们可以改为发布到我们的命名交换器

err = ch.ExchangeDeclare(
"logs", // name
"fanout", // 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", // exchange
"", // routing key
false, // mandatory
false, // immediate
amqp.Publishing{
ContentType: "text/plain",
Body: []byte(body),
})

临时队列

您可能还记得,之前我们使用了具有特定名称的队列(还记得 hellotask_queue 吗?)。能够命名队列对我们至关重要——我们需要将工作者指向同一个队列。给队列命名对于我们希望在生产者和消费者之间共享队列非常重要。

但我们的日志记录器不是这种情况。我们想了解所有日志消息,而不仅仅是其中的一部分。我们只对当前正在流动的消息感兴趣,而不是旧消息。要解决这个问题,我们需要两件事。

首先,每次连接到 Rabbit 时,我们需要一个全新的、空的队列。为此,我们可以创建一个带有随机名称的队列,或者——更好的是——让服务器为我们选择一个随机队列名称。

其次,一旦我们断开消费者连接,队列应被自动删除。

amqp客户端中,当我们把队列名设置为空字符串时,我们会创建一个非持久化的、带有生成名称的队列。

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

当方法返回时,队列实例将包含 RabbitMQ 生成的随机队列名称。例如,它可能看起来像 amq.gen-JzTY20BRgKO-HjmUJj0wLg

当声明它的连接关闭时,队列将因为它被声明为独占而删除。

你可以在队列指南中了解更多关于 exclusive 标志和其他队列属性的信息。

绑定

我们已经创建了一个 fanout 交换器和一个队列。现在我们需要告诉交换器将消息发送到我们的队列。交换器和队列之间的这种关系称为绑定

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

从现在开始,logs 交换器将把消息附加到我们的队列。

列出绑定

您可以使用,您猜对了,列出现有绑定。

rabbitmqctl list_bindings

总而言之

生产者程序(发出日志消息的程序)看起来与上一篇教程差别不大。最重要的改变是我们现在希望将消息发布到我们的logs交换机,而不是匿名的交换机。发送消息时我们需要提供一个routingKey,但对于fanout交换机,其值会被忽略。以下是emit_log.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", // name
"fanout", // 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", // exchange
"", // 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) < 2) || os.Args[1] == "" {
s = "hello"
} else {
s = strings.Join(args[1:], " ")
}
return s
}

(emit_log.go 源码)

如您所见,建立连接后,我们声明了交换器。此步骤是必要的,因为禁止发布到不存在的交换器。

如果尚未将任何队列绑定到交换器,消息将丢失,但这对我们来说没关系;如果没有消费者在监听,我们可以安全地丢弃该消息。

receive_logs.go的代码

package main

import (
"log"

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", // name
"fanout", // 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")

err = ch.QueueBind(
q.Name, // queue name
"", // routing key
"logs", // 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
}

(receive_logs.go 源码)

如果您想将日志保存到文件,只需打开一个控制台并输入:

go run receive_logs.go &> logs_from_rabbit.log

如果您希望在屏幕上看到日志,请启动一个新的终端并运行:

go run receive_logs.go

当然,要发出日志,请键入:

go run emit_log.go

使用rabbitmqctl list_bindings可以验证代码是否如我们所愿地创建了绑定和队列。运行两个receive_logs.go程序,你应该会看到类似这样的输出:

sudo rabbitmqctl list_bindings
# => Listing bindings ...
# => logs exchange amq.gen-JzTY20BRgKO-HjmUJj0wLg queue []
# => logs exchange amq.gen-vso0PVvyiRIL2WoV3i48Yg queue []
# => ...done.

结果的解释很简单:来自 logs 交换器的数据进入了两个带有服务器分配名称的队列。这正是我们的意图。

要了解如何监听消息的子集,让我们继续学习教程 4

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