Go语言使用select{}阻塞main函数介绍


Posted in Golang onApril 25, 2021

很多时候我们需要让main函数不退出,让它在后台一直执行,例如:

func main() {
    for i := 0; i < 20; i++ { //启动20个协程处理消息队列中的消息
        c := consumer.New()
        go c.Start()
    }
    select {} // 阻塞
}

可能大多数人想到阻塞的方法是用channel,当然都是可以的,不过用select{}更加简洁 :)

补充:由浅入深聊聊Golang中select的实现机制

正文

话说今天在玩select的时候发现一个问题,是这样的:

片段1:

func main(){
 var count int
 for {
  select {
  case <-time.Tick(time.Millisecond * 500):
   fmt.Println("咖啡色的羊驼")
   count++
   fmt.Println("count--->" , count)
  case <-time.Tick(time.Millisecond * 499) :
   fmt.Println(time.Now().Unix())
   count++
   fmt.Println("count--->" , count)
  }
 }
}

片段2:

func main(){
 t1 := time.Tick(time.Second)
 t2 := time.Tick(time.Second)
 var count int
 for {
  select {
  case <-t1:
   fmt.Println("咖啡色的羊驼")
   count++
   fmt.Println("count--->" , count)
  case <-t2 :
   fmt.Println(time.Now().Unix())
   count++
   fmt.Println("count--->" , count)
  }
 }
}

两个问题:

1.以上片段的输出结果是?

2.如何解释?

第一个问题好解决,跑一下就是,很明显输出结果肯定不同。

片段1:

1535673600
count---> 1
1535673600
count---> 2
1535673601
count---> 3

片段2:

咖啡色的羊驼
count---> 1
1535673600
count---> 2
咖啡色的羊驼
count---> 3
1535673601
count---> 4

第二个好理解,因为select监听了两个time的通道,所以交替出现。

那么第一个为何只有出现1个?

为了这个问题不得不把select的实现机制走一波,所以有了此文。

select机制简述

select有这么几个需要关注的机制

1.select+case是用于阻塞监听goroutine的,如果没有case,就单单一个select{},则为监听当前程序中的goroutine,此时注意,需要有真实的goroutine在跑,否则select{}会报panic

2.select底下有多个可执行的case,则随机执行一个。

3.select常配合for循环来监听channel有没有故事发生。需要注意的是在这个场景下,break只是退出当前select而不会退出for,需要用break TIP / goto的方式。

4.无缓冲的通道,则传值后立马close,则会在close之前阻塞,有缓冲的通道则即使close了也会继续让接收后面的值

5.同个通道多个goroutine进行关闭,可用recover panic的方式来判断通道关闭问题

看完以上知识点其实还是没法解释本文的核心疑惑,继续往下!

select机制详解

select的机制可以查看/src/runtime/select.go来了解。

源码片段解读:

func selectgo(sel *hselect) int {
 // ...
 // case洗牌
 pollslice := slice{unsafe.Pointer(sel.pollorder), int(sel.ncase), int(sel.ncase)}
 pollorder := *(*[]uint16)(unsafe.Pointer(&pollslice))
 for i := 1; i < int(sel.ncase); i++ {
  //....
 }
 // 给case排序
 lockslice := slice{unsafe.Pointer(sel.lockorder), int(sel.ncase), int(sel.ncase)}
 lockorder := *(*[]uint16)(unsafe.Pointer(&lockslice))
 for i := 0; i < int(sel.ncase); i++ {
  // ...
 }
 for i := int(sel.ncase) - 1; i >= 0; i-- {
  // ...
 }
 // 加锁该select中所有的channel
 sellock(scases, lockorder)
 // 进入loop
loop:
 // ... 
 // pass 1 - look for something already waiting
 // 按顺序遍历case来寻找可执行的case
 for i := 0; i < int(sel.ncase); i++ {
  //...
  switch cas.kind {
  case caseNil:
   continue
  case caseRecv:
   // ... goto xxx
  case caseSend:
   // ... goto xxx
  case caseDefault:
   dfli = casi
   dfl = cas
  }
 }
 // 没有找到可以执行的case,但有default条件,这个if里就会直接退出了。
 if dfl != nil {
  // ...
 }
 // ...
 // pass 2 - enqueue on all chans
 // chan入等待队列
 for _, casei := range lockorder {
  // ...
  switch cas.kind {
  case caseRecv:
   c.recvq.enqueue(sg)
  case caseSend:
   c.sendq.enqueue(sg)
  }
 }
 // wait for someone to wake us up
 // 等待被唤起,同时解锁channel(selparkcommit这里实现的)
 gp.param = nil
 gopark(selparkcommit, nil, "select", traceEvGoBlockSelect, 1)
 
 // 突然有故事发生,被唤醒,再次该select下全部channel加锁
 sellock(scases, lockorder)
 // pass 3 - dequeue from unsuccessful chans
 // 本轮最后一次循环操作,获取可执行case,其余全部出队列丢弃
 casi = -1
 cas = nil
 sglist = gp.waiting
 // Clear all elem before unlinking from gp.waiting.
 for sg1 := gp.waiting; sg1 != nil; sg1 = sg1.waitlink {
  sg1.isSelect = false
  sg1.elem = nil
  sg1.c = nil
 }
 gp.waiting = nil
 for _, casei := range lockorder {
  // ...
  if sg == sglist {
   // sg has already been dequeued by the G that woke us up.
   casi = int(casei)
   cas = k
  } else {
   c = k.c
   if k.kind == caseSend {
    c.sendq.dequeueSudoG(sglist)
   } else {
    c.recvq.dequeueSudoG(sglist)
   }
  }
  // ...
 }
 // 没有的话,再走一次loop
 if cas == nil {
  goto loop
 }
 // ...
bufrecv:
 // can receive from buffer
bufsend:
 // ...
recv:
 // ...
rclose:
 // ...
send:
 // ...
retc:
 // ...
sclose:
 // send on closed channel
}

为了方便展示,专门搞了一张很丑的图,来说明流程:

Go语言使用select{}阻塞main函数介绍

大概就是说呢,select是分四步进行的。

本文的疑惑关键点就在于那个loop的时候,当接收到发现一个可执行的时候,本次select不会执行的那些case对应的channel给出队当前goroutine,就不管他们了,就丢了,由于time.Tick是现场在case里头创建的,而不是像片段二是处于全局栈中,所以当每次任何一个执行的时候,另一个就被抛弃了,再次selelct的时候有需要重新获取,又是新的需要重头再来。

以上为个人经验,希望能给大家一个参考,也希望大家多多支持三水点靠木。如有错误或未考虑完全的地方,望不吝赐教。

Golang 相关文章推荐
基于go interface{}==nil 的几种坑及原理分析
Apr 24 Golang
golang http使用踩过的坑与填坑指南
Apr 27 Golang
golang 比较浮点数的大小方式
May 02 Golang
Golang: 内建容器的用法
May 05 Golang
go语言使用Casbin实现角色的权限控制
Jun 26 Golang
Go 语言结构实例分析
Jul 04 Golang
Go 中的空白标识符下划线
Mar 25 Golang
golang为什么要统一错误处理
Apr 03 Golang
Go归并排序算法的实现方法
Apr 06 Golang
Go调用Rust方法及外部函数接口前置
Jun 14 Golang
GoFrame基于性能测试得知grpool使用场景
Jun 21 Golang
go goth封装第三方认证库示例详解
Aug 14 Golang
win10下go mod配置方式
Go语言-为什么返回值为接口类型,却返回结构体
Apr 24 #Golang
go:垃圾回收GC触发条件详解
Apr 24 #Golang
基于go interface{}==nil 的几种坑及原理分析
Apr 24 #Golang
golang interface判断为空nil的实现代码
Apr 24 #Golang
golang判断key是否在map中的代码
Apr 24 #Golang
Go语言操作数据库及其常规操作的示例代码
Apr 21 #Golang
You might like
德劲1103二次变频版的打磨
2021/03/02 无线电
Linux下进行MYSQL编程时插入中文乱码的解决方案
2007/03/15 PHP
php结合ajax实现赞、顶、踩功能实例
2014/05/12 PHP
javascript类继承机制的原理分析
2009/09/12 Javascript
判断JS对象是否拥有某种属性的两种方式
2013/12/02 Javascript
Javascript自定义函数判断网站访问类型是PC还是移动终端
2014/01/10 Javascript
jQuery对JSON数据进行排序输出的方法
2015/06/24 Javascript
javascript中$(function() {});写与不写有哪些区别
2015/08/10 Javascript
JavaScript脚本判断蜘蛛来源的方法
2015/09/22 Javascript
jquery mobile 移动web(5)
2015/12/20 Javascript
微信小程序 location API接口详解及实例代码
2016/10/12 Javascript
解决同一页面中两个iframe互相调用jquery,js函数的方法
2016/12/12 Javascript
微信小程序手势操作之单触摸点与多触摸点
2017/03/10 Javascript
JS中DOM元素的attribute与property属性示例详解
2018/09/04 Javascript
Vue-Router的使用方法
2018/09/05 Javascript
vue实现分页栏效果
2019/06/28 Javascript
vue中的面包屑导航组件实例代码
2019/07/01 Javascript
laypage+SpringMVC实现后端分页
2019/07/27 Javascript
原生js代码能实现call和bind吗
2019/07/31 Javascript
layui 表格操作列按钮动态显示的实现方法
2019/09/06 Javascript
vue实现计算器功能
2020/02/22 Javascript
uniapp,微信小程序中使用 MQTT的问题
2020/07/11 Javascript
[00:34]DOTA2上海特级锦标赛 Spirit战队宣传片
2016/03/04 DOTA
举例讲解Python面向对象编程中类的继承
2016/06/17 Python
Python变量和字符串详解
2017/04/29 Python
Python列表list内建函数用法实例分析【insert、remove、index、pop等】
2017/07/24 Python
TensorFlow中权重的随机初始化的方法
2018/02/11 Python
Python实现二叉树前序、中序、后序及层次遍历示例代码
2019/05/18 Python
python中通过pip安装库文件时出现“EnvironmentError: [WinError 5] 拒绝访问”的问题及解决方案
2020/08/11 Python
英国排名第一的礼品体验公司:Red Letter Days
2018/08/16 全球购物
艺术设计专业个人求职信
2013/09/21 职场文书
初三班主任寄语大全
2014/04/04 职场文书
交通事故协议书
2014/04/15 职场文书
安全生产月宣传标语
2014/10/06 职场文书
销售区域经理岗位职责
2015/04/10 职场文书
投诉信格式范文
2015/07/02 职场文书