好好爱自己!

golang中select case 的用途到底是啥

https://nanxiao.gitbooks.io/golang-101-hacks/content/posts/select-operation.html

---------------------------------------------------------------------------------

Select operation


Go's select operation looks similar to switch, but it's dedicated to poll send and receive operations channels. Check the following example:

package main

import (
        "fmt"
        "time"
)

func main() {
        ch1 := make(chan int)
        ch2 := make(chan int)

        go func(ch chan int) { <-ch }(ch1)
        go func(ch chan int) { ch <- 2 }(ch2)

        time.Sleep(time.Second)

        for {
                select {
                case ch1 <- 1:
                        fmt.Println("Send operation on ch1 works!")
                case <-ch2:
                        fmt.Println("Receive operation on ch2 works!")
                default:
                        fmt.Println("Exit now!")
                        return
                }
        }
}

The running result is like this:

Send operation on ch1 works!
Receive operation on ch2 works!
Exit now!

The select operation will check which case branch can be run, that means the send or receive action can be executed successfully. If more than one case are ready now, the select will randomly choose one to execute. If no case is ready, but there is a default branch, then the default block will be executed, else the select operation will block. In the above example, if the main goroutine doesn't sleep (time.Sleep(time.Second)), the other 2 func goroutines won't obtain the opportunity to run, so only default block in select statement will be executed.

The select statement won't process nil channel, so if a channel used for receive operation is closed, you should mark its value as nil, then it will be kicked out of the selection list. So a common pattern of selection on multiple receive channels looks like this:

for ch1 != nil && ch2 != nil {
    select {
    case x, ok := <-ch1:
        if !ok {
            ch1 = nil
            break
        }
        ......
    case x, ok := <-ch2:
        if !ok {
            ch2 = nil
            break
        }
        ......
    }
}  
posted @   立志做一个好的程序员  阅读(4557)  评论(0编辑  收藏  举报
编辑推荐:
· AI与.NET技术实操系列:基于图像分类模型对图像进行分类
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· 25岁的心里话
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现

不断学习创作,与自己快乐相处

点击右上角即可分享
微信分享提示