Go语言中for循环内初始化的time.NewTimer如何正确停止？

我有一段如下所示的Go语言程序：

package main

import (
    "fmt"
    "time"
)

func main() {
    ch := make(chan string)
    go endProgram(ch)
    printFunc(ch)
}

func printFunc(ch chan string) {
    for {
        timeout := time.NewTimer(getTimeoutDuration())
        defer timeout.Stop()
        select {
        case s := <-ch:
            fmt.Println(s)
            return
        case <-timeout.C:
            fmt.Println("Current value")
        }
    }
}

func endProgram(ch chan string) {
    time.Sleep(time.Second * 8)
    ch <- "Exit function"
}

func getTimeoutDuration() time.Duration {
    return time.Second * 3
}

我知道上述写法并不推荐，因为在for循环内部使用defer属于不良编程实践。另一种替代方案是在for循环内使用time.After代替time.NewTimer，无需手动调用停止方法，但如果函数在定时器触发前退出，time.After会引发资源泄漏。请问在这种场景下，停止超时定时器的最佳方式是什么？

这个问题确实是Go并发编程里常见的坑，咱们一步步拆解解决：

为什么原有写法有问题？ #

• 循环内用defer：defer是在函数退出时才执行的，而非循环迭代结束。这意味着每次循环创建的timer都会被defer挂起，直到整个printFunc退出才会批量Stop，期间会积累大量未关闭的timer，浪费系统资源。
• 用time.After替代的隐患：time.After本质是内部创建timer后启动了一个goroutine，等待超时后向通道发消息。如果函数在超时前通过ch通道退出，这个内部goroutine会一直等到超时才销毁，造成不必要的资源泄漏。

最佳解决方案：手动控制Timer的生命周期 #

正确的做法是在每次循环迭代中，创建timer后，在select分支处理完成后立即手动调用Stop，并按需处理定时器是否已触发的情况。

修改后的代码如下：

package main

import (
    "fmt"
    "time"
)

func main() {
    ch := make(chan string)
    go endProgram(ch)
    printFunc(ch)
}

func printFunc(ch chan string) {
    for {
        timeout := time.NewTimer(getTimeoutDuration())
        select {
        case s := <-ch:
            // 提前退出循环，先停止定时器
            if !timeout.Stop() {
                // Stop返回false说明定时器已触发，通道内有未接收的消息
                // 必须排空通道，避免对应goroutine阻塞泄漏
                <-timeout.C
            }
            fmt.Println(s)
            return
        case <-timeout.C:
            fmt.Println("Current value")
            // 定时器已触发，Stop返回false，无需额外处理
            _ = timeout.Stop()
        }
        // 当前迭代的timer已被处理，无残留资源
    }
}

func endProgram(ch chan string) {
    time.Sleep(time.Second * 8)
    ch <- "Exit function"
}

func getTimeoutDuration() time.Duration {
    return time.Second * 3
}

代码细节解释 #

• ch分支的Stop处理：当从ch接收到消息要退出时，定时器大概率还未触发，Stop()会返回true并关闭定时器通道。但如果刚好出现竞态（接收ch的同时定时器触发），Stop()会返回false，此时必须排空timeout.C通道，否则对应的goroutine会一直阻塞。
• timeout.C分支的Stop处理：这里定时器已经触发，Stop()返回false，我们可以忽略这个返回值——因为通道里的消息已经被接收，不会有残留的阻塞goroutine。
• 弃用defer：确保每次循环的timer都在当前迭代内被处理，不会积累未关闭的资源。

这种方式既规避了循环内defer的不良实践，又彻底解决了time.After的资源泄漏问题，是处理这类场景的标准做法。

内容的提问来源于stack exchange，提问作者Abhay Gupta