关于GOLANG的chan

前端之家收集整理的这篇文章主要介绍了关于GOLANG的chan前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

GOLANG CHAN

chan是golang中非常重要的一个东西,用来做goroutine的通信,因为golang程序必然会有多个goroutine,如何同步这些goroutine就很重要了。

使用chan时有几个心得:

  1. 首先,永远是符号<-进行读取或者写入,譬如v,ok := <-c是读取,而c <- v是写入。
  2. 其次,读取时,如果没有ok,也是可以读取的。不过如果closed也是能读的,没有赋值而已;如果要知道是否closed得加ok,也就是除非chan永远不关闭,否则读取应该用v,ok := <-c而不是用v := <-c的方式。
  3. 再次,不能向closed的chan写入,所以一般写入时需要用一个信号的chan(一般buffer为1),来判断是否写入或者放弃,用select判断是写入成功了,还是正在关闭需要放弃写入。
  4. 最后,如果closed后,chan有数据,ok还是true的,直到chan没有数据了才false。

读写Chan

永远是符号<-进行读取或者写入,譬如v,ok := <-c是读取,而c <- v是写入。

  1. c := make(chan int, 1)
  2. c <- 10 // 写入chan
  3. v := <- c // 从chan中读取

下面的例子判断chan是否关闭

  1. c := make(chan int, 1)
  2. c <- 10
  3. v,ok := <- c // 读取,v=10,ok=true
  4. close(c)
  5. v,ok := <- c // 读取,v=0,ok=false

如果写不进去就丢弃,可以用select:

  1. c := make(chan int, 1)
  2.  
  3. select {
  4. case c <- 10: // c中放入了10,因为chan的buffer为1
  5. default:
  6. }
  7.  
  8. select {
  9. case c <- 11:
  10. default: // c中只有10,没有11
  11. }
  12.  
  13. select {
  14. case v,ok := <- c:
  15. // 读出来一个,v=10,ok=true
  16. default:
  17. }
  18.  
  19. select {
  20. case v,ok := <- c:
  21. default: // 没有可读的,走这个分支
  22. }

还可以用超时之类的,也是一个chan,time.After(xxx)返回的就是chan。

判断closed

读取时,如果没有ok,也是可以读取的。不过如果closed也是能读的,没有赋值而已;如果要知道是否closed得加ok,也就是除非chan永远不关闭,否则读取应该用v,ok := <-c而不是用v := <-c的方式。

  1. c := make(chan int, 1)
  2. c <- 10
  3. close(c)
  4.  
  5. v := <- c // c=10,读取出来一个
  6. v = <- c // c=0,实际上没有读出来,但是判断不了
  1. c := make(chan int, 1)
  2. c <- 10
  3. close(c)
  4.  
  5. v,ok := <- c // c=10,ok=true,读取出来一个
  6. v,ok = <- c // c=0,ok=false,实际上没有读出来

写入chan

不能向closed的chan写入,所以一般写入时需要用一个信号的chan,来判断是否写入或者放弃,用select判断是写入成功了,还是正在关闭需要放弃写入。

  1. type TcpListeners struct {
  2. conns chan *net.TCPConn
  3. closing chan bool
  4. wait *sync.WaitGroup
  5. }
  6.  
  7. func NewTcpListeners(addrs []string) (v *TcpListeners,err error) {
  8. v = &TcpListeners{
  9. addrs: addrs,conns: make(chan *net.TCPConn),closing: make(chan bool, 1),wait: &sync.WaitGroup{},}
  10.  
  11. return
  12. }
  13.  
  14. // Listen at addrs format as netowrk://laddr,for example,
  15. // tcp://:1935,tcp4://:1935,tcp6://1935,tcp://0.0.0.0:1935
  16. func (v *TcpListeners) ListenTCP() (err error) {
  17. for _,addr := range v.addrs {
  18. vs := strings.Split(addr,"://")
  19. network,laddr := vs[0],vs[1]
  20.  
  21. if l,err := net.Listen(network,laddr); err != nil {
  22. return nil,err
  23. } else {
  24. v.listeners = append(v.listeners,l.(*net.TCPListener))
  25. }
  26. }
  27.  
  28. v.wait.Add(len(v.listeners))
  29. for i,l := range v.listeners {
  30. addr := v.addrs[i]
  31. go func(l *net.TCPListener,addr string) {
  32. defer v.wait.Done()
  33. for {
  34. var conn *net.TCPConn
  35. if conn,err = l.AcceptTCP(); err != nil {
  36. return
  37. }
  38.  
  39. select {
  40. case v.conns <- conn:
  41. case c := <-v.closing:
  42. v.closing <- c
  43. conn.Close()
  44. }
  45. }
  46. }(l,addr)
  47. }
  48.  
  49. return
  50. }
  51.  
  52. func (v *TcpListeners) AcceptTCP() (c *net.TCPConn,err error) {
  53. var ok bool
  54. if c,ok = <- v.conns; !ok {
  55. return nil,ListenerDisposed
  56. }
  57. return
  58. }
  59.  
  60. func (v *TcpListeners) Close() (err error) {
  61. // unblock all listener internal goroutines
  62. v.closing <- true
  63.  
  64. // interrupt all listeners.
  65. for _,v := range v.listeners {
  66. if r := v.Close(); r != nil {
  67. err = r
  68. }
  69. }
  70.  
  71. // wait for all listener internal goroutines to quit.
  72. v.wait.Wait()
  73.  
  74. // clear the closing signal.
  75. _ = <-v.closing
  76.  
  77. // close channels to unblock the user goroutine to AcceptTCP()
  78. close(v.conns)
  79.  
  80. return
  81. }

这样在关闭Listener时,不会导致ListenTCP的goroutine写入closed的chan而导致错误

Closed Chan

如果closed后,chan有数据,ok还是true的,直到chan没有数据了才false。

  1. c := make(chan int,ok := <- c // v=10,ok=true,虽然c关闭了,但是有数据,ok依然是true
  2. v,ok <- c // v=0,ok=false,读失败了。

猜你在找的Go相关文章