在这个具体案例中,应该把通道关闭在哪里?
来源: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)
编写代码时遇到的另一个结构问题是 gen
和 fetchurl
都会创建通道,运行所有应该写入通道的代码,并仅在这些编写器完成后返回通道;由于在函数返回之前无法从通道中读取任何内容,因此这将导致死锁。您可以通过在顶层创建所有通道并将它们传递到生成器函数来解决此问题。
如果您希望两个工作线程从同一个 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学习网公众号。
-
502 收藏
-
502 收藏
-
501 收藏
-
501 收藏
-
501 收藏
-
139 收藏
-
204 收藏
-
325 收藏
-
477 收藏
-
486 收藏
-
439 收藏
-
- 前端进阶之JavaScript设计模式
- 设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
- 立即学习 542次学习
-
- GO语言核心编程课程
- 本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
- 立即学习 507次学习
-
- 简单聊聊mysql8与网络通信
- 如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
- 立即学习 497次学习
-
- JavaScript正则表达式基础与实战
- 在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
- 立即学习 487次学习
-
- 从零制作响应式网站—Grid布局
- 本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
- 立即学习 484次学习