【golang】select和channel
场景1：实现并发调用多个方法，有一个返回结果就都返回
思路：可以使用 select 语句结合通道来等待多个 goroutine 返回结果，一旦有 goroutine 返回结果，就立即处理该结果并结束程序。
/**
* @desc
* @date 2025/4/19
* @user yangshuo
*/
package main
import (
"fmt"
"time"
)
func https(resultChan chan string) {
// 模拟 HTTP 请求
// t := time.NewTicker(time.Duration(rand.Intn(5)+1) * time.Second)
time.Sleep(time.Duration(4+1) * time.Second)
resultChan <- "http"
println("1")
}
func redis(resultChan chan string) {
// 模拟 HTTP 请求
// t := time.NewTicker(time.Duration(rand.Intn(5)+1) * time.Second)
// <-t.C
time.Sleep(time.Duration(5+1) * time.Second)
resultChan <- "redis"
println("2")
}
func mysql(resultChan chan string) {
// 模拟 HTTP 请求
// t := time.NewTicker(time.Duration(rand.Intn(5)+1) * time.Second)
// <-t.C
time.Sleep(time.Duration(3+1) * time.Second)
resultChan <- "mysql"
println("3")
}
func main() {
resultChan := make(chan string)
defer close(resultChan)
go https(resultChan)
go mysql(resultChan)
go redis(resultChan)
for {
select {
case result := <-resultChan:
fmt.Println("返回结果：", result)
return
}
}
}
场景2：实现并发调用多个校验方法，有一个成功就返回成功，所有方法都失败就返回失败。
思路：与场景1不同的是，失败情况需要等待所有goroutine都执行完。
/**
* @desc
* @date 2025/4/20
* @user yangshuo
*/
package main
import (
"fmt"
"math/rand"
"time"
)
func validateFunc1() bool {
// 模拟验证逻辑
// 设置随机种子
return false
rand.Seed(time.Now().UnixNano())
// 生成随机数（0 或 1）
randomNum := rand.Intn(2)
// 根据随机数返回 true 或 false
result := randomNum == 1
fmt.Println(result)
return result
}
func validateFunc2() bool {
return false
// 设置随机种子
rand.Seed(time.Now().UnixNano())
// 生成随机数（0 或 1）
randomNum := rand.Intn(2)
// 根据随机数返回 true 或 false
result := randomNum == 1
fmt.Println(result)
return result
}
func validateFunc3() bool {
return false
// 模拟验证逻辑
// 设置随机种子
rand.Seed(time.Now().UnixNano())
// 生成随机数（0 或 1）
randomNum := rand.Intn(2)
// 根据随机数返回 true 或 false
result := randomNum == 1
fmt.Println(result)
return result
}
func main() {
// 定义验证函数
validateFuncs := []func() bool{validateFunc1, validateFunc2, validateFunc3}
// 使用 channel 通知是否有函数通过验证
passed := make(chan bool)
// 并发调用验证函数
for _, vf := range validateFuncs {
go func(fn func() bool) {
if fn() {
passed <- true
}
}(vf)
}
// 检查是否有函数通过验证
for range validateFuncs {
select {
case <-passed:
fmt.Println("Validation passed")
return
}
fmt.Println("Validation failed")
}
需要注意的是：当所有 goroutines 都未通过验证时，由于没有 goroutine 向 passed 通道发送消息，select 语句会一直阻塞等待消息，从而导致程序发生死锁。为了避免这种情况，可以加上default分支。
// 检查是否有函数通过验证
for range validateFuncs {
select {
case <-passed:
fmt.Println("Validation passed")
return
default:
}
}