基于Golang实现的Rabbitmq 连接池

来源：互联网发布：魔兽世界数据库wlk 编辑：程序博客网时间：2024/06/06 08:45

原文出处：个人博客地址：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.RWMutexConnection *amqp.ConnectionMqUri      string}type ChannelContext struct {Exchange     stringExchangeType stringRoutingKey   stringReliable     boolDurable      boolChannelId    stringChannel      *amqp.Channel}type BaseMq struct {MqConnection *MqConnection//channel cacheChannelContexts map[string]*ChannelContext}

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

func TestMyMqConnection1() {var mq1 *mq.BaseMqmq1 = 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

0 0