go select 使用总结

转载请注明出处：

　　在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 main

import (
    "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("超时：没有接收到任何数据")
    }
}

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

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

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

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

package main
import (
    "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")
}

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

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

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

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

posted @ 2023-07-05 23:15 香吧香阅读(307) 评论(0) 编辑收藏举报

刷新页面返回顶部

香吧香

go select 使用总结

公告