基于Golang实现的Rabbitmq 连接池

时间:2025-03-25 08:34:30

 原文出处:个人博客地址:/?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       
	Connection *
	MqUri      string
}

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

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

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

func TestMyMqConnection1() {
	var mq1 *
	mq1 = ("manager")
	channleContxt := {Exchange: "docker-exchange1",
		ExchangeType: "direct",
		RoutingKey:   "docker1",
		Reliable:     true,
		Durable:      false}
	for {
		("sending message111")
		(&channleContxt, "helllosaga")
		(10 * )
	}
}

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