实现 Context 超时控制

题目描述

实现一个带超时的 HTTP 请求函数。

参考答案

func FetchWithTimeout(url string, timeout time.Duration) (string, error) {
    ctx, cancel := context.WithTimeout(context.Background(), timeout)
    defer cancel()
    
    req, err := http.NewRequestWithContext(ctx, "GET", url, nil)
    if err != nil {
        return "", err
    }
    
    resp, err := http.DefaultClient.Do(req)
    if err != nil {
        return "", err
    }
    defer resp.Body.Close()
    
    body, err := io.ReadAll(resp.Body)
    if err != nil {
        return "", err
    }
    
    return string(body), nil
}

使用示例

result, err := FetchWithTimeout("https://example.com", 5*time.Second)
if err != nil {
    log.Printf("error: %v", err)
    return
}
fmt.Println(result)

关键点

1. context.WithTimeout：创建超时上下文
2. defer cancel()：确保资源释放
3. http.NewRequestWithContext：绑定上下文