为了账号安全,请及时绑定邮箱和手机立即绑定

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

标签:
Go

背景

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


点击查看更多内容
TA 点赞

若觉得本文不错,就分享一下吧!

评论

作者其他优质文章

正在加载中
  • 推荐
  • 评论
  • 收藏
  • 共同学习,写下你的评论
感谢您的支持,我会继续努力的~
扫码打赏,你说多少就多少
赞赏金额会直接到老师账户
支付方式
打开微信扫一扫,即可进行扫码打赏哦
今天注册有机会得

100积分直接送

付费专栏免费学

大额优惠券免费领

立即参与 放弃机会
意见反馈 帮助中心 APP下载
官方微信

举报

0/150
提交
取消