在上一教程中,我们构建了一个简单的日志记录系统。我们能够向许多消息消费者推送广播日志消息。
在本教程中,我们将为其添加一个功能 - 我们将使仅订阅消息的子集成为可能。为例如,我们将只能将关键错误消息定向到日志文件(以节省磁盘空间),同时仍然能够打印所有 控制台上的日志消息。
在前面的示例中,我们已经创建好了绑定,你可能还记得如下的代码:
err = ch.QueueBind(
q.Name, // queue name
"", // routing key
"logs", // exchange
false,
nil)
绑定操作是指交换机和队列之间的绑定关系。绑定可以采用额外的 routing_key
参数进行设置。为了避免与 Channel.Publish
参数混淆,我们将它称为绑定键。这就是我们如何使用键创建绑定:
err = ch.QueueBind(
q.Name, // queue name
"black", // routing key
"logs", // exchange
false,
nil)
绑定键的含义取决于交换类型。我们之前使用的 fanout
交换类型,只是忽略了它的价值。
上一教程中的日志记录系统广播所有消息给所有消费者。我们希望扩展它以允许过滤消息 基于其严重程度。例如,我们可能需要脚本将日志消息写入磁盘以仅接收严重错误,以及 不要在警告或信息日志消息上浪费磁盘空间。
我们使用 fanout
交换机类型,这并没有给我们太多的灵活性。它只是盲目的将所有的消息广播到所有的消费者。
我们将改用 direct
交换机类型。背后的路由算法 direct
交换很简单 - 消息转到绑定键与消息的路由键完全匹配的队列。
为了说明这一点,请考虑以下设置:
在此设置中,我们可以看到绑定了两个队列的 direct
交换 X
到它。第一个队列绑定绑定键 orange
,第二个队列绑定绑定键有两个绑定,一个带有绑定键 block
,另一个带有绑定键是 green
。
在此类设置中,发布到交换器的带有 orange
路由密钥的消息将被路由到队列 Q1
。路由密钥为 block
或 green
的消息将转到第 Q2
。至于其他消息将被丢弃。
使用同一绑定绑定多个队列是完全合法的 .key
。在我们的示例中,我们可以在 X
和 Q1
之间添加一个绑定键 block
。在这种情况下,direct
交换将表现良好, 像 fanout
一样,会将消息广播到所有匹配队列。路由密钥为 block
的消息将同时传递到 Q1
和 Q2
。
我们将此模型用于我们的日志记录系统。我们将不是 fanout
类型的交换机。我们将提供日志严重性为路由密钥。这样接收脚本将能够选择它想要接收的严重性。让我们专注于发出日志。
与往常一样,我们需要首先创建一个交换机:
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.'