Golang 封装文件读写的上下文取消机制-Golang-PHP中文网

os.file.read/write 无视 context.done() 因其为同步系统调用且不接收或检查 context；推荐用 io.copycontext（go 1.19+）实现上下文感知的流复制，或手动分块 + select 实现细粒度控制。

golang 封装文件读写的上下文取消机制

os.File.Read/Write 为什么无视 context.Done()

因为 os.File 的 Read 和 Write 是纯同步系统调用，不接收 context.Context 参数，也不检查 ctx.Done()。哪怕你传了带超时的 context 进 goroutine，Read 仍会卡在内核态等数据就绪——直到磁盘返回、出错或被信号中断（而 Go 运行时默认不发信号）。这不是 bug，是底层抽象层级决定的：IO 操作和 context 控制层是解耦的。

io.CopyContext 是最简可行方案（Go 1.19+）

如果你只是做文件到文件、文件到 pipe 或类似流式复制，直接用 io.CopyContext 替代 io.Copy 即可。它在每次 Read 后主动检查 ctx.Err()，一旦上下文取消或超时，立刻返回 context.Canceled 或 context.DeadlineExceeded。

只适用于 io.Reader → io.Writer 场景，不支持 ReadAt/WriteAt 或 mmap
缓冲区大小由内部控制（默认 32KB），无需手动指定
性能损耗极小，对 GB 级文件影响可忽略
错误需用 errors.Is(err, context.Canceled) 判断，不能只看 err == context.Canceled

ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
defer cancel()
n, err := io.CopyContext(dst, src)
if err != nil {
    if errors.Is(err, context.DeadlineExceeded) {
        log.Printf("复制超时，已写入 %d 字节", n)
    }
    return err
}

需要 ReadAt/自定义缓冲/进度上报？自己分块 + select

当标准 io.CopyContext 不够用（比如要读取特定偏移、上报进度、或控制内存占用），就得手动实现分块循环，并在每次迭代前监听 ctx.Done()。

Golang Naming

Go（Golang）命名规范 — 包括包、构造函数、结构体、接口、常量、枚举、错误、布尔值、接收器、getter/setter、函数等。

下载

缓冲区建议 64KB～1MB：太小增加系统调用开销，太大易触发 OOM
必须把 Read 放在 select 的 default 分支，否则可能饿死取消信号
每次 Read 前先 select，避免刚进入阻塞就被取消却无法响应
记得处理部分读取：如果 Read 返回 n > 0 但 err == nil，说明有数据可用，别丢弃

buf := make([]byte, 64*1024)
for {
    select {
    case  0 {
        // 处理 buf[:n]
        if _, werr := dst.Write(buf[:n]); werr != nil {
            return werr
        }
    }
    if err == io.EOF {
        break
    }
    if err != nil {
        return err
    }
}