温馨提示×

如何在Linux中用Golang实现消息队列

小樊
46
2026-05-23 20:51:10
栏目: 编程语言

在Linux中使用Golang实现消息队列,你可以选择多种消息队列系统,如RabbitMQ、Kafka、ZeroMQ等。这里以RabbitMQ为例,介绍如何使用Golang的streadway/amqp库来实现消息队列。

步骤1:安装RabbitMQ

首先,你需要在你的Linux系统上安装RabbitMQ。以下是在Ubuntu上的安装步骤:

sudo apt-get update
sudo apt-get install rabbitmq-server

启动RabbitMQ服务:

sudo systemctl start rabbitmq-server

确保RabbitMQ服务正在运行:

sudo systemctl status rabbitmq-server

步骤2:安装Golang的AMQP库

在你的Golang项目中,你需要安装streadway/amqp库。可以使用以下命令安装:

go get github.com/streadway/amqp

步骤3:编写生产者代码

创建一个名为producer.go的文件,并添加以下代码来创建一个生产者,它将消息发送到RabbitMQ队列:

package main

import (
	"fmt"
	"log"

	"github.com/streadway/amqp"
)

func failOnError(err error, msg string) {
	if err != nil {
		log.Fatalf("%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()

	q, err := ch.QueueDeclare(
		"hello", // name
		true,    // durable
		false,   // delete when unused
		false,   // exclusive
		false,   // no-wait
		nil,     // arguments
	)
	failOnError(err, "Failed to declare a queue")

	body := "Hello World!"
	err = ch.Publish(
		"",     // exchange
		q.Name, // routing key
		false,  // mandatory
		false,  // immediate
		amqp.Publishing{
			ContentType: "text/plain",
			Body:        []byte(body),
		})
	failOnError(err, "Failed to publish a message")
	fmt.Println(" [x] Sent", body)
}

步骤4:编写消费者代码

创建一个名为consumer.go的文件,并添加以下代码来创建一个消费者,它将从RabbitMQ队列接收消息:

package main

import (
	"fmt"
	"log"

	"github.com/streadway/amqp"
)

func failOnError(err error, msg string) {
	if err != nil {
		log.Fatalf("%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()

	q, err := ch.QueueDeclare(
		"hello", // name
		true,    // durable
		false,   // delete when unused
		false,   // exclusive
		false,   // no-wait
		nil,     // arguments
	)
	failOnError(err, "Failed to declare 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")

	forever := make(chan bool)

	go func() {
		for d := range msgs {
			fmt.Printf("Received a message: %s\n", d.Body)
		}
	}()

	fmt.Println(" [*] Waiting for messages. To exit press CTRL+C")
	<-forever
}

步骤5:运行生产者和消费者

首先,运行消费者程序:

go run consumer.go

然后,在另一个终端中运行生产者程序:

go run producer.go

你应该会看到消费者接收到生产者发送的消息。

以上就是在Linux中使用Golang和RabbitMQ实现消息队列的基本步骤。你可以根据需要调整代码,例如添加更多的队列、交换机、绑定等。

0