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
}
  1. 注册监听
producer.channel.NotifyClose(producer.notifyClose)
producer.channel.NotifyPublish(producer.notifyConfirm)
  1. 发了就不管
    直接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(),
        },
    )
  1. 三次重传的发消息
    这里主要通过time.NewTicker来实现超时重发
        ticker := time.NewTicker(resendDelay)
        select {
        case confirm := <-producer.notifyConfirm:
            if confirm.Ack {
                producer.logger.Println("Push confirmed!")
                return nil
            }
        case <- ticker.C:
        }

推荐阅读更多精彩内容