Golang通道的无阻塞读写的方法示例

更新时间：2018年11月04日 16:39:23 作者：shitaibin

这篇文章主要介绍了Golang通道的无阻塞读写的方法示例，详细的介绍了哪些情况会存在阻塞，以及如何使用select解决阻塞，非常具有实用价值，需要的朋友可以参考下

无论是无缓冲通道，还是有缓冲通道，都存在阻塞的情况，但其实有些情况，我们并不想读数据或者写数据阻塞在那里，有1个唯一的解决办法，那就是使用select结构。

这篇文章会介绍，哪些情况会存在阻塞，以及如何使用select解决阻塞。

阻塞场景

阻塞场景共4个，有缓存和无缓冲各2个。

无缓冲通道的特点是，发送的数据需要被读取后，发送才会完成，它阻塞场景：

通道中无数据，但执行读通道。
通道中无数据，向通道写数据，但无协程读取。

// 场景1
func ReadNoDataFromNoBufCh() {
 noBufCh := make(chan int)

 <-noBufCh
 fmt.Println("read from no buffer channel success")

 // Output:
 // fatal error: all goroutines are asleep - deadlock!
}

// 场景2
func WriteNoBufCh() {
 ch := make(chan int)

 ch <- 1
 fmt.Println("write success no block")
 
 // Output:
 // fatal error: all goroutines are asleep - deadlock!
}

注：示例代码中的Output注释代表函数的执行结果，每一个函数都由于阻塞在通道操作而无法继续向下执行，最后报了死锁错误。

有缓存通道的特点是，有缓存时可以向通道中写入数据后直接返回，缓存中有数据时可以从通道中读到数据直接返回，这时有缓存通道是不会阻塞的，它阻塞的场景是：

通道的缓存无数据，但执行读通道。
通道的缓存已经占满，向通道写数据，但无协程读。

// 场景1
func ReadNoDataFromBufCh() {
 bufCh := make(chan int, 1)

 <-bufCh
 fmt.Println("read from no buffer channel success")

 // Output:
 // fatal error: all goroutines are asleep - deadlock!
}

// 场景2
func WriteBufChButFull() {
 ch := make(chan int, 1)
 // make ch full
 ch <- 100

 ch <- 1
 fmt.Println("write success no block")
 
 // Output:
 // fatal error: all goroutines are asleep - deadlock!
}

使用Select实现无阻塞读写

select是执行选择操作的一个结构，它里面有一组case语句，它会执行其中无阻塞的那一个，如果都阻塞了，那就等待其中一个不阻塞，进而继续执行，它有一个default语句，该语句是永远不会阻塞的，我们可以借助它实现无阻塞的操作。

下面示例代码是使用select修改后的无缓冲通道和有缓冲通道的读写，以下函数可以直接通过main函数调用，其中的Ouput的注释是运行结果，从结果能看出，在通道不可读或者不可写的时候，不再阻塞等待，而是直接返回。

// 无缓冲通道读
func ReadNoDataFromNoBufChWithSelect() {
 bufCh := make(chan int)

 if v, err := ReadWithSelect(bufCh); err != nil {
  fmt.Println(err)
 } else {
  fmt.Printf("read: %d\n", v)
 }

 // Output:
 // channel has no data
}

// 有缓冲通道读
func ReadNoDataFromBufChWithSelect() {
 bufCh := make(chan int, 1)

 if v, err := ReadWithSelect(bufCh); err != nil {
  fmt.Println(err)
 } else {
  fmt.Printf("read: %d\n", v)
 }

 // Output:
 // channel has no data
}

// select结构实现通道读
func ReadWithSelect(ch chan int) (x int, err error) {
 select {
 case x = <-ch:
  return x, nil
 default:
  return 0, errors.New("channel has no data")
 }
}

// 无缓冲通道写
func WriteNoBufChWithSelect() {
 ch := make(chan int)
 if err := WriteChWithSelect(ch); err != nil {
  fmt.Println(err)
 } else {
  fmt.Println("write success")
 }

 // Output:
 // channel blocked, can not write
}

// 有缓冲通道写
func WriteBufChButFullWithSelect() {
 ch := make(chan int, 1)
 // make ch full
 ch <- 100
 if err := WriteChWithSelect(ch); err != nil {
  fmt.Println(err)
 } else {
  fmt.Println("write success")
 }

 // Output:
 // channel blocked, can not write
}

// select结构实现通道写
func WriteChWithSelect(ch chan int) error {
 select {
 case ch <- 1:
  return nil
 default:
  return errors.New("channel blocked, can not write")
 }
}

使用Select+超时改善无阻塞读写

使用default实现的无阻塞通道阻塞有一个缺陷：当通道不可读或写的时候，会即可返回。实际场景，更多的需求是，我们希望，尝试读一会数据，或者尝试写一会数据，如果实在没法读写，再返回，程序继续做其它的事情。

使用定时器替代default可以解决这个问题。比如，我给通道读写数据的容忍时间是500ms，如果依然无法读写，就即刻返回，修改一下会是这样：

func ReadWithSelect(ch chan int) (x int, err error) {
 timeout := time.NewTimer(time.Microsecond * 500)

 select {
 case x = <-ch:
  return x, nil
 case <-timeout.C:
  return 0, errors.New("read time out")
 }
}

func WriteChWithSelect(ch chan int) error {
 timeout := time.NewTimer(time.Microsecond * 500)

 select {
 case ch <- 1:
  return nil
 case <-timeout.C:
  return errors.New("write time out")
 }
}

结果就会变成超时返回：

read time out
write time out
read time out
write time out

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

go语言base64用法实例
这篇文章主要介绍了go语言base64用法,实例分析了Go语言base64编码的实用技巧,具有一定参考借鉴价值,需要的朋友可以参考下
2015-02-02
详解Golang 推荐的命名规范
这篇文章主要介绍了详解Golang 推荐的命名规范，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2019-02-02
go install/build生成的文件命名和路径操作
这篇文章主要介绍了go install/build生成的文件命名和路径操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-12-12
详解Go语言如何使用标准库sort对切片进行排序
Sort 标准库提供了对基本数据类型的切片和自定义类型的切片进行排序的函数。今天主要分享的内容是使用 Go 标准库 sort 对切片进行排序，感兴趣的可以了解一下
2022-12-12
Go中使用单调时钟获得准确的时间间隔问题
这篇文章主要介绍了Go中使用单调时钟获得准确的时间间隔,在go语言中，没有直接调用时钟的函数，可以通过 time.Now() 获得带单调时钟的 Time 结构体，并通过Since和Until获得相对准确的时间间隔，需要的朋友可以参考下
2022-06-06
golang实现通过smtp发送电子邮件的方法
这篇文章主要介绍了golang实现通过smtp发送电子邮件的方法,实例分析了Go语言基于SMTP协议发送邮件的相关技巧,需要的朋友可以参考下
2016-07-07
利用Go语言搭建WebSocket服务端方法示例
这篇文章主要给大家介绍了利用Go语言搭建WebSocket服务端方法，文中通过示例代码介绍的非常详细，需要的朋友们可以参考借鉴，下面来一起看看吧。
2017-04-04
Go语言实现登录验证代码案例
这篇文章主要介绍了Go语言实现登录验证代码案例，代码和图文讲解的很清晰，有感兴趣的可以学习下
2021-03-03
Go语言配置解析库viper的使用指南
viper 配置管理解析库，是由大神 Steve Francia 开发，本文就来和大家详细讲讲它的具体使用，文中的示例代码讲解详细，需要的可以收藏一下
2023-06-06
go json编译原理XJSON实现四则运算
这篇文章主要为大家介绍了go json编译原理XJSON实现四则运算示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2022-07-07