Golang实现文件上传下载功能详解

从现在开始，我们要努力学习啦！今天我给大家带来《Golang实现文件上传与下载功能》，感兴趣的朋友请继续看下去吧！下文中的内容我们主要会涉及到等等知识点，如果在阅读本文过程中有遇到不清楚的地方，欢迎留言呀！我们一起讨论，一起学习！

首先实现文件上传与下载功能，通过net/http解析multipart表单获取文件并保存；其次设置响应头触发浏览器下载，防止路径穿越；最后通过限制大小、校验类型、使用随机命名等措施保障安全。

实现文件上传与下载功能在Golang中非常常见，尤其在构建Web服务时。通过标准库net/http和os等，可以轻松完成这些操作。下面介绍如何实现一个简洁、安全、可扩展的文件上传与下载服务。

文件上传处理

文件上传的核心是解析HTTP请求中的multipart/form-data数据，提取上传的文件并保存到服务器指定目录。

关键步骤包括：

• 使用http.Request的ParseMultipartForm方法解析请求体
• 通过FormFile获取上传的文件句柄
• 使用os.Create创建目标文件，并用io.Copy将上传内容写入磁盘
• 设置合理的内存和磁盘大小限制，防止恶意大文件攻击

func uploadHandler(w http.ResponseWriter, r *http.Request) {
    if r.Method != "POST" {
        http.Error(w, "仅支持POST请求", http.StatusMethodNotAllowed)
        return
    }

    // 解析multipart表单，最大内存10MB
    r.ParseMultipartForm(10 << 20)

    file, handler, err := r.FormFile("file")
    if err != nil {
        http.Error(w, "无法获取上传文件", http.StatusBadRequest)
        return
    }
    defer file.Close()

    // 创建保存文件
    dst, err := os.Create("./uploads/" + handler.Filename)
    if err != nil {
        http.Error(w, "无法创建文件", http.StatusInternalServerError)
        return
    }
    defer dst.Close()

    // 复制文件内容
    io.Copy(dst, file)
    fmt.Fprintf(w, "文件 %s 上传成功", handler.Filename)
}

文件下载实现

文件下载的关键是设置正确的响应头，让浏览器触发下载行为，而不是直接显示内容。

主要操作包括：

• 验证请求的文件名，防止路径穿越攻击（如 ../../etc/passwd）
• 使用os.Open打开文件
• 设置Content-Disposition头为attachment模式
• 设置Content-Type为application/octet-stream以强制下载
• 使用http.ServeContent安全地发送文件流

func downloadHandler(w http.ResponseWriter, r *http.Request) {
    filename := r.URL.Query().Get("file")
    if filename == "" {
        http.Error(w, "缺少文件名参数", http.StatusBadRequest)
        return
    }

    // 简单路径安全检查
    if strings.Contains(filename, "..") || strings.Contains(filename, "/") {
        http.Error(w, "非法文件名", http.StatusBadRequest)
        return
    }

    path := "./uploads/" + filename
    file, err := os.Open(path)
    if err != nil {
        http.Error(w, "文件不存在", http.StatusNotFound)
        return
    }
    defer file.Close()

    // 设置响应头
    w.Header().Set("Content-Disposition", "attachment; filename="+filename)
    w.Header().Set("Content-Type", "application/octet-stream")

    http.ServeContent(w, r, filename, time.Now(), file)
}

安全与优化建议

在生产环境中，需额外考虑安全性与性能：

• 限制上传文件大小，避免资源耗尽
• 校验文件类型（如通过magic number），防止恶意文件上传
• 使用随机文件名或哈希命名，避免覆盖和信息泄露
• 对上传目录做权限控制，禁止执行权限
• 考虑使用临时目录和异步处理提升响应速度

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~