GolangHTTP文件上传与下载示例

net/http

multipart/form-data

net/http

Request

ResponseWriter

package main

import (
    "fmt"
    "io"
    "net/http"
    "os"
    "path/filepath"
    "time"
)

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

    // 限制上传文件大小，例如10MB
    r.ParseMultipartForm(10 << 20) // 10MB
    if r.MultipartForm == nil {
        http.Error(w, "无法解析多部分表单", http.StatusBadRequest)
        return
    }

    // 获取文件
    file, handler, err := r.FormFile("uploadFile") // "uploadFile" 是表单中文件字段的名称
    if err != nil {
        fmt.Printf("获取文件失败: %v\n", err)
        http.Error(w, "获取上传文件失败", http.StatusBadRequest)
        return
    }
    defer file.Close()

    // 检查文件类型（简单示例，实际应用中应更严格）
    // 例如，只允许图片或PDF
    ext := filepath.Ext(handler.Filename)
    if ext != ".jpg" && ext != ".png" && ext != ".pdf" {
        http.Error(w, "不支持的文件类型，只允许JPG, PNG, PDF", http.StatusBadRequest)
        return
    }

    // 创建目标文件
    // 为了避免文件名冲突，可以加上时间戳或UUID
    newFileName := fmt.Sprintf("%d%s", time.Now().UnixNano(), ext)
    dstPath := filepath.Join("./uploads", newFileName) // 假设存在一个名为 'uploads' 的目录
    dst, err := os.Create(dstPath)
    if err != nil {
        fmt.Printf("创建目标文件失败: %v\n", err)
        http.Error(w, "服务器无法创建文件", http.StatusInternalServerError)
        return
    }
    defer dst.Close()

    // 将上传文件内容拷贝到目标文件
    if _, err := io.Copy(dst, file); err != nil {
        fmt.Printf("拷贝文件内容失败: %v\n", err)
        http.Error(w, "保存文件失败", http.StatusInternalServerError)
        return
    }

    fmt.Fprintf(w, "文件 '%s' 上传成功，保存为 '%s'！", handler.Filename, newFileName)
}

// downloadHandler 处理文件下载请求
func downloadHandler(w http.ResponseWriter, r *http.Request) {
    // 从URL路径中获取文件名，例如 /download/myimage.jpg
    fileName := filepath.Base(r.URL.Path) // 只取文件名部分，防止路径遍历攻击
    filePath := filepath.Join("./uploads", fileName)

    // 检查文件是否存在
    _, err := os.Stat(filePath)
    if os.IsNotExist(err) {
        http.Error(w, "文件不存在", http.StatusNotFound)
        return
    } else if err != nil {
        http.Error(w, "无法访问文件", http.StatusInternalServerError)
        return
    }

    // 设置响应头，告诉浏览器这是一个下载文件
    w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=\"%s\"", fileName))
    w.Header().Set("Content-Type", "application/octet-stream") // 常见的二进制流类型
    // 也可以根据文件扩展名设置更具体的Content-Type，例如 image/jpeg, application/pdf
    // mimeType := mime.TypeByExtension(filepath.Ext(fileName))
    // if mimeType != "" {
    //  w.Header().Set("Content-Type", mimeType)
    // }

    // 打开并传输文件
    http.ServeFile(w, r, filePath) // net/http 包提供了便捷的 ServeFile 函数
}

func main() {
    // 确保上传目录存在
    if _, err := os.Stat("./uploads"); os.IsNotExist(err) {
        os.Mkdir("./uploads", 0755)
    }

    http.HandleFunc("/upload", uploadHandler)
    http.HandleFunc("/download/", downloadHandler) // 注意这里的斜杠，表示匹配 /download/ 后的所有路径
    fmt.Println("服务器正在监听 :8080...")
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        fmt.Printf("服务器启动失败: %v\n", err)
    }
}

curl -X POST -F "uploadFile=@/path/to/your/image.jpg" http://localhost:8080/upload

/path/to/your/image.jpg

curl -O http://localhost:8080/download/1678888888888888888.jpg # 替换成你上传后的文件名

-O

multipart/form-data

io.Copy

io.CopyBuffer

defer

Range

Range

http.StatusBadRequest

http.StatusInternalServerError

http.StatusNotFound

r.FormFile

os.Create

defer file.Close()

"Internal Server Error"

log.Printf

logrus

zap

fmt.Printf("创建目标文件失败: %v\n", err)