基于Golang实现的Rabbitmq 连接池

原文出处：个人博客地址：http://www.damonyi.cc/?p=11

之前项目中需要写个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