golang中select和超时

select作用

Go里面提供了一个关键字select,通过select可以监听channel上的数据流动。

select的用法与switch语言非常类似,由select开始一个新的选择块,每个选择条件由case语句来描述。

与switch语句可以选择任何可使用相等比较的条件相比, select有比较多的限制,其中最大的一条限制就是每个case语句里必须是一个IO操作,大致的结构如下:

select{

case<-chan1:

//如果chan1成功读到数据,则进行该case处理语句

casechan2<-1:

//如果成功向chan2写入数据,则进行该case处理语句

default:

//如果上面都没有成功,则进入default处理流程

}

在一个select语句中,Go语言会按顺序从头至尾评估每一个发送和接收的语句。

如果其中的任意一语句可以继续执行(即没有被阻塞),那么就从那些可以执行的语句中任意选择一条来使用。

如果没有任意一条语句可以执行(即所有的通道都被阻塞),那么有两种可能的情况:

l 如果给出了default语句,那么就会执行default语句,同时程序的执行会从select语句后的语句中恢复。

l 如果没有default语句,那么select语句将被阻塞,直到至少有一个通信可以进行下去。

示例代码:

funcfibonacci(c,quitchanint){

x,y:=1,1

for{

select{

casec<-x:

x,y=y,x+y

case<-quit:

fmt.Println("quit")

return

}

}

}

funcmain(){

c:=make(chanint)

quit:=make(chanint)

gofunc(){

fori:=0;i<6;i++{

fmt.Println(<-c)

}

quit<-0

}()

fibonacci(c,quit)

}

运行结果如下:

图片3.png

超时

有时候会出现goroutine阻塞的情况,那么我们如何避免整个程序进入阻塞的情况呢?我们可以利用select来设置超时,通过如下的方式实现:

funcmain(){

c:=make(chanint)

o:=make(chanbool)

gofunc(){

for{

select{

casev:=<-c:

fmt.Println(v)

case<-time.After(5*time.Second):

fmt.Println("timeout")

o<-true

break

}

}

}()

//c<-666//注释掉,引发timeout

<-o

}

代码如下所示:

// code_049_select_and_timeout project main.go

package main

import (

  "fmt"

  "time"

)

/*

1)Go里面提供了一个关键字select,通过select可以监听channel上的数据流动。

2)select的用法与switch语言非常类似,由select开始一个新的选择块,每个选择条件由case语句来描述。

3)与switch语句可以选择任何可使用相等比较的条件相比, select有比较多的限制,其中最大的一条限制就是每个case语句里必须是一个IO操作

*/

func fibonacci(c, quit chan int) {

  x, y := 1, 1

  for {

    select {

    case c <- x: //此处执行了发送,如果成功向c写入数据,则进行该case处理语句

      x, y = y, x+y

    case <-quit: //此处执行了接收, 如果quit成功读到数据,则进行该case处理语句

      //从输出结果可以看到, 此代码一直不断地向channel中写入数据。 在main函数中只是读取6个数据, 而写入的数据多于读取的数据

      fmt.Println("quit:y=", y)

      return

    }

  }

}

func main() {

  //1) select的使用

  c1 := make(chan int)

  quit := make(chan int)

  go func() {

    for i := 0; i < 6; i++ {

      fmt.Println(<-c1) //读取数据

    }

    quit <- 0

  }()

  fibonacci(c1, quit)

  //2) goroutine阻塞,利用select设置超时

  c := make(chan int)

  o := make(chan bool)

  go func() {

    for {

      select {

      case v := <-c:

        fmt.Println(v)

      case <-time.After(5 * time.Second): //返回值为channel类型,输出值为Time当前时间

        fmt.Println("timeout")

        o <- true

        break

      }

    }

  }()

  //c <-666 //注释掉,引发timeout

  <-o

}

本文来自:51CTO博客

感谢作者:ck_god

查看原文:golang中select和超时

• 编辑
• 预览