迭代通道时关闭通道的最佳时间

你在学习Golang相关的知识吗？本文《迭代通道时关闭通道的最佳时间》，主要介绍的内容就涉及到，如果你想提升自己的开发能力，就不要错过这篇文章，大家要知道编程理论基础和实战操作都是不可或缺的哦！

我正在使用 golang，我创建了这个小应用程序来使用 goroutine 进行多个并发 api 调用。

当应用程序工作时，调用完成后，应用程序会卡住，这是有道理的，因为它无法退出 range c 循环，因为通道未关闭。

我不知道在这种模式下哪里可以更好地关闭通道。

package main

import "fmt"
import "net/http"

func main() {
    links := []string{
        "https://github.com/fabpot",
        "https://github.com/andrew",
        "https://github.com/taylorotwell",
        "https://github.com/egoist",
        "https://github.com/HugoGiraudel",
    }

    checkUrls(links)
}

func checkUrls(urls []string) {
    c := make(chan string)

    for _, link := range urls {
        go checkUrl(link, c)
    }

    for msg := range c {
        fmt.Println(msg)
    }

    close(c) //this won't get hit
}

func checkUrl(url string, c chan string) {
    _, err := http.Get(url)

    if err != nil {
        c <- "We could not reach:" + url
    } else {
        c <- "Success reaching the website:" + url
    }
}

解决方案

当没有更多的值要发送时，您将关闭通道，因此在本例中，是在所有 checkurl goroutines 完成时关闭通道。

var wg sync.waitgroup

func checkurls(urls []string) {
    c := make(chan string)

    for _, link := range urls {
        wg.add(1)
        go checkurl(link, c)
    }

    go func() {
        wg.wait()
        close(c)
    }()

    for msg := range c {
        fmt.println(msg)
    }
}

func checkurl(url string, c chan string) {
    defer wg.done()
    _, err := http.get(url)

    if err != nil {
        c <- "we could not reach:" + url
    } else {
        c <- "success reaching the website:" + url
    }
}

（请注意，来自 http.get 的 error 仅反映连接和协议错误。如果您也期望这些错误，则它不会包含 http 服务器错误，您必须了解如何检查这些错误路径而不仅仅是主机。）

当使用通道和 goroutine 在 go 中编写程序时，始终要考虑谁（哪个函数）拥有通道。我更喜欢让拥有通道的函数关闭它的做法。如果我要写这个，我会这样做。

注意：处理此类情况的更好方法是扇出、扇入并发模式。参考(https://blog.golang.org/pipelines)Go Concurrency Patterns

package main

import "fmt"
import "net/http"
import "sync"

func main() {
    links := []string{
        "https://github.com/fabpot",
        "https://github.com/andrew",
        "https://github.com/taylorotwell",
        "https://github.com/egoist",
        "https://github.com/HugoGiraudel",
    }


    processURLS(links)
    fmt.Println("End of Main")
}

func processURLS(links []string) {
    resultsChan := checkUrls(links)

    for msg := range resultsChan {
        fmt.Println(msg)
    }

}     

func checkUrls(urls []string) chan string {

    outChan := make(chan string)

    go func(urls []string) {
       defer close(outChan)

       var wg sync.WaitGroup
       for _, url := range urls {
         wg.Add(1)
          go checkUrl(&wg, url, outChan)
       }
       wg.Wait()

    }(urls)

    return outChan
}

func checkUrl(wg *sync.WaitGroup, url string, c chan string) {
    defer wg.Done()
    _, err := http.Get(url)

    if err != nil {
        c <- "We could not reach:" + url
    } else {
        c <- "Success reaching the website:" + url
    }
}

好了，本文到此结束，带大家了解了《迭代通道时关闭通道的最佳时间》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！