跳至主要内容

RabbitMQ 教程 - 发布/订阅

发布/订阅

(使用 Go RabbitMQ 客户端)

信息

先决条件

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

获取帮助

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

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

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

在我们的日志系统中,接收器程序的每个运行副本都将获得消息。这样,我们就可以运行一个接收器并将日志定向到磁盘;同时,我们也可以运行另一个接收器并在屏幕上查看日志。

本质上,发布的日志消息将广播到所有接收器。

交换机

在教程的前几部分,我们向队列发送和接收消息。现在是时候介绍 Rabbit 中完整的消息模型了。

让我们快速回顾一下我们在之前的教程中介绍的内容

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

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

© 2024 RabbitMQ. All rights reserved.