go - 如何同时收听两个 Golang channel ？

Question

我基本上是通过爬虫收集数据，然后想以两种方式处理这些数据:拒绝或接受。为了不减慢抓取速度，我将数据传递到拒绝 channel 或接受 channel 。我需要这些 channel 同时收听各自 channel 中的数据。到目前为止，这是我的尝试:

func main() {
  okChannel := make(chan Article)
  rejectChannel := make(chan Article)

  scrape := new(scrapers)
  go Init(okChannel, rejectChannel)

  go func(okChannel chan Article) {
    for article := range okChannel {
        SaveArticle(article)
        fmt.Printf("SAVED")
    }
  }(okChannel)

  go func(rejectChannel chan Article) {
    for article := range rejectChannel {
        RejectArticle(article)
        fmt.Printf("REJECTED")
    }
  }(rejectChannel)

请帮忙

Answer 1

我不是很清楚你的意思。您可能想要做的是拥有一个 channel ，根据包含信息的数据类型为您提供所需的所有信息。这是我的意思的一个例子。

type ArticleClassifier struct {
    Rejected bool
    Article  Article
}

func main() {
    classifierChannel := make(chan ArticleClassifier)
    go Init(classifierChannel)

    go func(ch chan ArticleClassifier) {
        for ac := range classifierChannel {
            if ac.Rejected {
                // Reject article
            } else {
                // Save article
            }
        }
    }
}

但是您上面的解决方案似乎没有必要这样做。也许您的问题是该计划将提前结束？如果那是问题所在，那就不要让第二个监听器成为 goroutine。这是您示例中的一些重构代码:

func main() {
    okChannel := make(chan Article)
    rejectChannel := make(chan Article)

    scrape := new(scrapers)
    go Init(okChannel, rejectChannel)

    go func(okChannel chan Article) {
        for article := range okChannel {
            SaveArticle(article)
            fmt.Printf("SAVED")
        }
    }(okChannel)

    for article := range rejectChannel {
        RejectArticle(article)
        fmt.Printf("REJECTED")
    }
}