当前位置：首页 > article >正文

【go从零单排】通道select、通道timeout、Non-Blocking Channel Operations非阻塞通道操作

article 2024/11/14 21:52:59

挪威特罗姆瑟夜景

🌈Don’t worry , just coding!
内耗与overthinking只会削弱你的精力，虚度你的光阴，每天迈出一小步，回头时发现已经走了很远。

📗概念

select 语句是 Go 的一种控制结构，用于等待多个通道操作。它类似于 switch 语句，但用于通道的接收和发送

超时是指在一定时间内未能完成某项操作（如接收通道消息或发送消息），从而采取某种措施（如执行默认操作或返回错误）

非阻塞通道操作是指通过 select 语句的 default 分支，或者直接在通道操作中，尝试发送或接收数据而不会导致 goroutine 阻塞。

💻代码

通道select

package main

import (
	"fmt"
	"time"
)

func main() {
	c1 := make(chan string) // 创建一个字符串类型的通道 c1
	c2 := make(chan string) // 创建一个字符串类型的通道 c2

	// 启动第一个 goroutine
	go func() {
		time.Sleep(1 * time.Second) // 暂停 1 秒
		c1 <- "one"                 // 向 c1 通道发送消息 "one"
	}()

	// 启动第二个 goroutine
	go func() {
		time.Sleep(2 * time.Second) // 暂停 2 秒
		c2 <- "two"                 // 向 c2 通道发送消息 "two"
	}()

	// 循环接收消息
	for i := 0; i < 2; i++ {
		select {
		case msg1 := <-c1: // 从 c1 通道接收消息
			fmt.Println("received", msg1) // 打印接收到的消息
		case msg2 := <-c2: // 从 c2 通道接收消息
			fmt.Println("received", msg2) // 打印接收到的消息
		}
	}
}

//输出
//启动等待一秒后打印 received one
//启动等待2秒后打印 received two，因为两个等待是同时执行的，所以总执行时间是2秒

通道timeout

package main

import (
	"fmt"
	"time"
)

func main() {
	c1 := make(chan string, 1) // 创建一个缓冲通道 c1，容量为 1
	go func() {
		time.Sleep(2 * time.Second) // 暂停 2 秒
		c1 <- "result 1"            // 向 c1 通道发送 "result 1"
	}()

	// 第一个 select 语句
	select {
	case res := <-c1: // 尝试从 c1 接收消息
		fmt.Println(res) // 打印接收到的消息
	case <-time.After(1 * time.Second): // 如果 1 秒后还没有消息，则执行此 case
		fmt.Println("timeout 1") // 打印超时消息
	}

	c2 := make(chan string, 1) // 创建另一个缓冲通道 c2，容量为 1
	go func() {
		time.Sleep(2 * time.Second) // 暂停 2 秒
		c2 <- "result 2"            // 向 c2 通道发送 "result 2"
	}()

	// 第二个 select 语句
	select {
	case res := <-c2: // 尝试从 c2 接收消息
		fmt.Println(res) // 打印接收到的消息
	case <-time.After(3 * time.Second): // 如果 3 秒后还没有消息，则执行此 case
		fmt.Println("timeout 2") // 打印超时消息
	}
}
//输出
//timeout 1
//result 2

Non-Blocking Channel Operations非阻塞通道操作

package main

import "fmt"

func main() {
	messages := make(chan string) // 创建一个字符串类型的通道 messages
	signals := make(chan bool)    // 创建一个布尔类型的通道 signals

	// 第一个 select 语句
	select {
	case msg := <-messages: // 尝试从 messages 通道接收消息
		fmt.Println("received message", msg)
	default: // 如果没有消息可接收，则执行此分支
		fmt.Println("no message received")
	}

	msg := "hi" // 定义消息内容
	// 第二个 select 语句
	select {
	case messages <- msg: // 尝试向 messages 通道发送消息
		fmt.Println("sent message", msg)
	default: // 如果通道满或没有接收者，则执行此分支
		fmt.Println("no message sent")
	}

	// 第三个 select 语句
	select {
	case msg := <-messages: // 尝试从 messages 通道接收消息
		fmt.Println("received message", msg)
	case sig := <-signals: // 尝试从 signals 通道接收信号，这里是bool类型的通道，并没有消息发送进来
		fmt.Println("received signal", sig)
	default: // 如果没有消息或信号可接收，则执行此分支
		fmt.Println("no activity")
	}
}

//输出
//no message received
//no message sent
//no activity