为何context.WithTimeout有时会执行失败？

问题分析 #

你的测试偶尔失败的核心原因是时间窗口过小，受系统调度和time.Sleep精度波动影响：

• 测试中设置超时时间为10ms，函数执行时间为11ms，两者仅差1ms。但time.Sleep的精度并非绝对精确，加上Go goroutine的调度延迟，可能出现两种异常情况：
  1. doCall中的time.Sleep(11ms)因系统调度提前唤醒，实际在10ms内完成，导致callDone通道先被触发，函数返回ErrFuncTimeProcess，测试失败。
  2. 上下文超时的触发因调度延迟被推迟，同样可能导致callDone信号先被接收。
• 另外，当前实现存在goroutine泄漏风险：当超时发生后，doCall仍会继续执行完time.Sleep，再向通道发送数据（虽不会阻塞，但属于不必要的资源消耗）。

解决方案 #

1. 增大时间差（快速修复测试） #

将测试中的时间差拉大，大幅降低调度波动的影响：

func TestTimeout(t *testing.T) {
    err := internal.Timeout(100*time.Millisecond, 200*time.Millisecond)
    if err != internal.ErrTimeout {
        t.Errorf("unexpected error. wants timeout error, got: %v", err)
    }
}

2. 用Context控制goroutine（优化核心实现） #

让doCall监听上下文，一旦超时就立即退出，既避免goroutine泄漏，也能让超时逻辑更可靠：

package internal

import (
    "context"
    "errors"
    "time"
)

var (
    ErrTimeout         = errors.New("timeout error")
    ErrFuncTimeProcess = errors.New("function time process error")
)

func Timeout(timeout, funcTimeProcess time.Duration) error {
    ctx, cancel := context.WithTimeout(context.Background(), timeout)
    defer cancel()

    callDone := make(chan struct{}, 1)
    go doCall(ctx, callDone, funcTimeProcess)

    select {
    case <-ctx.Done():
        return ErrTimeout
    case <-callDone:
        return ErrFuncTimeProcess
    }
}

func doCall(ctx context.Context, c chan<- struct{}, timeout time.Duration) {
    select {
    case <-ctx.Done():
        // 超时直接退出，不再执行sleep
        return
    case <-time.After(timeout):
        c <- struct{}{}
    }
}

3. 补充反向测试用例 #

为覆盖两种逻辑场景，建议添加函数执行时间短于超时的测试，确保逻辑双向正确：

func TestNoTimeout(t *testing.T) {
    err := internal.Timeout(200*time.Millisecond, 100*time.Millisecond)
    if err != internal.ErrFuncTimeProcess {
        t.Errorf("unexpected error. wants function process error, got: %v", err)
    }
}

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