• 设为首页
  • 点击收藏
  • 手机版
    手机扫一扫访问
    迪恩网络手机版
  • 关注官方公众号
    微信扫一扫关注
    公众号

go检查channel是否关闭

原作者: [db:作者] 来自: [db:来源] 收藏 邀请

 

golang 中channel 即使已经关闭了, 仍然可以将channel中的数据读出来, 并不会报错。

一般的写法: data, ok := <- chan,

只有当channel无数据channel被close了,才会返回ok=false。

 

 

 

package main

import (
  "context"
  "fmt"
  "reflect"
  "time"
  "unsafe"
)

func isChanClosed(ch interface{}) bool {
  if reflect.TypeOf(ch).Kind() != reflect.Chan {
    panic("only channels!")
  }

  // get interface value pointer, from cgo_export
  // typedef struct { void *t; void *v; } GoInterface;
  // then get channel real pointer
  cptr := *(*uintptr)(unsafe.Pointer(
    unsafe.Pointer(uintptr(unsafe.Pointer(&ch)) + unsafe.Sizeof(uint(0))),
  ))

  // this function will return true if chan.closed > 0
  // see hchan on https://github.com/golang/go/blob/master/src/runtime/chan.go
  // type hchan struct {
  // qcount   uint           // total data in the queue
  // dataqsiz uint           // size of the circular queue
  // buf      unsafe.Pointer // points to an array of dataqsiz elements
  // elemsize uint16
  // closed   uint32
  // **

  cptr += unsafe.Sizeof(uint(0)) * 2
  cptr += unsafe.Sizeof(unsafe.Pointer(uintptr(0)))
  cptr += unsafe.Sizeof(uint16(0))
  return *(*uint32)(unsafe.Pointer(cptr)) > 0
}

func main() {
  c := make(chan int, 10)
  c <- 1
  c <- 2
  c <- 3

  ctx, cancel := context.WithCancel(context.Background())
  close(c)
  cancel()
  fmt.Println("whether channel is closed:", isChanClosed(c), "\n")

exit:
  for {
    select {
    case i, ok := <-c:
      fmt.Println(ok)
      if !ok {
        fmt.Println("channel closed!")
        break
      }
      fmt.Println(i)

    case <-ctx.Done():
      break exit
    }
  }
  fmt.Println("aaaaaaaa")
  time.Sleep(time.Second * 2)
}

  

我们都知道data, ok := <- chan第一个变量表示读出的数据,第二个变量表示是否成功读取了数据,有意思的是,第二个变量并不用于指示管道的关闭的状态。第二个变量常常被误以为关闭状态是因为它确实和管道状态有关,确切的来说,是和管道缓冲区是否有数据有关。

如果判断golang的channel是否关闭,data, ok := <- chan,当ok不是true的时候,说明是channel关闭了。 那么问题来了,channel关闭了,我们是否可以立马获取到channel被关闭的状态?我想这个问题不少人没有去想吧?为什么有这样的问题?  来自我的一个bug,我期初认为close了一个channel,消费端的goroutine自然是可以拿到channel的关闭状态。然而事实并不是这样的。 只有当channel无数据,且channel被close了,才会返回ok=false。  所以,只要有堆积,就不会返回关闭状态。导致我的服务花时间来消费堆积,才会退出。
————————————————
版权声明:本文为CSDN博主「github_zwl」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/github_34457546/article/details/109687879

 

 

 

参考: https://blog.csdn.net/github_34457546/article/details/109687879

https://stackoverflow.com/questions/16105325/how-to-check-a-channel-is-closed-or-not-without-reading-it

----------------------------------------------


鲜花

握手

雷人

路过

鸡蛋
该文章已有0人参与评论

请发表评论

全部评论

专题导读
热门推荐
热门话题
阅读排行榜

扫描微信二维码

查看手机版网站

随时了解更新最新资讯

139-2527-9053

在线客服(服务时间 9:00~18:00)

在线QQ客服
地址:深圳市南山区西丽大学城创智工业园
电邮:jeky_zhao#qq.com
移动电话:139-2527-9053

Powered by 互联科技 X3.4© 2001-2213 极客世界.|Sitemap