golang select case语句代码示例

雨点打透心脏的1/2处 2023-10-12 12:16 16阅读 0赞

## select介绍 ##

`select` 是 Go 语言中用于处理并发操作的控制结构，它可以在多个通信操作中选择一个可执行的操作进行处理。`select` 语句可以让程序在多个通道操作中进行非阻塞的选择，并执行相应的代码块。

`select` 语句的基本语法如下：

select {
        
    case <-ch1:
        // 执行操作1
    case data := <-ch2:
        // 执行操作2
    case ch3 <- value:
        // 执行操作3
    default:
        // 默认操作
    }

`select` 语句由多个 `case` 语句组成，每个 `case` 表示一个通信操作。可以包含发送操作、接收操作或使用 `default` 关键字定义的默认操作。

*  `<-ch1` 表示从通道 `ch1` 接收数据，当 `ch1` 有数据可接收时，对应的代码块将被执行。
 *  `data := <-ch2` 表示从通道 `ch2` 接收数据，并将接收到的值赋给变量 `data`。当 `ch2` 有数据可接收时，对应的代码块将被执行。
 *  `ch3 <- value` 表示向通道 `ch3` 发送数据，将 `value` 发送到通道 `ch3` 中。当 `ch3` 可以接收数据时，对应的代码块将被执行。
 *  `default` 关键字定义了默认操作，当没有其他的通信操作可执行时，将执行默认操作。

`select` 语句的执行规则如下：

1.  如果多个通信操作可以执行，Go 语言会随机选择一个可执行的操作来执行。
2.  如果没有任何通信操作可以执行，且存在默认操作，则执行默认操作。
3.  如果没有任何通信操作可以执行，且没有默认操作，则 `select` 语句将被阻塞，直到至少有一个通信操作可以执行。

`select` 语句的使用场景包括：

*  并发任务的监控和处理：可以通过在多个通道上等待操作完成，并执行相应的处理逻辑。
 *  超时控制：可以使用 `time.After` 结合 `select` 语句来设置超时时间，当超过指定时间时执行相应的处理操作。
 *  多路复用：可以同时监听多个通道的操作，并根据情况选择合适的操作进行处理。
 *  与 `default` 结合使用实现非阻塞操作：可以通过设置默认操作，在没有其他操作可执行时，执行默认操作，避免程序阻塞。

通过使用 `select` 语句，我们可以更加灵活地处理并发操作，实现高效的并发编程。它是 Go 语言中处理并发任务的重要工具之一。

## 代码示例 ##

以下是一个使用 Go 语言中的 `select` 语句的示例代码：

package main
    
    import (
    	"fmt"
    	"time"
    )
    
    func main() {
        
    	ch1 := make(chan int)
    	ch2 := make(chan int)
    
    	go func() {
        
    		time.Sleep(2 * time.Second)
    		ch1 <- 1
    	}()
    
    	go func() {
        
    		time.Sleep(3 * time.Second)
    		ch2 <- 2
    	}()
    
    	select {
        
    	case <-ch1:
    		fmt.Println("Received from ch1")
    	case <-ch2:
    		fmt.Println("Received from ch2")
    	case <-time.After(4 * time.Second):
    		fmt.Println("Timeout")
    	}
    }

在这个示例中，我们创建了两个整数类型的通道 `ch1` 和 `ch2`。然后，我们使用两个匿名函数分别在不同的 goroutine 中向通道发送数据。第一个匿名函数在2秒后向 `ch1` 发送数据，第二个匿名函数在3秒后向 `ch2` 发送数据。

在 `select` 语句中，我们使用 `<-ch1` 和 `<-ch2` 分别监听 `ch1` 和 `ch2` 的接收操作。`select` 语句会等待多个通道中的一个操作完成，并执行相应的代码块。当其中一个通道成功发送数据时，对应的代码块会被执行。如果在指定的时间内没有任何通道操作完成，`select` 语句会执行 `time.After` 中的操作，即超时处理。

在本示例中，由于 `ch1` 的发送操作会在2秒后完成，因此执行 `<-ch1` 对应的代码块，打印"Received from ch1"。而 `ch2` 的发送操作会在3秒后完成，因此 `<-ch2` 对应的代码块不会执行。由于超时时间设置为4秒，所以最终会执行 `<-time.After(4 * time.Second)` 对应的代码块，打印"Timeout"。

通过使用 `select` 语句，我们可以在多个通道之间进行非阻塞的操作选择，实现对并发事件的监控和处理。这对于处理并发任务、超时控制和通信等场景非常有用。