【Golang】高性能编程之超时退出协程

2021/12/14 17:18:11

本文主要是介绍【Golang】高性能编程之超时退出协程,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

  • 超时控制在网络编程中是非常常见的,利用 context.WithTimeouttime.After都能够很轻易地实现。
func doBadthing(done chan bool) {
	time.Sleep(time.Second)
	done <- true
}

func timeout(f func(chan bool)) error {
	done := make(chan bool)
	go f(done)
	select {
	case <-done:
		fmt.Println("done")
		return nil
	case <-time.After(time.Millisecond):
		return fmt.Errorf("timeout")
	}
}

// timeout(doBadthing)

对于这段代码,会出现超时,子协程是不能够全部正常退出的。参见如何退出协程 goroutine 超时场景

  • 它的问题是,它使用的是无缓冲的channel,当select执行到超时,doBadthing发送给done的信息没有接收方,因为进到了超时分支,所以,发送方doBadthing会一直阻塞,导致协程不能退出。
    那如果在实际的业务中,我们使用了上述的代码,那越来越多的协程会残留在程序中,最终会导致内存耗尽(每个协程约占 2K 空间),程序崩溃。
  • 如何避免呢?
    1.创建有缓冲区的channel,使得即使没有接收方,发送方也不会发生阻塞。
    2.使用select尝试发送
func doGoodthing(done chan bool) {
	time.Sleep(time.Second)
	select {
	case done <- true:
	default:
		return
	}
}

如果发送失败则,会直接退出。
3.拆分任务,分为多段,配合select使用,判断哪一段超时。

  • 强制Kill goroutine
    goroutine 只能自己退出,而不能被其他 goroutine 强制关闭或杀死。
    goroutine 被设计为不可以从外部无条件地结束掉,只能通过 channel 来与它通信。也就是说,每一个 goroutine 都需要承担自己退出的责任。(A goroutine cannot be programmatically killed. It can only commit a cooperative suicide.)
  • 建议
    因为 goroutine 不能被强制 kill,在超时或其他类似的场景下,为了 goroutine 尽可能正常退出,建议如下:
    1.尽量使用非阻塞 I/O(非阻塞 I/O 常用来实现高性能的网络库),阻塞 I/O 很可能导致 goroutine 在某个调用一直等待,而无法正确结束。
    2.业务逻辑总是考虑退出机制,避免死循环
    3.任务分段执行超时后即时退出,避免 goroutine 无用的执行过多,浪费资源。

参考

如何退出协程 goroutine 超时场景



这篇关于【Golang】高性能编程之超时退出协程的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程