登录
首页 >  Golang >  Go问答

在这个具体案例中,应该把通道关闭在哪里?

来源:stackoverflow

时间:2024-02-29 16:33:26 477浏览 收藏

golang学习网今天将给大家带来《在这个具体案例中,应该把通道关闭在哪里?》,感兴趣的朋友请继续看下去吧!以下内容将会涉及到等等知识点,如果你是正在学习Golang或者已经是大佬级别了,都非常欢迎也希望大家都能给我建议评论哈~希望能帮助到大家!

问题内容

我只是编写一个简单的 go 管道,目标是获取 url 并打印状态。

在 fetchurl 上,我需要关闭通道来通知 main,将没有数据传入,因此释放 main go 例程。但是,我无法在循环后真正关闭 fetchurl 函数上的通道,因为这太快了。我不想在应用程序中添加等待组,因为目前我的整个目标是了解渠道。

在 fetchurl 函数中,调用 2 的通道只是为了确保同时只有 2 个作业。

package main

import (
    "fmt"
    "net/http"
    "os"
)

func gen(val []string) <-chan string {
    out := make(chan string, len(val))
    for _, val := range val {
        out <- val
    }
    close(out)
    return out
}

func fetchUrl(in <-chan string) <-chan string {
    out := make(chan string)
    two := make(chan struct{}, 2)
    fmt.Println("blocked")
    for url := range in {
        two <- struct{}{}
        go fetchWorker(url, two, out)
    }

    return out
}

func fetchWorker(url string, two chan struct{}, out chan string) {
    res, err := http.Get("https://" + url)
    if err != nil {
        panic(err)
    }
    <-two
    out <- fmt.Sprintf("[%d] %s\n", res.StatusCode, url)
}

func main() {
    for val := range fetchUrl(gen(os.Args[1:])) {
        fmt.Println(val)
    }
}

解决方案


在将每个结果写入 out 通道后,您需要关闭该通道。判断这一点的最简单方法是当所有工作协程都退出时,而判断这一点的最简单方法是使用 sync.waitgroup。 (在 go 中,通道和 goroutine 是非常密切相关的概念,因此 goroutine 管理是通道工作的一部分。)

在现有代码中,您可以将其绑定到 fetchurl 函数中:

var wg sync.waitgroup
for url := range in {
    two <- struct{}{}
    wg.add(1)
    go func() {
        defer wg.done()
        fetchworker(url, two, out)
    }()
}
wg.wait()
close(out)

编写代码时遇到的另一个结构问题是 genfetchurl 都会创建通道,运行所有应该写入通道的代码,并仅在这些编写器完成后返回通道;由于在函数返回之前无法从通道中读取任何内容,因此这将导致死锁。您可以通过在顶层创建所有通道并将它们传递到生成器函数来解决此问题。

如果您希望两个工作线程从同一个 url 队列中读取数据,标准模式是启动两个 goroutine 从相同的通道中读取和写入。例如,您可以将 fetchworker 重写为

func fetchworker(urls <-chan string, out chan<- string) {
    for url := range urls {
        res, err := http.get("https://" + url)
        if err != nil {
            panic(err)
        }
        out <- fmt.sprintf("[%d] %s\n", res.statuscode, url)
    }
}

在顶层,创建通道、创建工作人员、提供输入并使用输出。

func main() {
    urls := make(chan string)
    out := make(chan string)

    // Launch a goroutine to feed data into urls, then
    // close(urls), then stop
    go gen(os.Args[1:], urls)

    // Launch worker goroutines
    workerCount := 2
    var wg sync.WaitGroup
    for i := 0; i < workerCount; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            fetchWorker(urls, out)
        }()
    }

    // Launch a dedicated goroutine to close the channel
    go func() {
        wg.Wait()
        close(out)
    }()

    // Read the results
    for result := range(out) {
        fmt.Println(result)
    }
}

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

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>