理解并发编程：Goroutines、Channels、WaitGroups和select

亲爱的编程学习爱好者，如果你点开了这篇文章，说明你对《理解并发编程：Goroutines、Channels、WaitGroups和select》很感兴趣。本篇文章就来给大家详细解析一下，主要介绍一下，希望所有认真读完的童鞋们，都有实质性的提高。

我不一定要完成一些特定的事情，更多只是了解 goroutine、通道、waitgroup 和 select（在通道上）如何一起发挥作用。我正在编写一个简单的程序，它循环遍历 url 片段，获取 url，然后基本上就结束了。简单的想法是，我希望所有的提取都发生并返回，通过通道发送数据，然后在所有提取发生后结束。我快到了，我知道我的选择中缺少一些将结束循环的东西，一些东西说“嘿等待组现在是空的”，但我不确定如何最好地做到这一点。介意看一下并帮我清理一下吗？现在一切都运行得很好，只是没有终止，所以显然我错过了一些东西和/或不理解其中一些组件应该如何协同工作。

package main

import (
    "fmt"
    "io/ioutil"
    "net/http"
    "sync"
)

var urls = []string{
    "https://www.google.com1",
    "https://www.gentoo.org",
}

var wg sync.WaitGroup

// simple struct to store fetching
type urlObject struct {
    url     string
    success bool
    body    string
}

func getPage(url string, channelMain chan urlObject, channelError chan error) {

    // increment waitgroup, defer decrementing
    wg.Add(1)
    defer wg.Done()
    fmt.Println("fetching " + url)

    // create a urlObject
    uO := urlObject{
        url:     url,
        success: false,
    }

    // get URL
    response, getError := http.Get(url)

    // close response later on
    if response != nil {
        defer response.Body.Close()
    }

    // send error over error channel if one occurs
    if getError != nil {
        channelError <- getError
        return
    }

    // convert body to []byte
    body, conversionError := ioutil.ReadAll(response.Body)
    // convert []byte to string
    bodyString := string(body)

    // if a conversion error happens send it over the error channel
    if conversionError != nil {
        channelError <- conversionError
    } else {
        // if not send a urlObject over the main channel
        uO.success = true
        uO.body = bodyString
        channelMain <- uO
    }

}

func main() {

    var channelMain = make(chan urlObject)
    var channelError = make(chan error)

    for _, v := range urls {
        go getPage(v, channelMain, channelError)
    }

    // wait on goroutines to finish
    wg.Wait()

    for {
        select {
        case uO := <-channelMain:
            fmt.Println("completed " + uO.url)
        case err := <-channelError:
            fmt.Println("error: " + err.Error())
        }
    }

}

解决方案

您需要进行以下更改：

• 正如人们所提到的，在调用 goroutine 之前，您可能需要在 main 函数中调用 wg.add(1) 。这样您就知道它发生在 defer wg.done() 调用之前。
• 你的通道读取将会阻塞，除非你能找到一种方法来关闭你的 goroutine 中的通道，或者让它们缓冲。也许最简单的方法是让它们缓冲，例如 var channelmain = make(chan urlobject, len(urls))
• select 语句中的 break 将仅退出 select，而不退出包含 for 循环。您可以标记 for 循环并对其进行中断，或者使用某种条件变量。

工作版本演示链接：https://play.golang.org/p/WH1fm2MhP-L

package main

import (
    "fmt"
    "io/ioutil"
    "net/http"
    "sync"
)

var urls = []string{
    "https://www.google.com1",
    "https://www.gentoo.org",
}

var wg sync.WaitGroup

// simple struct to store fetching
type urlObject struct {
    url     string
    success bool
    body    string
}

func getPage(url string, channelMain chan urlObject, channelError chan error) {

    // increment waitgroup, defer decrementing
    defer wg.Done()
    fmt.Println("fetching " + url)

    // create a urlObject
    uO := urlObject{
        url:     url,
        success: false,
    }

    // get URL
    response, getError := http.Get(url)

    // close response later on
    if response != nil {
        defer response.Body.Close()
    }

    // send error over error channel if one occurs
    if getError != nil {
        channelError <- getError
        return
    }

    // convert body to []byte
    body, conversionError := ioutil.ReadAll(response.Body)
    // convert []byte to string
    bodyString := string(body)

    // if a conversion error happens send it over the error channel
    if conversionError != nil {
        channelError <- conversionError
    } else {
        // if not send a urlObject over the main channel
        uO.success = true
        uO.body = bodyString
        channelMain <- uO
    }

}

func main() {

    var channelMain = make(chan urlObject, len(urls))
    var channelError = make(chan error, len(urls))

    for _, v := range urls {
        wg.Add(1)
        go getPage(v, channelMain, channelError)
    }

    // wait on goroutines to finish
    wg.Wait()

    for done := false; !done; {
        select {
        case uO := <-channelMain:
            fmt.Println("completed " + uO.url)
        case err := <-channelError:
            fmt.Println("error: " + err.Error())
        default:
            done = true
        }
    }

}

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。