RabbitMQ(4)：路由

引言

在rabbitmq的发布订阅一节中，我们构建了一个简单的日志记录系统。我们能够向许多接收者广播日志消息。但是每个接收者接收到的消息都是相同的。在有一些场景中，我们希望只接收自己想要的，比如日志系统中我们选择只将error级别的日志保存在文件中。

本节内容中，我们将向它添加一个特性-我们将使它能够只订阅消息的一个子集。例如，我们将只能将关键错误消息定向到日志文件（以节省磁盘空间），同时仍然能够在控制台上打印所有日志消息。

绑定

在前面的示例中，我们已经在创建绑定。如下：

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",   // binding key
  "logs",    // exchange
  false,
  nil)

绑定密钥binding key的含义取决于交换器的类型。我们以前使用的fanout交换器只是忽略了这个值。

直连交换器

我们之前的日志系统是将所有消息广播给所有消费者。我们希望扩展这一点，允许根据消息的种类过滤消息。例如，我们可能希望将日志消息写入磁盘的脚本只接收严重错误，而不会在warning或info日志消息上浪费磁盘空间。

我们之前使用的是fanout交换器，这并没有给我们很大的灵活性——它只能无脑地对所有绑定了的队列进行广播。

我们将使用direct交换器。direct交换器背后的路由算法很简单——消息进入其binding key（队列和交换器绑定时的参数）与消息的routing key（消息发布时的参数）完全匹配的队列。

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

在此设置中，我们可以看到绑定了两个队列的direct交换器X。第一个队列绑定键为orange，第二个队列绑定为两个，一个绑定键为black，另一个为green。

在这种设置中，使用orange路由键发布到交换器的消息将被路由到队列Q1。路由键为black或green的消息将转到Q2。所有其他消息将被丢弃。

多重绑定

用相同的绑定键绑定多个队列是完全合法的。在我们的示例中，我们可以使用绑定键black在X和Q1之间添加绑定。在这种情况下，direct交换器的行为将类似fanout，并将消息广播到所有匹配的队列。带有black路由键的消息将同时传递给Q1和Q2。

发布日志

我们将在日志系统中使用这个模型。我们将发送消息到direct交换器，而不是fanout交换器。我们将提供日志级别作为路由键。这样，接收脚本将能够选择其想要接收的日志级别。让我们首先关注发送日志。

// 创建一个direct交换器
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")

body := bodyFrom(os.Args)
err = ch.Publish(
  "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,             // binding key
    "logs_direct", // exchange
    false,
    nil)
  failOnError(err, "Failed to bind a queue")
}

完整示例

生产者P通过direct交换器发布日志消息，C1只接收error等级日志，C2全部都接收

生产者

emit_log_direct.go

package main

import (
	"log"
	"os"
	"strings"

	"github.com/streadway/amqp"
)

func main() {
	conn, err := amqp.Dial("amqp://root:123456@127.0.0.1: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()

	// 创建direct交换器
	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")

	body := bodyFrom(os.Args)
	err = ch.Publish(
		"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
}

func failOnError(err error, msg string) {
	if err != nil {
		log.Fatalf("%s: %s", msg, err)
	}
}

消费者

receive_logs_direct.go

package main

import (
	"log"
	"os"

	"github.com/streadway/amqp"
)

func main() {
	conn, err := amqp.Dial("amqp://root:123456@127.0.0.1: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")

	go func() {
		for d := range msgs {
			log.Printf(" [x] %s", d.Body)
		}
	}()

	log.Printf(" [*] Waiting for logs. To exit press CTRL+C")
	select {}
}

func failOnError(err error, msg string) {
	if err != nil {
		log.Fatalf("%s: %s", msg, err)
	}
}

如果你只想将“warning”和“err”（而不是“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.'