Go 超时

超时对那些连接外部资源的程序来说是很重要的，否则就需要限定执行时间。在Go里面实现超时很简单。我们可以使用channel和select很容易地做到。

  1. package main
  3. import "time"
  4. import "fmt"
  6. func main() {
  8.     // 在这个例子中，假设我们执行了一个外部调用，2秒之后将结果写入c1
  9.     c1 := make(chan string, 1)
  10.     go func() {
  11.         time.Sleep(time.Second * 2)
  12.         c1 <- "result 1"
  13.     }()
  15.     // 这里使用select来实现超时，`res := <-c1`等待通道结果，
  16.     // `<- Time.After`则在等待1秒后返回一个值，因为select首先
  17.     // 执行那些不再阻塞的case，所以这里会执行超时程序，如果
  18.     // `res := <-c1`超过1秒没有执行的话
  19.     select {
  20.     case res := <-c1:
  21.         fmt.Println(res)
  22.     case <-time.After(time.Second * 1):
  23.         fmt.Println("timeout 1")
  24.     }
  26.     // 如果我们将超时时间设为3秒，这个时候`res := <-c2`将在
  27.     // 超时case之前执行，从而能够输出写入通道c2的值
  28.     c2 := make(chan string, 1)
  29.     go func() {
  30.         time.Sleep(time.Second * 2)
  31.         c2 <- "result 2"
  32.     }()
  33.     select {
  34.     case res := <-c2:
  35.         fmt.Println(res)
  36.     case <-time.After(time.Second * 3):
  37.         fmt.Println("timeout 2")
  38.     }
  39. }

运行结果

  1. timeout 1
  2. result 2