golang实现rabbitmq消费者模式断线重连机制

在连接第三方组件的时候实现断线重连机制是很有必要的事情，因为你不知道在啥时候他突然就抽风挂掉了。以rabbitmq为例，在突然面对大流量写入，或者连接数被打满（比如在资讯，直播等模块使用rabbitmq(惨痛的教训)）mq就挂掉了，如果你的消费者没有重连机制，你的消费者进程在mq挂掉后，也会自动挂掉。然后等运维修复了mq，但是你的消费者进程却无法再次连接消费了，这就有点恐怖了。要不就写个shell 脚本，每隔30S或者1分钟去检测进程活度，挂掉了就重启消费者。当然最简单的还是在消费者的代码里实现断线重连机制。代码如下

package rabbitmq

import (
   "encoding/json"
   "fmt"
   "github.com/streadway/amqp"
   "time"
)

//测试用例
type Obj struct {
   Item1 string `json:"item1"`
   Item2 string `json:"item2"`
   Item3 string `json:"item3"`
}

func StartAMQPConsume() {
   defer func() {
      if err := recover(); err != nil {
         time.Sleep(3 * time.Second)
         fmt.Println("休息3秒")
         StartAMQPConsume()
      }
   }()
   conn, err := amqp.Dial("amqp://guest:guest@127.0.0.1:5672/") //mq连接地址
   if err != nil {
      fmt.Println(err)
   }
   defer conn.Close()
   ch, err := conn.Channel()
   if err != nil {
      fmt.Println(err)
   }
   defer ch.Close()
   closeChan := make(chan *amqp.Error, 1)
   notifyClose := ch.NotifyClose(closeChan) //一旦消费者的channel有错误，产生一个amqp.Error，channel监听并捕捉到这个错误
   closeFlag := false
   msgs, err := ch.Consume(
      "xly.test.queue",
      "",
      true,
      false,
      false,
      false, nil)
   var obj Obj
   for {
      select {
      case e := <-notifyClose:
         fmt.Println("chan通道错误,e:%s", e.Error())
         close(closeChan)
         time.Sleep(5 * time.Second)
         StartAMQPConsume()
         closeFlag = true
      case msg := <-msgs:
         //fmt.Println()
         if err := json.Unmarshal(msg.Body, &obj); err != nil {

            fmt.Println(err.Error())
         }
         fmt.Println(obj.Item1)

      }
      if closeFlag {
         break
      }
   }
}

只需要在main 方法里调用这个方法即可，或者go协程调用，效果如下

当我关闭mq服务时会有如下效果