golang rabbitMQ 生产者复用channel以及生产者组分发策略

文章详细描述了在Go语言中使用rabbitMQ时如何实现连接和channel的复用,以提高效率和可靠性。通过创建Connection和Channel结构体,初始化时动态配置生产者和消费者的连接数及channel数量。消费者和生产者分别通过工作模式运行,生产者发送消息时随机选择connection和channel,确保负载均衡。代码示例展示了具体实现过程。

引用的是rabbitMQ官方示例的库github.com/rabbitmq/amqp091-go

在网络编程中我们知道tcp连接的创建、交互、销毁等相关操作的"代价"都是很高的,所以就要去实现如何复用这些连接,并要做到高效并可靠。

预期效果:

项目初始化构建时可以自定义选择生产者开启多个connection,每个connection可以启动多少个channel【都是全局复用的】,因为rabbitMQ所有的命令都是基本都是通过channel去操作完成的,所以这个channel很重要,也是我们想要复用的重点。

初始化创建完connection和channel后,当生产者需要发送一条消息的时候,我们可以通过一些策略去选择它发送到哪个connection和channel,我这里采用的就是随机选择,也可以采用哈希取模、轮询权重算法等,这个可以根据自身业务来做。

我简单画了一个效果图:

定义RabbitMQ的Config、Connection、Channel结构体

type Config struct {
    Host     string
    Port     int
    User     string
    Password string
}

type Channel struct {
    m  *sync.Mutex
    ch *amqp.Channel
}

type Connection struct {
    ctx  context.Context
    n    int
    conn *amqp.Connection
    ch   []*Channel
}

实例化RabbitMQ结构体

func (mq *Connection) New(config Config) (rabbitmq *Connection) {

    configString := fmt.Sprintf("amqp://%s:%s@%s:%d/", config.User, config.Password, config.Host, config.Port)

    conn, err := amqp.Dial(configString)
    if err != nil {
        log.Panicf("amqp connect error: %v \n", err)
    }

    rabbitmq = &Connection{
        ctx:  context.Background(),
        conn: conn,
    }

    return
}

一、创建消费者

// ConsumeWithWork rabbitmq消费消息[work模式 channelNums可以设置当前连接开启多少个channel]
func (mq *Connection) ConsumeWithWork(queueName string, channelNums int) {
    for i := 0; i < channelNums; i++ {
        go func(i int) {

            ch, err := mq.conn.Channel()
            if err !=
评论 1
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值