探索Go中的Select语句

运维开发王义杰

发布于 2023-08-10 18:14:00

1420

发布于 2023-08-10 18:14:00

文章被收录于专栏：运维开发王义杰运维开发王义杰

大家好，欢迎再次回到我的Go语言专栏。今天我们将探索Go中的一个非常强大的并发特性：Select语句。

Select语句使我们能够在多个不同的Channel上进行等待。这是非常有用的，因为它为我们提供了一种方式来同时管理多个Channel。

1. Select语句基础

Select语句的基本语法如下：

select {
    case sendChan <- value:
        // 发送操作
    case x = <-receiveChan:
        // 接收操作
    case <-time.After(time.Second * 1):
        // 超时操作
    default:
        // 默认操作
}

Select语句的工作原理是，它会等待case中的任何一条语句能够执行，然后执行那条语句。如果有多个case同时满足条件，则随机选择一个执行。

2. 使用Select语句进行非阻塞的读/写操作

使用default语句，我们可以进行非阻塞的读或者写操作。如果所有的Channel都不能立即进行读或者写操作，那么default case将被执行。

select {
    case sendChan <- value:
        fmt.Println("wrote value to sendChan")
    case x = <-receiveChan:
        fmt.Println("received value from receiveChan")
    default:
        fmt.Println("no value ready to receive or send")
}

3. 使用Select语句进行超时操作

我们还可以使用Select语句进行超时操作，这可以通过使用time.After函数实现。

select {
    case res := <-responseChan:
        fmt.Println(res)
    case <-time.After(time.Second * 1):
        fmt.Println("request timed out")
}

在上面的示例中，如果在1秒内没有从responseChan接收到数据，那么超时case将被执行，程序将打印出"request timed out"。

4. Select语句实例

以下是一个简单的示例，说明如何使用Select语句：

package main

import (
  "fmt"
  "time"
)

func server1(ch chan string) {
  time.Sleep(time.Second * 1)
  ch <- "from server1"
}

func server2(ch chan string) {
  time.Sleep(time.Second * 2)
  ch <- "from server2"
}

func main() {
  output1 := make(chan string)
  output2 := make(chan string)

  go server1(output1)
  go server2(output2)

  select {
  case s1 := <-output1:
    fmt.Println(s1)
  case s2 := <-output2:
    fmt.Println(s2)
  }
}

在这个示例中，我们有两个服务器，每个服务器都在其自己的Goroutine中运行，并在完成处理后向其自己的Channel发送数据。主函数中的select语句等待两个服务器中的任何一个完成其处理。

这就是Go中Select语句的基本使用。在下一篇文章中，我们将探讨Go语言中的Mutex以及如何使用它来避免竞争条件。敬请期待！

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2023-07-17，如有侵权请联系 cloudcommunity@tencent.com 删除