golang rabbitMQ 生产者复用channel以及生产者组分发策略

原创

已于 2023-02-13 11:56:59 修改 · 3.5k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#golang #后端 #rabbitmq

于 2023-02-10 18:44:14 首次发布

文章详细描述了在Go语言中使用rabbitMQ时如何实现连接和channel的复用，以提高效率和可靠性。通过创建Connection和Channel结构体，初始化时动态配置生产者和消费者的连接数及channel数量。消费者和生产者分别通过工作模式运行，生产者发送消息时随机选择connection和channel，确保负载均衡。代码示例展示了具体实现过程。

引用的是rabbitMQ官方示例的库：github.com/rabbitmq/amqp091-go

在网络编程中我们知道tcp连接的创建、交互、销毁等相关操作的"代价"都是很高的，所以就要去实现如何复用这些连接，并要做到高效并可靠。

预期效果：

项目初始化构建时可以自定义选择生产者开启多个connection，每个connection可以启动多少个channel【都是全局复用的】，因为rabbitMQ所有的命令都是基本都是通过channel去操作完成的，所以这个channel很重要，也是我们想要复用的重点。

初始化创建完connection和channel后，当生产者需要发送一条消息的时候，我们可以通过一些策略去选择它发送到哪个connection和channel，我这里采用的就是随机选择，也可以采用哈希取模、轮询权重算法等，这个可以根据自身业务来做。

我简单画了一个效果图：

定义RabbitMQ的Config、Connection、Channel结构体

type Config struct {
    Host     string
    Port     int
    User     string
    Password string
}

type Channel struct {
    m  *sync.Mutex
    ch *amqp.Channel
}

type Connection struct {
    ctx  context.Context
    n    int
    conn *amqp.Connection
    ch   []*Channel
}

实例化RabbitMQ结构体

func (mq *Connection) New(config Config) (rabbitmq *Connection) {

    configString := fmt.Sprintf("amqp://%s:%s@%s:%d/", config.User, config.Password, config.Host, config.Port)

    conn, err := amqp.Dial(configString)
    if err != nil {
        log.Panicf("amqp connect error: %v \n", err)
    }

    rabbitmq = &Connection{
        ctx:  context.Background(),
        conn: conn,
    }

    return
}

一、创建消费者

// ConsumeWithWork rabbitmq消费消息[work模式 channelNums可以设置当前连接开启多少个channel]
func (mq *Connection) ConsumeWithWork(queueName string, channelNums int) {
    for i := 0; i < channelNums; i++ {
        go func(i int) {

            ch, err := mq.conn.Channel()
            if err !=