如何优雅的关闭Golang Channel?

Channel关闭原则

不要在消费端关闭channel,不要在有多个并行的生产者时对channel执行关闭操作。
也就是说应该只在[唯一的或者最后唯一剩下]的生产者协程中关闭channel,来通知消费者已经没有值可以继续读了。只要坚持这个原则,就可以确保向一个已经关闭的channel发送数据的情况不可能发生。

暴力关闭channel的正确方法

如果想要在消费端关闭channel,或者在多个生产者端关闭channel,可以使用recover机制来上个保险,避免程序因为panic而崩溃。

func SafeClose(ch chan T) (justClosed bool) {
     defer func() {
        if recover() != nil {
            justClosed = false
        }
    }()
    close(ch)
    return true
}

使用这种方法明显违背了上面的channel关闭原则,然后性能还可以,毕竟在每个协程只会调用一次SafeClose,性能损失很小。
同样也可以在生产消息的时候使用recover方法。
1103448-20181025110425474-555840324.png

礼貌关闭channel方法

还有不少人经常使用sync.Once来关闭channel,这样可以确保只会关闭一次
1103448-20181025111926714-1296498789.png

同样我们也可以使用sync.Mutex达到同样的目的。
1103448-20181025112558312-993308002.png
要知道golang的设计者不提供SafeClose或者SafeSend方法是有原因的,
他们本来就不推荐在消费端或者在并发的多个生产端关闭channel,
比如关闭只读channel在语法上就彻底被禁止使用了。

优雅的关闭channel的方法

多个消费者,单个生产者.

1103448-20181025132249937-1812266034.png

多个生产者,单个消费者。

1103448-20181025133925886-253655430.png
就上面这个例子,生产者同时也是退出信号channel的接受者,退出信号channel仍然是由它的生产者

多个生产者,多个消费者

1103448-20181025140716233-1842118870.png

转载于:https://www.cnblogs.com/Csir/p/9848439.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值