Golang如何搭建文件上传与下载服务_Golang 文件传输服务实践

首先实现文件上传接口,通过ParseMultipartForm解析multipart/form-data请求,限制内存使用32MB,提取文件并保存;接着实现下载功能。

golang如何搭建文件上传与下载服务_golang 文件传输服务实践

在Go语言中搭建一个高效的文件上传与下载服务并不复杂。利用标准库中的 net/http 和文件操作功能,可以快速实现稳定、安全的文件传输服务。下面从基础结构出发,介绍如何构建一个支持文件上传与下载的HTTP服务,并给出实际可运行的代码示例。

1. 实现文件上传接口

文件上传的核心是处理 multipart/form-data 类型的请求。Go 的 http.Request 提供了 ParseMultipartForm 方法来解析表单数据,包括文件字段。

以下是一个简单的文件上传处理函数:

func uploadHandler(w http.ResponseWriter, r *http.Request) {    if r.Method != "POST" {        http.Error(w, "仅支持 POST 请求", http.StatusMethodNotAllowed)        return    }    // 解析 multipart 表单,限制内存使用 32MB    err := r.ParseMultipartForm(32 << 20)    if err != nil {        http.Error(w, "解析表单失败", http.StatusBadRequest)        return    }    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()    // 将上传的文件内容复制到本地文件    _, err = io.Copy(dst, file)    if err != nil {        http.Error(w, "保存文件失败", http.StatusInternalServerError)        return    }    w.WriteHeader(http.StatusOK)    w.Write([]byte("文件上传成功: " + handler.Filename))}

注意点:

立即学习“go语言免费学习笔记(深入)”;

确保 ./uploads 目录存在,否则创建失败 限制上传大小防止恶意攻击 对文件名做安全校验,避免路径穿越(如:../../../etc/passwd)

2. 实现文件下载接口

文件下载本质上是读取服务器上的文件并通过 HTTP 响应返回,设置正确的响应头让浏览器触发下载行为。

示例如下:

func downloadHandler(w http.ResponseWriter, r *http.Request) {    filename := r.URL.Query().Get("file")    if filename == "" {        http.Error(w, "缺少文件名参数", http.StatusBadRequest)        return    }    filepath := "./uploads/" + filename    // 检查文件是否存在且不包含非法路径    if !strings.HasPrefix(filepath, "./uploads/") {        http.Error(w, "非法路径", http.StatusForbidden)        return    }    _, err := os.Stat(filepath)    if os.IsNotExist(err) {        http.Error(w, "文件不存在", http.StatusNotFound)        return    }    // 设置响应头,触发浏览器下载    w.Header().Set("Content-Disposition", "attachment; filename="+filename)    w.Header().Set("Content-Type", "application/octet-stream")    http.ServeFile(w, r, filepath)}

关键点:

验证文件路径合法性,防止目录遍历攻击 使用 Content-Disposition: attachment 强制下载 大文件可考虑分块传输或使用 io.Copy 控制缓冲

3. 启动HTTP服务并路由

将上传和下载处理函数注册到路由,并启动服务:

func main() {    // 确保上传目录存在    os.MkdirAll("./uploads", os.ModePerm)    http.HandleFunc("/upload", uploadHandler)    http.HandleFunc("/download", downloadHandler)    // 可选:提供静态页面用于测试上传    http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {        html := `                                                                                                `        w.Write([]byte(html))    })    println("服务已启动,访问 http://localhost:8080")    log.Fatal(http.ListenAndServe(":8080", nil))}

这样就完成了一个基本可用的文件传输服务。

4. 安全与优化建议

生产环境中还需考虑以下几点:

文件名安全:使用 UUID 或哈希重命名文件,避免覆盖或执行风险 类型检查:校验 MIME 类型或扩展名,限制可上传类型 权限控制:添加身份验证中间件(如 JWT)保护接口 超时设置:为大文件上传设置合理的 read/write 超时 并发限制:避免过多连接耗尽资源基本上就这些。Go 的简洁语法和强大标准库让文件服务实现变得直接高效,适合中小型项目快速落地。

以上就是Golang如何搭建文件上传与下载服务_Golang 文件传输服务实践的详细内容,更多请关注创想鸟其它相关文章!

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1426887.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2025年12月16日 20:54:32
下一篇 2025年12月16日 20:54:50

相关推荐

发表回复

登录后才能评论
关注微信