深入理解Go语言time.AfterFunc函数及其潜在阻塞问题
本文将详细讲解time.AfterFunc函数的用法，并重点分析其常见误用场景及解决方法。time.AfterFunc函数用于在指定时间后执行一个函数，这在Go语言的并发编程中非常实用。然而，直接依赖其返回的Timer的C通道来等待函数执行完成，并非总是可靠的，甚至可能导致程序阻塞。
让我们先看一段容易产生阻塞的代码：
import (
"fmt"
"time"
)
func main() {
t := time.AfterFunc(3 * time.Second, func() {
fmt.Println("done")
})
// time.Sleep(4 * time.Second) // 避免阻塞的错误方法
for {
select {
// 此处t.C通道在该场景下为nil，导致程序阻塞
case <-t.C:
fmt.Println("ok")
}
}
}
这段代码意图在3秒后打印"done"，然后打印"ok"。然而，它会发生阻塞，无法按预期顺序打印。这是因为time.AfterFunc返回的Timer的C通道在此场景下是nil。time.AfterFunc本身并不提供直接的机制来同步等待回调函数的执行完成。
立即学习“go语言免费学习笔记（深入）”；
为了实现"先打印'done'，再打印'ok'"的效果，并避免在主线程使用time.Sleep，我们需要借助Go语言的通道机制进行同步。改进后的代码如下：
package main
import (
"fmt"
"time"
)
func main() {
c := make(chan struct{}) // 使用无缓冲通道进行同步
time.AfterFunc(3*time.Second, func() {
fmt.Println("done")
close(c) // 关闭通道，通知主 goroutine
})
<-c // 阻塞等待通道关闭
fmt.Println("ok")
}
在这个改进版本中，我们创建了一个无缓冲通道c。在time.AfterFunc的回调函数中，我们打印"done"后，关闭通道c来通知主goroutine。主goroutine通过<-c阻塞等待通道c关闭，从而确保"done"打印完成后再打印"ok"，避免了阻塞问题，并且没有使用time.Sleep。 这种方法更优雅且高效地解决了同步问题。
