RabbitMQ 教程 - 路由
路由
(使用 Go RabbitMQ 客户端)
先决条件
本教程假设 RabbitMQ 已安装并在 localhost 上的标准端口 (5672) 上运行。如果您使用不同的主机、端口或凭据,则需要调整连接设置。
在哪里获得帮助
如果您在学习本教程时遇到问题,可以通过 GitHub Discussions 或 RabbitMQ 社区 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。路由键为 black 或 green 的消息将发送到 Q2。所有其他消息将被丢弃。
多重绑定
使用相同的绑定键绑定多个队列是完全合法的。在我们的示例中,我们可以添加 X 和 Q1 之间绑定键为 black 的绑定。在这种情况下,direct 交换机将像 fanout 一样工作,并将消息广播到所有匹配的队列。路由键为 black 的消息将传递到 Q1 和 Q2。
发出日志
我们将把此模型用于我们的日志系统。我们将消息发送到 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),
})
为了简化操作,我们将假设“严重性”可以是 info、warning 或 error 之一。
订阅
接收消息的工作方式与之前的教程中一样,但有一个例外 - 我们将为我们感兴趣的每个严重性创建一个新绑定。
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,了解如何根据模式侦听消息。