Go语言中处理multipart/form-data文件上传

package main

import (
    "fmt"
    "io"
    "mime/multipart"
    "net/http"
    "os"
    "strconv"
)

// uploadHandler 处理文件上传请求
func uploadHandler(w http.ResponseWriter, r *http.Request) {
    // 确保请求方法是POST
    if r.Method != http.MethodPost {
        http.Error(w, "只支持POST请求", http.StatusMethodNotAllowed)
        return
    }

    // 定义内存限制，例如24KB。超过此大小的文件将写入临时文件。
    const maxMemory = (1 << 10) * 24 // 24KB
    err := r.ParseMultipartForm(maxMemory)
    if err != nil {
        http.Error(w, fmt.Sprintf("解析multipart表单失败: %v", err), http.StatusInternalServerError)
        return
    }

    // 遍历所有上传的文件
    // r.MultipartForm.File 是一个 map[string][]*multipart.FileHeader
    // 其中 key 是表单中文件字段的名称
    for formFieldName, fileHeaders := range r.MultipartForm.File {
        for _, header := range fileHeaders {
            // 打开上传的文件
            infile, err := header.Open()
            if err != nil {
                http.Error(w, fmt.Sprintf("打开上传文件失败: %v", err), http.StatusInternalServerError)
                return
            }
            defer infile.Close() // 确保文件流被关闭

            // 准备目标文件路径
            // 这里假设有一个名为 "uploaded" 的目录用于存放文件
            // 实际应用中需要确保该目录存在且有写入权限
            uploadDir := "./uploaded"
            if _, err := os.Stat(uploadDir); os.IsNotExist(err) {
                os.Mkdir(uploadDir, 0755) // 如果目录不存在则创建
            }

            destFilePath := uploadDir + "/" + header.Filename
            outfile, err := os.Create(destFilePath)
            if err != nil {
                http.Error(w, fmt.Sprintf("创建目标文件失败: %v", err), http.StatusInternalServerError)
                return
            }
            defer outfile.Close() // 确保目标文件被关闭

            // 将上传文件内容复制到目标文件
            written, err := io.Copy(outfile, infile)
            if err != nil {
                http.Error(w, fmt.Sprintf("写入文件失败: %v", err), http.StatusInternalServerError)
                return
            }

            // 响应客户端上传成功信息
            responseMsg := fmt.Sprintf("文件 '%s' (字段: %s) 上传成功，大小: %s 字节。\n", header.Filename, formFieldName, strconv.FormatInt(written, 10))
            w.Write([]byte(responseMsg))
            fmt.Printf(responseMsg) // 打印到服务器控制台
        }
    }

    // 清理临时文件
    // r.ParseMultipartForm 会在内部创建临时文件，
    // r.MultipartForm.RemoveAll() 会删除这些临时文件。
    // 建议在处理完请求后调用，尤其是在 defer 中。
    defer r.MultipartForm.RemoveAll()

    if len(r.MultipartForm.File) == 0 {
        w.Write([]byte("没有文件被上传。\n"))
    }
}

func main() {
    http.HandleFunc("/upload", uploadHandler)
    fmt.Println("服务器正在监听 :8080")
    http.ListenAndServe(":8080", nil)
}