当前位置 博文首页 > golang中for循环遍历channel时需要注意的问题详解

    golang中for循环遍历channel时需要注意的问题详解

    作者:xialeistudio 时间:2021-06-25 18:26

    前言

    for循环是Go语言唯一的循环结构,最近在做一个基于RabbitMQ的应用,由于官方的qos没有golang的版本,所以出了一点问题。

    问题代码如下:

    _, ch, err := component.NewRabbitMQ()
    if err != nil {
     panic(err)
    }
    if err := ch.Qos(10, 0, true); err != nil {
     panic(err)
    }
    msgs, err := ch.Consume("push", "", false, false, false, false, nil)
    if err != nil {
     panic(err)
    }
    for m := range msgs {
     go func(d *amqp.Delivery) {
       defer func() { d.Ack(false) }
       // 处理消息
      }(&m)
     }

    发现消费到10条消息,进程就退出了,但是exit code为0,表示系统是正常退出,由于做了日志记录可以确定消费了10条,所以初步确定是qos相关问题。

    排查过程

    • 首先是把d的tag打印出来,发现全部是一样的,可以确定是重复的一条消息
    • 一开始想到可能是经典的go协程执行在for循环结束以后导致的,但是看我的代码不属于这种情况,有使用&m保证每一条消息都是不同循环传入的。所以判断可能是for循环的传递问题。
    • 确定方向之后开始写了一个测试项目用来验证我的想法是否正确。

    测试代码

    package main
    import "fmt"
    func main() {
     ch := make(chan int, 10)
     for i := 0; i < 10; i++ {
     ch <- i
     }
     close(ch)
     for v := range ch {
     fmt.Println(&v)
     }
    }

    执行输出

    0xc420086008
    0xc420086008
    0xc420086008
    0xc420086008
    0xc420086008
    0xc420086008
    0xc420086008
    0xc420086008
    0xc420086008
    0xc420086008

    到这里才焕然大悟,for循环中,如果循环变量不是指针,那么每次的变量是同一个,不过值变了。,所以上例中的RabbitMQ go协程消费消息那里,需要直接传递值而不是指针,经过测试之后发现,问题确实解决了。

    题外话

    测试代码那里,如果不close掉channel是会发生死锁的,原因是 当for循环读完channel的10个值之后会继续尝试读取下一个,而由于channel为空又没关闭,会一直阻塞形成死锁

    TOOD

    研究RabbitMQ Consumer部分的源码来看看消费channel被关闭的问题。

    总结

    js
    下一篇:没有了