Golang RabbitMQ 自动重连/重发生产者

背景

Golang里面采用 AMQP 来连接 rabbitmq , 使用之后发现这个库比较底层，只提供协议的封装。这个库用到生产环境不合适，包装了一层以提供更加稳定的功能， 代码地址

目标

• 断线重连： 这个需求不过分，生产环境各种网络问题，最起码的重连要支持，支持重连次数
• 消息重发： 这个要求不过分，Rabbitmq有消息确认机制

关键实现

PS：Golang的并发真的设计的很好，习惯之后用起来比多线程/锁的模式舒服一些。

1. 定义三个通道来进行并发

type Producer struct {
    name          string
    logger        *log.Logger
    connection    *amqp.Connection
    channel       *amqp.Channel
    done          chan bool   // 如果主动close，会接受数据
    notifyClose   chan *amqp.Error   // 如果异常关闭，会接受数据
    notifyConfirm chan amqp.Confirmation // 消息发送成功确认，会接受到数据
    isConnected   bool
}

2. 注册监听

producer.channel.NotifyClose(producer.notifyClose)
producer.channel.NotifyPublish(producer.notifyConfirm)

3. 发了就不管

   直接push消息,回传一个error

return producer.channel.Publish(
        "",         // Exchange
        producer.name, // Routing key
        false,      // Mandatory
        false,      // Immediate
        amqp.Publishing{
            DeliveryMode: 2,
            ContentType:  "application/json",
            Body:        data,
            Timestamp:  time.Now(),
        },
    )

4. 三次重传的发消息

   这里主要通过 time.NewTicker 来实现超时重发

ticker := time.NewTicker(resendDelay)
        select {
        case confirm := <-producer.notifyConfirm:
            if confirm.Ack {
                producer.logger.Println("Push confirmed!")
                return nil
            }
        case <- ticker.C:
        }