Golang优雅关闭channel的方法示例_Golang

Golang优雅关闭channel的方法示例

2020-05-11 10:44小谈 Golang

Goroutine和channel是Go在“并发”方面两个核心feature，下面这篇文章主要给大家介绍了关于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()

									}