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:
 }

有疑问加站长微信联系(非本文作者)

• 请尽量让自己的回复能够对别人有帮助
• 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
• 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
• 图片支持拖拽、截图粘贴等方式上传