golang关键字select的三个例子, time.After模拟socket/心跳超时

golang关键字select的三个例子, time.After模拟socket/心跳超时

 

例子1 select会随机选择一个可执行的case

 

// 这个例子主要说明select是随机选择一个可执行的case
func main() {
  // 定义两个chan, 用于存储数据
  // 由于我打算提前把数据存进去, 所以定义缓冲区为10
  chan1 := make(chan int, 10)
  chan2 := make(chan int, 10)
  // 在两个chan中都放入10个数据
  for i := 0; i < 10; i++ {
    chan1 <- i
  }
  for i := 0; i < 10; i++ {
    chan2 <- i
  }
  // 启动一个for循环, 无限循环
  // select的作用:
  // select会随机选择一个case, 判断该case后面的语句是否可以执行
  //     如果可以执行, 进入case的代码块中, 执行代码块
  //     如果不能只能, 则随机挑选另一个case来进行判断
  // select会持续这个随机挑选判断的过程, 直到某一个case可以被执行, 并且执行它
  // 也就是说select会阻塞在这里, 直到选择了某一个case

  // 特别说明: 判断的顺序是随机的, 看看第一个, 看看第二个, 再看看第二个, 再看看第一个
  // 这个和switch-case是不一样的
  fmt.Println("即将进入select")
  fmt.Println("你会发现chan 1和chan 2是随机打印的, 没有规  ")
  fmt.Println("验证了select是随机选择")
  for {
    select {
    case num1 := <-chan1:
      {
        fmt.Printf("从chan 1中取出了数据, 这是第%d次选择case 1\n", num1)
      }
    case num2 := <-chan2:
      {
        fmt.Printf("从chan 2中取出了数据, 这是第%d次选择case 2\n", num2)
      }
    case <-time.After(time.Second * 5):
      {
        // 请忽略这个case, 仅仅是为了退出程序
        // time.After的用法将在case 2中讲解
        fmt.Println("5秒了, 程序退出")
        return
      }
    }
  }
}

 

例子2 select会阻塞, 直到某一个case可以执行 && time.After的用法

 

// 这个例子主要验证select会阻塞, 直到某一个case可以执行
// 同时也会说明time.After的用法
func main() {
  // 创建一个通道, 在for循环中会从中取值
  dataChan := make(chan int)

  // 启动一个go程
  go func() {
    // 13秒之后, 我往这个dataChan中放入一个数据
    // 这个数据是什么不重要, 重要的是我放入了一个数据
    time.Sleep(time.Second * 13)
    fmt.Println("13秒到了, 往dataChan放入一个数据")
    dataChan <- 1
  }()

  // 这个go程只是用来打印时间和退出程序的, 对测试内容没有任何影响
  go func() {
    for i := 0; ; i++ {
      fmt.Printf("%d秒\n", i+1)
      time.Sleep(time.Second * 1)
      if i == 22 {
        fmt.Println("测试结束, 程序退出")
        os.Exit(0)
      }
    }
  }()

  // 启动一个for循环, 无限循环
  for {
    // 每次循环从这里开始
    fmt.Println("-----我是select, 我开始阻塞选择啦-----")
    select {
    case <-dataChan:
      {
        // 判断是否能从dataChan这个channel中取出数据来
        // 显然前13秒是取不出来的, 因为没有人给他放数据
        fmt.Println("13秒到了, 从dataChan中取出了数据")
        fmt.Println("注意观察, 下一次select的时候, time.After会重新计时5秒")
      }
    case <-time.After(time.Second * 5):
      // time.After的作用:
      // 每次执行到select时
      // 这个time.After会开始计时
      // (如果到了)5秒后, 就会生产出一个数据, select就会认为这个case可以执行, 选择这个case
      // 如果没有到5秒, select就选择了其他case, 那么这个计时器会清零
      // 下一次循环中, 会重新进行5秒计时
      fmt.Println("5秒过去啦, time.After生产出了数据, select选择了第二个case")
    }
  }
}

 

例子3 用select来判断socket/心跳超时

 

// 这个例子是用select来判断socket/心跳超时
func main() {
  // 用这个channel来模拟socket输入, 输入数据或者心跳
  socketChan := make(chan int)
  go func() {
    // 模拟输入/心跳, 每5秒输入一个数据/接收到心跳
    for i := 0; i < 2; i++ {
      time.Sleep(time.Second * 5)
      fmt.Printf("socket: 第%d个5秒到了, 我输入一个数据\n", i+1)
      socketChan <- i
    }
    // 10秒后, socket崩溃, 不再输入数据/不再发送心跳
    time.Sleep(time.Millisecond * 1)
    fmt.Println("socket: 我崩溃啦, 没有办法继续输入数据了")
    fmt.Println("开始计算超时时间, 预计10秒后超时")
  }()

  // 这个go程只是用来打印时间和退出程序的, 对测试内容没有任何影响
  go func() {
    for i := 0; ; i++ {
      fmt.Printf("%d秒\n", i+1)
      time.Sleep(time.Second * 1)
      if i == 21 {
        fmt.Println("测试结束, 程序退出")
        os.Exit(0)
      }
    }
  }()

  // 启动一个for循环, 无限循环
  for {
    fmt.Println("-----我是select, 我开始阻塞选择啦-----")
    select {
    case num := <-socketChan:
      {
        fmt.Printf("我是select: 我收到了socket输入 是第%d次\n", num+1)
      }
    case <-time.After(time.Second * 10):
      {
        // 超时时间设定为10秒
        fmt.Println("我是select: 连续10秒都没有收到socket输入, 超时了")
        return
      }
    }
  }
}