当前位置:首页 >> 脚本专栏

golang中的select关键字用法总结

1.官方解释

一个select语句用来选择哪个case中的发送或接收操作可以被立即执行。它类似于switch语句,但是它的case涉及到channel有关的I/O操作。即select就是用来监听和channel有关的IO操作,当 IO 操作发生时,触发相应的动作。

2.要点

如果有一个或多个IO操作可以完成,则Go运行时系统会随机的选择一个执行,否则的话,如果有default分支,则执行default分支语句,如果连default都没有,则select语句会一直阻塞,直到至少有一个IO操作可以进行

所有channel表达式都会被求值、所有被发送的表达式都会被求值。求值顺序:自上而下、从左到右.

3.用法

 1.使用 select 实现 timeout 机制

timeout := make (chan bool, 1)
  go func() {
    time.Sleep(1e9) // sleep one second
    timeout <- true
  }()
  select {
  case <- timeout:
    fmt.Println("timeout!")
  }

2.使用 select 语句来检测 chan 是否已经满了

ch2 := make (chan int, 1)
  ch2 <- 1
  select {
  case ch2 <- 2:
  default:
    fmt.Println("channel is full !")
  }

3. for-select

package main

import (
  "fmt"
  "time"
)

func main() {
  var errChan = make(chan int)
  //定时2s
  ticker := time.NewTicker(2 * time.Second)
  defer ticker.Stop()
  go func(a chan int) {
    //5s发一个信号
    time.Sleep(time.Second * 5)
    errChan <- 1
  }(errChan)
  LOOP:
    for {
      select {
        case <-ticker.C: {
          fmt.Println("Task still running")
        }
        case res, ok := <-errChan:
          if ok {
            fmt.Println("chan number:", res)
            break LOOP
          }
      }
    }
  fmt.Println("end!!!")
}
//输出结果:
//Task still running
//Task still running
//chan number: 1
//end!!!

附录:

select 是 golang 中的一个控制结构,类似于 switch. 每一个 case 都必须为一个通信操作,要么是发送要么是接受。
select 随机选择一个可运行的 case, 如果没有 case 可以运行,便会阻塞,直到有 case 可以运行。一个默认的字句总是可以运行的。

select {
  case communication clause :
    statement(s)
  case communication clause :
    statement(s)
  default :
    statement(s)
}

以下描述 select 语句的语法

  • 每个 case 都必须是一个通信
  • 所有 channel 表达式都会被求值
  • 所有被发送的表达式都会被求值
  • 如果任意某个通信可以执行,它就会执行;其他就会被忽略
  • 如果有多个 case 都可以运行,select 会随机公平的选出一个执行。其他不会执行。

否则

  • 如果有 default 子句,则执行该语句
  • 如果没有 default 子句,select 将阻塞,直到某个通信可以执行;channel 或者值不会被重复求值

示例

package main
import "fmt"
func fibonacci(c, quit chan int) {
  x, y := 0, 1
  for {
    select {
    case c <- x:
      x, y = y, x+y
    case <-quit:
      fmt.Println("quit")
      return
    }
  }
}
func main() {
  c := make(chan int)
  quit := make(chan int)
  // start a goroutine to print current result
  // no buffer in c and quit channel, so this code
  // would block when this goroutine try to print
  go func() {
    for i := 0; i < 10; i++ {
      fmt.Println(<-c)
    }
    quit <- 0
  }()
  fibonacci(c, quit)
}

总结