Golang优雅关闭channel的方法示例
前言
最近使用go开发后端服务，服务关闭需要保证channel中的数据都被读取完，理由很简单，在收到系统的中断信号后，系统需要做收尾工作，保证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()
}
这样就可以保证不会往已经关闭的 channel 中发送数据了。
总结
以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，如果有疑问大家可以留言交流，谢谢大家对脚本之家的支持。
相关文章
为了确保数据一致性，在项目中会经常用到事务处理，对于MySQL事务相信大家应该都不陌生。这篇文章主要总结一下在Go语言中Gorm是如何实现事务的；感兴趣的小伙伴们可以参考借鉴，希望对大家能有所帮助2022-09-09
在项目开发中，我们常常会遇到处理来自数百万个端点的大量POST请求，本文主要介绍了Golang实现处理每分钟100万个请求的方法，希望对大家有所帮助2023-04-04
GoFrame框架gset使用对比PHP Java Redis优势
这篇文章主要为大家介绍了GoFrame框架gset对比PHP Java Redis的使用优势详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-06-06
这篇文章主要介绍了GO 语言运行环境的基础知识的相关资料,需要的朋友可以参考下2022-09-09
golang跳转语句goto,break,continue的使用及区别说明
这篇文章主要介绍了golang跳转语句goto,break,continue的使用及区别说明，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-12-12
这篇文章主要介绍了Go语言编程的入门指南,包括对Go的变量及函数的基本介绍,需要的朋友可以参考下2016-01-01
这篇文章主要介绍了基于golang中container/list包的用法说明，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-04-04
这篇文章主要给大家介绍了关于Go语言学习技巧之如何合理使用Pool的相关资料，Pool用于存储那些被分配了但是没有被使用，而未来可能会使用的值，以减小垃圾回收的压力。文中通过示例代码介绍的非常详细，需要的朋友可以参考下。2017-12-12
这篇文章主要介绍了详解如何热重启golang服务器，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-08-08
这篇文章主要为大家介绍了go语言区块链学习如何调用以太坊的示例实现过程，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步2021-10-10
最新评论