Go语言高并发HTTP文件下载：揭秘os.File未关闭导致的完整性问题

// 核心下载逻辑片段
res,err := http.Get(addr)
defer res.Body.Close()

// ... 省略循环部分，假设每个vFile实例处理一个文件
v := &vFile{path,0}
bv :=bufio.NewWriterSize(v,1024*1024)
_, err:= io.Copy(bv,res.Body)
if err == nil { err = bv.Flush() }

// 自定义的vFile结构体及其Write方法
type vFile struct {
    path string
    cur int64 // 原始代码中此字段未被使用或更新
    err error // 原始代码中此字段用于存储错误
}
func (wtr *vFile) Write(buf []byte) {
    var f *os.File
    if wtr.cur == 0 { f,wtr.err = os.Create(wtr.path) }
    else { f,wtr.err = os.OpenFile(wtr.path,os.O_RDWR|os.O_APPEND,0666) }
    // 注意：原始代码中此处为 `f.WriteAt(buf, twr.path)`，
    // `twr.path` 显然是错误的，WriteAt第二个参数应为偏移量。
    // 假设其意图是顺序写入，这里应为 f.Write(buf) 或 f.WriteAt(buf, offset)。
    // 即使是 f.WriteAt(buf, wtr.cur)，wtr.cur也未更新。
    _, err := f.WriteAt(buf, twr.path) // 假设这里是 f.Write(buf) 且有错误处理
    // 关键问题：文件句柄 f 在此方法结束后未被关闭
}

package main

import (
    "bufio"
    "fmt"
    "io"
    "net/http"
    "os"
    "path/filepath"
    "sync" // 用于并发写入保护，如果每个vFile实例只被一个goroutine使用则非必需
)

// vFile 结构体用于管理文件写入，实现 io.Writer 和 io.Closer 接口
type vFile struct {
    file *os.File   // 存储已打开的文件句柄
    path string     // 文件路径
    mu   sync.Mutex // 保护文件写入操作，防止并发冲突
}

// NewVFile 创建并初始化一个 vFile 实例。
// 它负责打开文件，如果文件已存在则截断。
func NewVFile(filePath string) (*vFile, error) {
    // os.Create 会创建文件，如果文件已存在则截断为0字节
    f, err := os.Create(filePath)
    if err != nil {
        return nil, fmt.Errorf("无法创建文件 %s: %w", filePath, err)
    }
    return &vFile{
        file: f,
        path: filePath,
    }, nil
}

// Write 实现 io.Writer 接口。
// 它将字节切片写入到内部维护的文件句柄中。
func (wtr *vFile) Write(buf []byte) (n int, err error) {
    wtr.mu.Lock() // 锁定，防止多个goroutine同时写入同一个文件句柄
    defer wtr.mu.Unlock()

    // 直接使用已打开的文件句柄进行写入
    // 假设目的是顺序追加写入，使用 Write 方法更合适
    n, err = wtr.file.Write(buf)
    if err != nil {
        return n, fmt.Errorf("写入文件 %s 失败: %w", wtr.path, err)
    }
    return n, nil
}

// Close 实现 io.Closer 接口。
// 它负责关闭内部维护的文件句柄，确保所有缓冲数据被刷新到磁盘。
func (wtr *vFile) Close() error {
    wtr.mu.Lock()
    defer wtr.mu.Unlock()

    if wtr.file != nil {
        err := wtr.file.Close()
        wtr.file = nil // 清空句柄，防止重复关闭或对已关闭句柄操作
        if err != nil {
            return fmt.Errorf("关闭文件 %s 失败: %w", wtr.path, err)
        }
    }
    return nil
}

// downloadFile