从 Sarama 的错误通道中读取的正确方法是什么？

Question

当我生成消息时，我正在使用用 Go 编写的 Sarama 库从错误通道中读取。整个代码如下所示，包含在一个函数中：

producer.AsyncProducer.Input() <- &sarama.ProducerMessage{Topic: topic, Key: nil, Value: sarama.ByteEncoder(message)}
go func() {
    for err := range saramaProducer.Errors() {
        if producer.callbacks.OnError != nil {
            producer.callbacks.OnError(err)
        }
    }
}()

根据我对 goroutine 的理解，我的 goroutine 会不断迭代 Errors() 频道，直到它收到一个。有没有办法让它在我的函数执行完成后停止侦听错误？

Answer 1

您可以使用另一个频道和select 使循环返回。

var quit chan struct{}
go func() {
    for {
        select {
        case err:=<-saramaProducer.Errors():
            //handle errors
        case <-quit:
            return
        }
    }
}
defer func() { quit<-struct{}{} }()

原始的for ... range 循环在得到一个之前不会迭代通道。相反，它会一直阻塞，直到出现错误，处理它，然后再次等待新的错误，直到通道关闭或main 返回。

上面的代码有个小问题，当quit和error channel都准备好时，select会随机选择一个，可能会导致单个error丢失。如果这值得处理，只需将另一个 switch 与 default 放在一起以获取该错误，然后再添加 return。