基于Golang实现的Rabbitmq 连接池

       这个网站是新申请的网站,备案号还没有申请下来,想好好的做好这个博客。越来越发现做了很多事情,如果没有记录下来,过一段时间就忘记了,忘了在哪看到的一句话:不做笔记的学习工作,都是耍流氓,哈哈。。。

      之前项目中需要写个Python 版本的Agent,里面用到了Rabbitmq,中间遇到了好多坑啊,最主要的原因就是Python的Rabbitmq 库没有java的完善,像spring-Rabbitmq,什么都给你处理了。但是在Python中各种异常都需要自己处理。

     最近又开发了一个Golang的项目,里面也用到了rabbitmq ,google了一下,golang的Rabbitmq库,也不完善

        1.多个channel共享一个connection

        2.连接断开后的重建连接

  项目目前也主要是为了解决这个问题,因此整了个Rabbitmq的Connection Pool (实际上是Channel Pool,存储了一个Connection的多个Channel),实现思路可以参考设计的Struct

type MqConnection struct {
	Lock       sync.RWMutex
	Connection *amqp.Connection
	MqUri      string
}

type ChannelContext struct {
	Exchange     string
	ExchangeType string
	RoutingKey   string
	Reliable     bool
	Durable      bool
	ChannelId    string
	Channel      *amqp.Channel
}
type BaseMq struct {
	MqConnection *MqConnection

	//channel cache
	ChannelContexts map[string]*ChannelContext
}

查看我定义的Struct,可以看到就是定义了一个struct 保存了Connection信息和全部的Channel信息。当使用BasMq的Publish方法时,首先判断是否已经存在对应的Channel了,如果有,就用已经存在的Channel。发送消息代码如下

func TestMyMqConnection1() {
	var mq1 *mq.BaseMq
	mq1 = mq.GetConnection("manager")
	channleContxt := mq.ChannelContext{Exchange: "docker-exchange1",
		ExchangeType: "direct",
		RoutingKey:   "docker1",
		Reliable:     true,
		Durable:      false}
	for {
		fmt.Println("sending message111")
		mq1.Publish(&channleContxt, "helllosaga")
		time.Sleep(10 * time.Second)
	}
}

参数只传入Channel信息,RabbitmqServer 可以根据业务需求调整,具体可参考源码 rabbitmq-golangclient







喜欢 1