go select 使用总结-热文
2023-07-06 05:47:57 博客园

转载请注明出处:

在Go语言中,select语句用于处理多个通道的并发操作。它类似于switch语句,但是select语句用于通信操作,而不是条件判断。select语句会同时监听多个通道的操作,并选择其中一个可用的通道进行操作。 select语句的语法如下:

select {    case <-channel1:        // 执行channel1的操作    case data := <-channel2:        // 执行channel2的操作,同时将通道中的数据赋值给data变量    case channel3 <- data:        // 将data写入channel3    default:        // 当没有任何通道操作时执行default语句}

select语句中可以包含多个case子句,每个case子句表示一个通道操作。<-符号用于从通道中读取数据,channel <- data用于将数据写入通道。 select语句的执行流程为:


【资料图】

如果有多个通道都可以操作,则随机选择一个进行操作。

如果没有任何通道可以操作,则会执行default语句(如果有)。

如果没有default语句,则select语句会阻塞,直到至少有一个通道可以操作。

下面是一个使用select语句的示例:

package mainimport (    "fmt"    "time")func main() {    ch1 := make(chan int)    ch2 := make(chan int)    go func() {        time.Sleep(time.Second)        ch1 <- 1        fmt.Println("ch1 ending")    }()    go func() {        time.Sleep(2 * time.Second)        ch2 <- 2        fmt.Println("ch2 ending")    }()    select {        case data := <-ch1:            fmt.Println("收到ch1的数据:", data)        case data := <-ch2:            fmt.Println("收到ch2的数据:", data)        case <-time.After(100 * time.Second):            fmt.Println("超时:没有接收到任何数据")    }}

在上面的例子中,我们创建了两个通道ch1ch2,并分别在不同的goroutine中向它们发送数据。然后使用select语句监听这两个通道的操作,当其中一个通道可用时,就会执行对应的case语句。

在这个例子中,由于ch1的数据发送操作会在1秒后执行,而ch2的数据发送操作会在2秒后执行,因此select语句会等待1秒后,执行ch1case语句,输出"收到ch1的数据: 1"。如果我们将ch1的发送操作改为在2秒后执行,那么select语句将会等待2秒后,执行ch2case语句,输出"收到ch2的数据: 2"。

需要注意的是,select语句的执行顺序是随机的,所以不能依赖于某个通道的操作先于其他通道。这也是使用select语句时需要注意的地方之一。

如果希望持续监听多个通道的操作,可以将select语句放在一个无限循环中。

package mainimport (    "fmt"    "time")func main() {    channel1 := make(chan int)    channel2 := make(chan string)    go func() {        for i := 0; i < 5; i++ {            time.Sleep(time.Second)            channel1 <- i        }        close(channel1)    }()    go func() {        for i := 0; i < 5; i++ {            time.Sleep(time.Second)            channel2 <- fmt.Sprintf("Message %d", i)        }        close(channel2)    }()    for {        select {        case data1, ok := <-channel1:            if ok {                fmt.Println("Received from channel1:", data1)            } else {                fmt.Println("Channel1 closed")                channel1 = nil            }        case data2, ok := <-channel2:            if ok {                fmt.Println("Received from channel2:", data2)            } else {                fmt.Println("Channel2 closed")                channel2 = nil            }        }        if channel1 == nil && channel2 == nil {            break        }    }    fmt.Println("Done")}

在这个示例中,创建了两个通道channel1channel2,分别用于发送不同类型的数据。然后分别启动两个goroutine,每个goroutine向对应的通道发送一些数据,然后关闭通道。其运行得结果如图:

上述代码中ok是从通道的属性中获取的。在Go语言中,当从通道接收数据时,会返回两个值:接收到的数据和一个表示通道是否已关闭的布尔值。这个布尔值就是ok。 当通道已关闭且没有数据可读取时,会返回通道元素类型的零值和false。当通道还未关闭且有数据可读取时,会返回通道中的数据和true因此,使用data, ok := <-channel的语法可以同时接收通道中的数据和判断通道是否已关闭。data表示接收到的数据,ok表示通道是否还有数据可读取。如果okfalse,则表示通道已关闭,没有数据可读取。

在主函数中,我们使用无限循环和select语句来持续监听这两个通道的操作。每次循环时,select语句会选择其中一个可用的通道进行操作。如果通道关闭,我们会将对应的通道设置为nil,以便在后续的循环中跳过该通道的操作。当两个通道都关闭,即channel1channel2都为nil时,我们跳出循环,程序结束。

运行上述代码,你会看到程序持续监听两个通道的操作,并打印接收到的数据,直到两个通道都关闭。最后,程序输出"Done"并结束。

相关新闻: