golang中的select关键字用法总结

脚本专栏 2024/12/25 佚名

3 2 1

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)
}

总结

golang,select关键字用法,golang,select关键字,golang,select

标签：