如何在超时后终止Go语言中长时间运行的函数？

你的代码失效的核心原因很明确：select只在每次for循环的开头执行一次。当进入default分支执行那段长时间运行的代码时，当前goroutine会完全被这段代码占用，根本没有机会去检查abort通道的状态。只有等这段长代码执行完毕，回到for循环开头，才会再次进入select判断通道状态——这就导致通道关闭的信号无法及时被响应。

可行的实现方案 #

1. 拆分长运行代码，插入中断检查 #

如果你的长运行代码可以拆分成多个独立的小任务，就在每个小任务执行完毕后，重新检查abort通道：

func foo(abort <-chan struct{}) {
    for {
        select {
        case <-abort:
            return
        default:
            // 拆分长代码为多个小步骤，每完成一步就回到select
            for i := 0; i < 1000; i++ {
                // 单次耗时极短的操作
            }
        }
    }
}

这种方式能保证abort信号被及时响应，但前提是代码具备可拆分性。

2. 将长代码放入独立goroutine，监听双信号 #

如果长代码无法拆分（比如调用第三方阻塞函数），可以把它放到单独的goroutine中，同时监听abort信号和代码完成信号：

func foo(abort <-chan struct{}) {
    done := make(chan struct{})
    go func() {
        // 执行长时间运行的代码
        ///long running code
        close(done)
    }()

    // 等待abort或任务完成
    select {
    case <-abort:
        // 注意：这里只是让foo返回，执行长代码的goroutine可能仍在运行
        // 若需彻底终止，需依赖代码自身支持取消，或用子进程替代goroutine
        return
    case <-done:
        return
    }
}

3. 使用标准库context实现取消（推荐） #

Go生态的路由框架（如Gin、Echo）的超时终止功能，几乎都是基于context包实现的，这也是官方推荐的取消方式：

import (
    "context"
    "time"
)

func foo(ctx context.Context) {
    done := make(chan struct{})
    go func() {
        ///long running code
        close(done)
    }()

    select {
    case <-ctx.Done():
        // 触发超时或主动取消，直接返回
        return
    case <-done:
        // 任务正常完成
        return
    }
}

// 调用示例：设置5秒超时
func main() {
    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
    defer cancel() // 确保资源释放
    foo(ctx)
}

注意事项 #

Go语言本身不支持强制终止goroutine，因为这可能导致内存泄漏、锁未释放等问题。如果你的长运行代码是CPU密集型且完全无法支持中断检查，唯一彻底的解决方案是将这段代码放到独立的子进程中，超时则通过操作系统信号杀掉子进程。

内容的提问来源于stack exchange，提问作者Michal Žídek