温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

使用Golang怎么关闭channel

发布时间:2021-04-08 16:13:32 来源:亿速云 阅读:1376 作者:Leah 栏目:编程语言

使用Golang怎么关闭channel?相信很多没有经验的人对此束手无策,为此本文总结了问题出现的原因和解决方法,通过这篇文章希望你能解决这个问题。

关于Go channel设计和规范的批评:

  • 在不能更改channel状态的情况下,没有简单普遍的方式来检查channel是否已经关闭了

  • 关闭已经关闭的channel会导致panic,所以在closer(关闭者)不知道channel是否已经关闭的情况下去关闭channel是很危险的

  • 发送值到已经关闭的channel会导致panic,所以如果sender(发送者)在不知道channel是否已经关闭的情况下去向channel发送值是很危险的

所以Golang 内建的 close 方法可以关闭 channel,如果往已经关闭的 channel 发送数据,则会报错:panic: close of closed channel.

看如下代码,在一段时间内,生产者可以不断往 channel 写入数据,消费者进行处理,一段时间后 channel 关闭了,这个时候如果还有数据往 channel 发送,程序就会报错。

package main   import (  "fmt"  "sync"  "time" )   func main() {  jobs := make(chan int)  var wg sync.WaitGroup  go func() {  time.Sleep(time.Second * 3)  close(jobs)  }()  go func() {  for i := 0; ; i++ {  jobs <- i  fmt.Println("produce:", i)  }  }()  wg.Add(1)  go func() {  defer wg.Done()  for i := range jobs {  fmt.Println("consume:", i)  }  }()  wg.Wait() }

多运行几次出错的概率会比较大:

produce: 33334 consume: 33334 consume: 33335 produce: 33335 produce: 33336 consume: 33336 consume: 33337 produce: 33337 produce: 33338 consume: 33338 consume: 33339 produce: 33339 produce: 33340 consume: 33340 panic: send on closed channel   goroutine 19 [running]: panic(0x49b660, 0xc042410bb0)   C:/Go/src/runtime/panic.go:500 +0x1af main.main.func2(0xc04203a180)   C:/Users/tanteng/Go/src/examples/channel_close.go:18 +0x6b created by main.main   C:/Users/tanteng/Go/src/examples/channel_close.go:21 +0xb8 exit status 2

如何优雅关闭 channel

那么在往通道发数据前如何判断通道是否关闭呢?

1._,ok := <- jobs

此时如果 channel 关闭,ok 值为 false,如果 channel 没有关闭,则会漏掉一个 jobs

2.使用 select 方式

再创建一个 channel,叫做 timeout,如果超时往这个 channel 发送 true,在生产者发送数据给 jobs 的 channel,用 select 监听 timeout,如果超时则关闭 jobs 的 channel.

完整代码如下:

package main   import (  "fmt"  "sync"  "time" )   func main() {  jobs := make(chan int)  timeout := make(chan bool)  var wg sync.WaitGroup  go func() {  time.Sleep(time.Second * 3)  timeout <- true  }()  go func() {  for i := 0; ; i++ {  select {  case <-timeout:  close(jobs)  return    default:  jobs <- i  fmt.Println("produce:", i)  }  }  }()  wg.Add(1)  go func() {  defer wg.Done()  for i := range jobs {  fmt.Println("consume:", i)  }  }()  wg.Wait() }

看完上述内容,你们掌握使用Golang怎么关闭channel的方法了吗?如果还想学到更多技能或想了解更多相关内容,欢迎关注亿速云行业资讯频道,感谢各位的阅读!

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI