如何改写Go定时循环：实现首次立即执行且支持Channel停止

问题描述 #

我原本用这样的for循环：

for ; true; <-time.After(duration) {
    // 任务逻辑
}

这个循环会立即执行第一次，之后每隔duration时长执行一次。现在我想改写成能通过done通道终止的版本，但用select实现时，第一次迭代就会等待<-time.After(duration)，没办法立即执行，请问该怎么解决？

我目前写出的代码是这样的：

for {
    select {
    case <-done:
        return
    case t := <-time.After(duration):
        fmt.Println("Tick at", t)
    }
}

解决方案 #

方法1：先执行一次任务，再进入循环等待

最直接的思路是把任务逻辑抽离，先手动执行一次，再进入select循环处理定时和终止信号：

// 抽离任务逻辑，避免重复代码
executeTask := func(t time.Time) {
    fmt.Println("Tick at", t)
}

// 首次立即执行任务
executeTask(time.Now())

for {
    select {
    case <-done:
        return
    case t := <-time.After(duration):
        executeTask(t)
    }
}

这种方式简单直观，适合任务逻辑不复杂的场景，无额外资源开销。

方法2：用初始触发通道在循环内完成所有逻辑

如果希望逻辑都封装在循环体里，可以创建一个预先带有值的通道触发首次执行，之后替换成定时通道：

// 初始化带初始值的通道，触发首次执行
trigger := make(chan time.Time, 1)
trigger <- time.Now()

for {
    select {
    case <-done:
        return
    case t := <-trigger:
        fmt.Println("Tick at", t)
        // 首次执行后，将触发源换成定时通道
        trigger = time.After(duration)
    }
}

方法3：使用time.NewTicker（推荐长期运行场景）

如果定时逻辑需要长期运行，time.NewTicker比重复创建time.After更高效（每次time.After都会生成新定时器，长期循环会造成不必要的资源浪费），同样先手动执行一次任务：

ticker := time.NewTicker(duration)
// 退出时停止ticker，避免资源泄漏
defer ticker.Stop()

// 首次立即执行任务
fmt.Println("Tick at", time.Now())

for {
    select {
    case <-done:
        return
    case t := <-ticker.C:
        fmt.Println("Tick at", t)
    }
}

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