Golang优雅关闭channel的方法示例

作者:小谈 时间:2024-05-03 15:05:44 

前言

最近使用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 中发送数据了。

来源:https://blog.tanteng.me/2017/11/golang-close-channel/

标签:golang,优雅关闭,channel
0
投稿

猜你喜欢

  • python+opencv实现霍夫变换检测直线

    2021-07-30 04:08:57
  • chatGPT之Python API启用上下文管理案例详解

    2022-09-21 13:51:43
  • MySql闪退和服务无法启动的解决方法

    2024-01-21 07:53:12
  • Python @property使用方法解析

    2021-10-27 05:42:46
  • Asp 操作Cookies(包括设置[赋值]、读取、删除[设置过期时间])

    2011-03-10 11:06:00
  • 使用 Python 处理3万多条数据只要几秒钟

    2023-08-03 15:21:06
  • 用css+js给网页背景插入flash播放器

    2007-10-21 09:27:00
  • 15个Pythonic的代码示例(值得收藏)

    2022-07-27 21:21:19
  • python神经网络使用Keras构建RNN训练

    2021-07-19 21:12:15
  • 256种编程语言大汇总

    2023-05-30 19:06:01
  • Python绘制1000响大地红鞭炮动态效果

    2021-08-21 20:44:26
  • MySQL存储引擎InnoDB与Myisam的区别分析

    2024-01-18 13:34:50
  • Django实现学生管理系统

    2023-07-22 18:05:59
  • Python的内置数据类型中的数字

    2023-12-29 19:36:37
  • 微信小程序创建自定义全局函数以及其调用方法详解

    2023-08-24 20:43:22
  • 五分钟学会Python 模块和包、文件

    2023-06-01 20:05:38
  • 如何在pycharm中安装第三方包

    2021-10-13 21:30:01
  • golang goquery selector选择器使用示例大全

    2023-10-14 15:40:58
  • flash(swf)遮住网页内容div的解决

    2007-10-31 07:29:00
  • go语言中函数与方法介绍

    2024-04-23 09:34:13
  • asp之家 网络编程 m.aspxhome.com