继续浏览精彩内容
慕课网APP
程序员的梦工厂
打开
继续
感谢您的支持,我会继续努力的
赞赏金额会直接到老师账户
将二维码发送给自己后长按识别
微信支付
支付宝支付

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

慕姐8265434
关注TA
已关注
手记 1309
粉丝 222
获赞 1065

背景

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



作者:OhBonsai
链接:https://www.jianshu.com/p/da8c18bc3455


打开App,阅读手记
0人推荐
发表评论
随时随地看视频慕课网APP