面试题：Go 中复杂请求链下 Go Context 的传递与控制

1. Go Context 的传递与使用方法

传递 Context：在微服务架构中，从请求的入口开始创建一个 Context，通常是 context.Background()，然后在每次调用下游服务或开启子协程时，将这个 Context 传递下去。例如，在一个 HTTP 处理函数中：

package main

import (
    "context"
    "fmt"
    "net/http"
    "time"
)

func downstreamService(ctx context.Context) {
    select {
    case <-ctx.Done():
        fmt.Println("downstreamService cancelled")
        return
    case <-time.After(2 * time.Second):
        fmt.Println("downstreamService completed")
    }
}

func handler(w http.ResponseWriter, r *http.Request) {
    ctx := context.Background()
    // 假设这里要调用下游服务
    go downstreamService(ctx)
    // 处理请求的其他逻辑
    fmt.Fprintf(w, "Request processed")
}

func main() {
    http.HandleFunc("/", handler)
    http.ListenAndServe(":8080", nil)
}

在子协程中使用 Context：在子协程内部，通过 select 语句监听 ctx.Done() 通道，当这个通道接收到信号时，意味着 Context 被取消或超时，子协程应立即结束操作。例如：

func worker(ctx context.Context) {
    for {
        select {
        case <-ctx.Done():
            fmt.Println("worker cancelled")
            return
        default:
            // 执行工作逻辑
            fmt.Println("worker working")
            time.Sleep(100 * time.Millisecond)
        }
    }
}

2. 可能遇到的问题及解决方案

问题：未正确传递 Context，导致部分子协程无法感知取消信号。
- 解决方案：确保在每一层调用中都正确传递 Context，包括函数调用、方法调用以及开启子协程时。可以通过代码审查和测试来保证这一点。例如，编写测试函数来验证子协程在父 Context 取消时是否能正确终止。
问题：Context 超时设置不合理，可能导致请求处理时间过长或过短。
- 解决方案：根据业务需求和系统性能测试来合理设置 Context 的超时时间。可以通过配置文件或动态配置的方式，使得超时时间可以根据不同环境或业务场景进行调整。例如，对于一些耗时较长的查询操作，可以适当延长超时时间；而对于实时性要求高的操作，设置较短的超时时间。
问题：嵌套的 Context 可能导致逻辑混乱，例如外层 Context 取消后，内层 Context 未正确处理。
- 解决方案：尽量保持 Context 传递的清晰和简单，避免不必要的嵌套。如果确实需要嵌套，确保内层 Context 能正确响应外层 Context 的取消信号。例如，可以通过 context.WithCancelCause 等函数来创建内层 Context，并在取消时传递外层 Context 的取消原因。

星途面试题库

面试题：Go 中复杂请求链下 Go Context 的传递与控制

知识考点

面试题答案

1. Go Context 的传递与使用方法

2. 可能遇到的问题及解决方案