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

Go中select用法(go for select)

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

什么是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

相关推荐

每天一个编程技巧!掌握这7个神技,代码效率飙升200%

“同事6点下班,你却为改BUG加班到凌晨?不是你不努力,而是没掌握‘偷懒’的艺术!本文揭秘谷歌工程师私藏的7个编程神技,每天1分钟,让你的代码从‘能用’变‘逆天’。文末附《Python高效代码模板》,...

Git重置到某个历史节点(Sourcetree工具)

前言Sourcetree回滚提交和重置当前分支到此次提交的区别?回滚提交是指将改动的代码提交到本地仓库,但未推送到远端仓库的时候。...

git工作区、暂存区、本地仓库、远程仓库的区别和联系

很多程序员天天写代码,提交代码,拉取代码,对git操作非常熟练,但是对git的原理并不甚了解,借助豆包AI,写个文章总结一下。Git的四个核心区域(工作区、暂存区、本地仓库、远程仓库)是版本控制的核...

解锁人生新剧本的密钥:学会让往事退场

开篇:敦煌莫高窟的千年启示在莫高窟321窟的《降魔变》壁画前,讲解员指着斑驳色彩说:"画师刻意保留了历代修补痕迹,因为真正的传承不是定格,而是流动。"就像我们的人生剧本,精彩章节永远...

Reset local repository branch to be just like remote repository HEAD

技术背景在使用Git进行版本控制时,有时会遇到本地分支与远程分支不一致的情况。可能是因为误操作、多人协作时远程分支被更新等原因。这时就需要将本地分支重置为与远程分支的...

Git恢复至之前版本(git恢复到pull之前的版本)

让程序回到提交前的样子:两种解决方法:回退(reset)、反做(revert)方法一:gitreset...

如何将文件重置或回退到特定版本(怎么让文件回到初始状态)

技术背景在使用Git进行版本控制时,经常会遇到需要将文件回退到特定版本的情况。可能是因为当前版本出现了错误,或者想要恢复到之前某个稳定的版本。Git提供了多种方式来实现这一需求。...

git如何正确回滚代码(git命令回滚代码)

方法一,删除远程分支再提交①首先两步保证当前工作区是干净的,并且和远程分支代码一致$gitcocurrentBranch$gitpullorigincurrentBranch$gi...

[git]撤销的相关命令:reset、revert、checkout

基本概念如果不清晰上面的四个概念,请查看廖老师的git教程这里我多说几句:最开始我使用git的时候,我并不明白我为什么写完代码要用git的一些列指令把我的修改存起来。后来用多了,也就明白了为什么。gi...

利用shell脚本将Mysql错误日志保存到数据库中

说明:利用shell脚本将MYSQL的错误日志提取并保存到数据库中步骤:1)创建数据库,创建表CreatedatabaseMysqlCenter;UseMysqlCenter;CREATET...

MySQL 9.3 引入增强的JavaScript支持

MySQL,这一广泛采用的开源关系型数据库管理系统(RDBMS),发布了其9.x系列的第三个更新版本——9.3版,带来了多项新功能。...

python 连接 mysql 数据库(python连接MySQL数据库案例)

用PyMySQL包来连接Python和MySQL。在使用前需要先通过pip来安装PyMySQL包:在windows系统中打开cmd,输入pipinstallPyMySQL ...

mysql导入导出命令(mysql 导入命令)

mysql导入导出命令mysqldump命令的输入是在bin目录下.1.导出整个数据库  mysqldump-u用户名-p数据库名>导出的文件名  mysqldump-uw...

MySQL-SQL介绍(mysql sqlyog)

介绍结构化查询语言是高级的非过程化编程语言,允许用户在高层数据结构上工作。它不要求用户指定对数据的存放方法,也不需要用户了解具体的数据存放方式,所以具有完全不同底层结构的不同数据库系统,可以使用相同...

MySQL 误删除数据恢复全攻略:基于 Binlog 的实战指南

在MySQL的世界里,二进制日志(Binlog)就是我们的"时光机"。它默默记录着数据库的每一个重要变更,就像一位忠实的史官,为我们在数据灾难中提供最后的救命稻草。本文将带您深入掌握如...