面试题：Go中复杂场景下context取消信号传递的优化

设计思路

全局Context贯穿：从顶层函数传入一个context.Context，该context作为整个程序的根context，所有的嵌套函数和并发操作都基于这个根context创建子context。这样，当顶层的context被取消时，所有依赖它的子context也会自动取消。
合理创建子Context：在每个需要并发操作的地方，使用context.WithCancel、context.WithTimeout或context.WithDeadline基于父context创建子context。WithCancel用于手动取消，WithTimeout和WithDeadline用于在特定时间后自动取消。
及时取消未使用的Context：在不需要继续使用某个context时，及时调用取消函数（对于context.WithCancel创建的context），释放相关资源。
传播取消信号：在嵌套调用中，将context作为参数传递给下层函数，确保取消信号能够一直传递到最底层的函数和并发操作中。

关键实现要点

正确传递Context：在函数定义和调用中，确保context.Context作为第一个参数传递。例如：

func innerFunction(ctx context.Context, otherArgs ...interface{}) error {
    // 函数逻辑
}

func outerFunction(ctx context.Context) error {
    err := innerFunction(ctx, otherArgs)
    if err != nil {
        return err
    }
    return nil
}

并发操作中的Context处理：在使用go关键字启动并发操作时，将context传递给并发函数，并在函数内部使用ctx.Done()通道来监听取消信号。例如：

func worker(ctx context.Context) {
    for {
        select {
        case <-ctx.Done():
            return
        default:
            // 执行任务
        }
    }
}

func main() {
    ctx, cancel := context.WithCancel(context.Background())
    defer cancel()

    go worker(ctx)

    // 主逻辑
}

超时和截止时间设置：对于有时间限制的操作，使用context.WithTimeout或context.WithDeadline。例如：

func main() {
    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
    defer cancel()

    err := someLongRunningOperation(ctx)
    if err != nil {
        if err == context.DeadlineExceeded {
            // 处理超时
        }
    }
}

资源清理：在context取消时，确保所有相关的资源（如文件句柄、网络连接等）都被正确关闭或释放。可以使用defer语句在函数结束时执行清理操作。例如：

func someFunction(ctx context.Context) {
    file, err := os.Open("example.txt")
    if err != nil {
        return
    }
    defer file.Close()

    for {
        select {
        case <-ctx.Done():
            // 额外的清理逻辑
            return
        default:
            // 读取文件内容
        }
    }
}

星途面试题库

面试题：Go中复杂场景下context取消信号传递的优化

知识考点

面试题答案

设计思路

关键实现要点