Golang Web表单文件流处理优化示例

func uploadHandler(w http.ResponseWriter, r *http.Request) {
    // 限制请求体大小，防止恶意大文件
    r.ParseMultipartForm(32 << 20) // 32MB
<pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">file, header, err := r.FormFile("file")
if err != nil {
    http.Error(w, "无法获取文件", http.StatusBadRequest)
    return
}
defer file.Close()

// 打印文件信息
log.Printf("文件名: %s, 大小: %d", header.Filename, header.Size)

// 流式写入磁盘（也可转发到OSS、S3等）
outFile, err := os.Create("/tmp/" + header.Filename)
if err != nil {
    http.Error(w, "创建文件失败", http.StatusInternalServerError)
    return
}
defer outFile.Close()

// 使用 io.Copy 边读边写，不占内存
_, err = io.Copy(outFile, file)
if err != nil {
    http.Error(w, "保存文件失败", http.StatusInternalServerError)
    return
}

w.Write([]byte("上传成功"))

buffer := make([]byte, 512)
_, err := file.Read(buffer)
if err != nil && err != io.EOF {
    // 处理错误
}
<p>// 检测 MIME 类型
fileType := http.DetectContentType(buffer)
if !strings.HasPrefix(fileType, "image/") {
http.Error(w, "仅支持图片格式", http.StatusBadRequest)
return
}</p><p>// 重置文件指针以便后续写入
file.Seek(0, 0)

server := &http.Server{
    Addr:         ":8080",
    ReadTimeout:  30 * time.Second,
    WriteTimeout: 60 * time.Second,
}

go func() {
    processLargeFile("/tmp/" + filename)
}()
w.Write([]byte("接收成功，后台处理中"))

limitedReader := io.LimitReader(file, 32<<20) // 最多读 32MB
_, err = io.Copy(outFile, limitedReader)
if err != nil {
    // 可能是超出大小
}