Go和RabbitMQ构建高效的消息队列系统

package mq

import (
	"fmt"
	"yunpan/config"

	"github.com/streadway/amqp"
)

var conn *amqp.Connection
var channel *amqp.Channel

// initChannel 初始化 RabbitMQ 的连接和通道
func initChannel() bool {
	// 检查是否已经存在有效的通道
	if channel != nil {
		return true
	}

	// 尝试建立到 RabbitMQ 的连接
	var err error
	conn, err = amqp.Dial(config.RabbitURL)
	if err != nil {
		fmt.Println("Failed to connect to RabbitMQ:", err.Error())
		return false
	}

	// 打开一个通道，用于消息的发布与接收
	channel, err = conn.Channel()
	if err != nil {
		fmt.Println("Failed to open a channel:", err.Error())
		return false
	}

	fmt.Println("Successfully initialized RabbitMQ channel.")
	return true
}

// Publish 向指定交换机发送消息
func Publish(exchange string, routingKey string, msg []byte) bool {
	// 确保通道已初始化
	if !initChannel() {
		fmt.Println("Failed to initialize channel")
		return false
	}

	// 发布消息到指定的交换机和路由键
	err := channel.Publish(
		exchange,     // 交换机名称
		routingKey,   // 路由键
		false,        // 是否强制转发
		false,        // 是否立即送达（此参数在新版本中没有实际效果）
		amqp.Publishing{
			ContentType: "text/plain",
			Body:        msg,
		},
	)
	if err != nil {
		fmt.Printf("Failed to publish message: %s\n", err.Error())
		return false
	}

	fmt.Println("Message published successfully.")
	return true
}

package config

var RabbitURL = "amqp://guest:guest@localhost:5672/"