golang select 机制和超时

简介: golang select 机制和超时

golang 中的协程使用非常方便,但是协程什么时候结束是一个控制问题,可以用 select 配合使用。

子协程和父协程的通信通常用 context 或者 chan。我遇到一个通常的使用场景,在子协程中尝试多次处理,父协程等待一段时间超时,我选择用 chan 实现。我以为 select 和 C++ 中 switch 类似,所以最开始代码类似如下:


for {
    select {
        case <-ctx.Done():
            // process ctx done
        case <-time.After(time.Second * 3):
            // process after
        default:
            // process code
    }
}


测试发现无法实现 timeout,又仔细查看文档,才发现 golang 中 select 另有玄机。废话少说,直接总结要点:

  • select 中的 case 必须是进行 chan 的手法操作,也就是只能在 case 中操作 chan,并且是非阻塞接收
  • select 中的 case 是同时监听的,多个 case 同时操作,并未 switch 中一个个顺序判断。如果多个 case 满足要求,随机执行一个,如果一个没有则阻塞当前的协程(没有 default 情况下)。很类似 Linux 文件符操作的 select 语义
  • 上面说的阻塞是没有 default 的情况下,如果有 default,则执行 default,然后退出 select,也就是不会阻塞当前协程。

回到上述代码,我这个 select 会一直不断的执行 default,time.After 生成的 chan 并不会被阻塞判断,所以根本无法完成我想要的效果。理解了之后重新修改代码:


done := make(char int)
go func(c chan int) {
    for {
        // process code
        if {
            c <- 1
            return
        }
    }
    c <- 0
}(done)
select {
    case <-ctx.Done():
        // process ctx done
    case <-time.After(time.Second * 3):
        // process after
    case <-done:
        // process code
}


开一个新的协程去不断尝试,在外的三个 case 有一个满足,则会执行。但是这里有一个问题非常需要注意:子协程什么时候退出?

因为 gorountine 不能被强制 kill,所以在上述超时的情况下,select 语句执行 case time.After 之后退出,done 这个 chan 已经没有接受方了,因此既没有接受者,又没有缓冲区,结合 chan 的特性,则子协程会一直阻塞无法退出,所以本质上这个实现会导致子协程累积下去,也就是协程泄露,可能会使资源耗尽。

如何避免上述问题呢?一个很简单的想法就是提供缓冲区,done := make(char int, 1),这样即使没有接收方,子协程也能完成发送,不会被阻塞。

还要一种办法,上面说了,select 操作 chan,并且可以指定 default,那是不是有思路了呢?


if {
    select {
        case done <- 1:
        default:
            return
    }
}


我们尝试往 chan 中发送,如果发不出去,则就退出,也实现了目的。

最后总结一下,goroutine 泄露的防范条例:

  • 创建 goroutine 时就要想好该 goroutine 该如何结束。
  • 使用 chan 时,要考虑到 chan 阻塞时协程可能的行为。
  • 实现循环语句时注意循环的退出条件,避免死循环。
目录
相关文章
|
6月前
|
存储 Go
Golang底层原理剖析之slice类型与扩容机制
Golang底层原理剖析之slice类型与扩容机制
72 0
|
6月前
|
程序员 Go
Golang深入浅出之-Select语句在Go并发编程中的应用
【4月更文挑战第23天】Go语言中的`select`语句是并发编程的关键,用于协调多个通道的读写。它会阻塞直到某个通道操作可行,执行对应的代码块。常见问题包括忘记初始化通道、死锁和忽视`default`分支。要解决这些问题,需确保通道初始化、避免死锁并添加`default`分支以处理无数据可用的情况。理解并妥善处理这些问题能帮助编写更高效、健壮的并发程序。结合使用`context.Context`和定时器等工具,可提升`select`的灵活性和可控性。
101 2
|
2月前
|
安全 Go
Golang语言goroutine协程并发安全及锁机制
这篇文章是关于Go语言中多协程操作同一数据问题、互斥锁Mutex和读写互斥锁RWMutex的详细介绍及使用案例,涵盖了如何使用这些同步原语来解决并发访问共享资源时的数据安全问题。
80 4
|
2月前
|
Go
Golang语言错误处理机制
这篇文章是关于Golang语言错误处理机制的教程,介绍了使用defer结合recover捕获错误、基于errors.New自定义错误以及使用panic抛出自定义错误的方法。
45 3
|
3月前
|
监控 Go
|
3月前
|
Go 开发者
|
3月前
|
Go 开发者
Golang 中的异常处理机制详解
【8月更文挑战第31天】
47 0
|
5月前
|
监控 Go
博客园 ☜ golang select 的 case 执行顺序
Go 语言的 `select` 语句用于等待多个通道操作就绪。当非阻塞的 chan1, chan2, chan3 同时可读时,`select` 会随机选择一个执行,之后的循环中其他未选中的 case 仍有执行机会。如果所有 case 都未准备好,将执行 default case。
|
5月前
|
Go
【golang】使用select {}
【golang】使用select {}
43 0
|
6月前
|
负载均衡 算法 Go
Golang深入浅出之-Go语言中的服务注册与发现机制
【5月更文挑战第4天】本文探讨了Go语言中服务注册与发现的关键原理和实践,包括服务注册、心跳机制、一致性问题和负载均衡策略。示例代码演示了使用Consul进行服务注册和客户端发现服务的实现。在实际应用中,需要解决心跳失效、注册信息一致性和服务负载均衡等问题,以确保微服务架构的稳定性和效率。
145 3