如何同时收听两个 Golang 频道？

Question

我基本上是通过刮板收集数据，然后想以两种方式处理这些数据：拒绝或接受。为了不减慢爬虫的速度，我将数据传递给拒绝通道或接受通道。我需要这些频道同时收听各自频道中的数据。这是我迄今为止的尝试：

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

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

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")
    }
}