背景
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: }
作者:OhBonsai
链接:https://www.jianshu.com/p/da8c18bc3455