百度360必应搜狗淘宝本站头条
当前位置:网站首页 > IT技术 > 正文

Go中select用法(go for select)

wptr33 2025-03-25 18:09 26 浏览

什么是select

select 语句用于从多个发送/接收通道操作中进行选择。select 语句将一直阻塞,直到其中一个发送/接收操作准备就绪。如果多个操作准备就绪,则随机选择其中一个。语法类似于switch,不同之处在于每个case语句都是一个通道操作。让我们深入了解一些代码以更好地理解。

例子

package main

import (  
    "fmt"
    "time"
)

func server1(ch chan string) {  
    time.Sleep(6 * time.Second)
    ch <- "from server1"
}
func server2(ch chan string) {  
    time.Sleep(3 * time.Second)
    ch <- "from server2"

}
func main() {  
    output1 := make(chan string)
    output2 := make(chan string)
    go server1(output1)
    go server2(output2)
    select {
    case s1 := <-output1:
        fmt.Println(s1)
    case s2 := <-output2:
        fmt.Println(s2)
    }
}

在上面的程序中,server1 函数休眠6秒,然后将文本"from server1"写入通道ch。server2 函数休眠3秒钟,然后"from server2"写入通道ch。main 函数调用 go Goroutines server1 和 server2。

程序运行到达 select 语句,select 语句将阻塞,直到其中一个case准备就绪。在上面的程序中,server1 Goroutine在6秒后写入output1 channel,而server2在3秒后写入output2 channel。因此,select 语句将阻塞3秒,并等待 server2 Goroutine 写入output2 channle。3秒后,程序打印:

from server2 

之后程序中断。

select的实际用法

将上述程序中的函数命名为 server1 和 server2 的原因是为了说明 select 的实际用法。

假设我们有一个任务关键型应用程序,我们需要尽快将输出返回给用户。此应用程序的数据库被复制并存储在世界各地的不同服务器中。假设函数server1和server2实际上正在与2个这样的服务器通信,每个服务器的响应时间取决于每个服务器上的负载和网络延迟,我们将请求发送到两个服务器,然后使用 select 语句在相应的channel上等待响应。首先响应的服务器由select语句选择,另外的响应被忽略。这样,我们可以将相同的请求发送到多个服务器,并将最快的响应返回给用户。

select语句中的default case

当其他事例均未准备就绪时,将执行 select 语句中的default case。这通常用于防止 select 语句阻塞。

package main

import (  
    "fmt"
    "time"
)

func process(ch chan string) {  
    time.Sleep(10500 * time.Millisecond)
    ch <- "process successful"
}

func main() {  
    ch := make(chan string)
    go process(ch)
    for {
        time.Sleep(1000 * time.Millisecond)
        select {
        case v := <-ch:
            fmt.Println("received value: ", v)
            return
        default:
            fmt.Println("no value received")
        }
    }
}

在上面的程序中,process函数休眠10500毫秒(10.5 秒),然后将"process successful"写入 ch 通道。此函数在程序的main函数中被调用。同时调用Goroutine 后,在主 Goroutine 中启动一个无限的 for 循环。无限循环在每次迭代开始时休眠1000毫秒(1秒),然后执行select语句,在前10500毫秒内,select 语句的第一种情况,即 case v := <-ch,将不会准备就绪,因为进程 Goroutine 只会在10500毫秒后写入 ch 通道。因此,default case将在此期间执行,程序将打印10次"no value received"。在10.5 秒后,Goroutine 将"process successful"写入ch,将执行select语句的第一个情况,程序将打印"received value: process successful",然后它将终止。该程序将输出,

no value received  
no value received  
no value received  
no value received  
no value received  
no value received  
no value received  
no value received  
no value received  
no value received  
received value:  process successful  

死锁和default case

package main

func main() {  
    ch := make(chan string)
    select {
    case <-ch:
    }
}

在上面的程序中,创建了一个ch channel,尝试从select中的ch channel读取。select 语句将永远阻塞,因为没有其他 Goroutine 写入此通道,因此将导致死锁。该程序将在运行时死机并显示以下消息:

fatal error: all goroutines are asleep - deadlock!

goroutine 1 [chan receive]:  
main.main()  
    /tmp/sandbox627739431/prog.go:6 +0x4d

如果存在default case,则不会发生此死锁,因为default case将在没有其他case准备就绪时执行。上面的程序被重写如下:

package main

import "fmt"

func main() {  
    ch := make(chan string)
    select {
    case <-ch:
    default:
        fmt.Println("default case executed")
    }
}

上面的程序将打印:

default case executed  

同样地,如果select语句仅仅有nil channel,default case也会执行。

package main

import "fmt"

func main() {  
    var ch chan string
    select {
    case v := <-ch:
        fmt.Println("received value", v)
    default:
        fmt.Println("default case executed")

    }
}

上面的程序中,ch channel是nil,但是select语句中尝试从ch channel中读取值,如果select中没有default case,程序将一直阻塞并且导致死锁。因此当有nil channel时,在select语句中加上default case,此程序将输出如下:

default case executed 

随机选择

当在select语句中有多个case都准备好了,则只能随机其中的一个。

package main

import (  
    "fmt"
    "time"
)

func server1(ch chan string) {  
    ch <- "from server1"
}
func server2(ch chan string) {  
    ch <- "from server2"

}
func main() {  
    output1 := make(chan string)
    output2 := make(chan string)
    go server1(output1)
    go server2(output2)
    time.Sleep(1 * time.Second)
    select {
    case s1 := <-output1:
        fmt.Println(s1)
    case s2 := <-output2:
        fmt.Println(s2)
    }
}

上面的程序中,server1和server2 goroutines分别被调用,当主程序sleep 1秒后,output1和output2将分别写入字符,因此select语句中的两个case都准备就绪,这时将随机选择一个case执行。你可以本地执行验证它的随机性。

陷阱--空select

package main

func main() {  
    select {}
}

你认为上面的程序将输出什么?

我们都知道select语句会一直阻塞直到它的case执行。在这个例子中,select语句中没有任何case,因为它会一直阻塞,直至产生死锁。这个程序将会panic并由如下的输出:

fatal error: all goroutines are asleep - deadlock!

goroutine 1 [select (no cases)]:  
main.main()  
    /tmp/sandbox246983342/prog.go:4 +0x25

相关推荐

什么是Java中的继承?如何实现继承?

什么是继承?...

Java 继承与多态:从基础到实战的深度解析

在面向对象编程(OOP)的三大支柱中,继承与多态是构建灵活、可复用代码的核心。无论是日常开发还是框架设计,这两个概念都扮演着至关重要的角色。本文将从基础概念出发,结合实例与图解,带你彻底搞懂Java...

Java基础教程:Java继承概述_java的继承

继承概述假如我们要定义如下类:学生类,老师类和工人类,分析如下。学生类属性:姓名,年龄行为:吃饭,睡觉老师类属性:姓名,年龄,薪水行为:吃饭,睡觉,教书班主任属性:姓名,年龄,薪水行为:吃饭,睡觉,管...

java4个技巧:从继承和覆盖,到最终的类和方法

日复一日,我们编写的大多数Java只使用了该语言全套功能的一小部分。我们实例化的每个流以及我们在实例变量前面加上的每个@Autowired注解都足以完成我们的大部分目标。然而,有些时候,我们必须求助于...

java:举例说明继承的概念_java继承的理解

在现实生活中,继承一般指的是子女继承父辈的财产。在程序中,继承描述的是事物之间的所属关系,通过继承可以使多种事物之间形成一种关系体系。例如猫和狗都属于动物,程序中便可以描述为猫和狗继承自动物,同理,...

从零开始构建一款开源的 Vibe Coding 产品 Week1Day4:业界调研之 Agent 横向对比

前情回顾前面两天我们重点调研了了一下Cursor的原理和Cursor中一个关键的工具edit_file的实现,但是其他CodingAgent也需要稍微摸一下底,看看有没有优秀之处,下...

学会这几个插件,让你的Notepad++使用起来更丝滑

搞程序开发的小伙伴相信对Notepad++都不会陌生,是一个占用空间少、打开启动快的文件编辑器,很多程序员喜欢使用Notepad++进行纯文本编辑或者脚本开发,但是Notepad++的功能绝不止于此,...

将 node_modules 目录放入 Git 仓库的优点

推荐一篇文章Whyyoushouldcheck-inyournodedependencies[1]...

再度加码AI编程,腾讯发布AI CLI并宣布CodeBuddy IDE开启公测

“再熬一年,90%的程序员可能再也用不着写for循环。”凌晨两点半,王工还在公司敲键盘。他手里那份需求文档写了足足六页,产品经理反复改了三次。放在过去,光数据库建表、接口对接、单元测试就得写两三天。现...

git 如何查看stash的内容_git查看ssh key

1.查看Stash列表首先,使用gitstashlist查看所有已保存的stash:...

6万星+ Git命令懒人必备!lazygit 终端UI神器,效率翻倍超顺手!

项目概览lazygit是一个基于终端的Git命令可视化工具,通过简易的TUI(文本用户界面)提升Git操作效率。开发者无需记忆复杂命令,即可完成分支管理、提交、合并等操作。...

《Gemini CLI 实战系列》(一)Gemini CLI 入门:AI 上命令行的第一步

谷歌的Gemini模型最近热度很高,而它的...

deepin IDE新版发布:支持玲珑构建、增强AI智能化

IT之家8月7日消息,深度操作系统官方公众号昨日(8月6日)发布博文,更新推出新版deepin集成开发环境(IDE),重点支持玲珑构建。支持玲珑构建deepinIDE在本次重磅更...

狂揽82.7k的star,这款开源可视化神器,轻松创建流程图和图表

再不用Mermaid,你的技术文档可能已经在悄悄“腐烂”——图表版本对不上、同事改完没同步、评审会上被一句“这图哪来的”问得哑口无言。这不是危言耸听。GitHub2025年开发者报告显示,63%的新仓...

《Gemini CLI 实战系列》(五)打造专属命令行工具箱

在前几篇文章中,我们介绍了GeminiCLI的基础用法、效率提升、文件处理和与外部工具结合。今天我们进入第五篇...