背景
Golang里面采用AMQP来连接rabbitmq, 使用之后发现这个库比较底层，只提供协议的封装。这个库用到生产环境不合适，包装了一层以提供更加稳定的功能，代码地址
目标
- 断线重连： 这个需求不过分，生产环境各种网络问题，最起码的重连要支持，支持重连次数
- 消息重发： 这个要求不过分，Rabbitmq有消息确认机制
关键实现
PS：Golang的并发真的设计的很好，习惯之后用起来比多线程/锁的模式舒服一些。
- 定义三个通道来进行并发
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
}
- 注册监听
producer.channel.NotifyClose(producer.notifyClose)
producer.channel.NotifyPublish(producer.notifyConfirm)
- 发了就不管
直接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(),
},
)
- 三次重传的发消息
这里主要通过time.NewTicker来实现超时重发
ticker := time.NewTicker(resendDelay)
select {
case confirm := <-producer.notifyConfirm:
if confirm.Ack {
producer.logger.Println("Push confirmed!")
return nil
}
case <- ticker.C:
}
