首先利用net/http包创建HTTP服务,通过POST接口上传文件并保存,再通过GET接口读取文件实现下载。具体为:注册/upload处理POST请求,解析multipart/form-data表单,获取文件流并写入本地;注册/download接口,校验文件名安全性后设置Content-Disposition响应头,调用http.ServeFile发送文件;主函数中配置路由并启动服务。代码简洁且依赖标准库,需注意添加上传限制、类型检查和路径遍历防护等安全措施。
Go语言实现文件上传下载功能,关键在于利用net/http包构建HTTP服务,通过标准的请求响应机制完成数据传输。下面直接说明如何实现一个简单可靠的文件上传下载服务。
文件上传功能实现
文件上传通常使用HTTP的POST方法,
前端
通过multipart/form-data格式提交文件。Go服务端需解析该请求,提取文件内容并保存到指定路径。
核心步骤包括:
注册一个POST接口(如
/upload
)用于接收文件
使用
r.ParseMultipartForm()
解析请求体
调用
r.FormFile()
获取上传的文件句柄
创建本地文件并使用
io.Copy()
将上传内容写入磁盘
示例代码片段:
立即学习
“
go语言免费学习笔记(深入)
”;
文件下载功能实现
文件下载通过GET请求触发,服务端读取本地文件并设置合适的响应头,让
浏览器
自动下载。
Comet AI 浏览器
下载 Comet AI 浏览器,体验由 Perplexity AI 驱动的革命性上网方式。内置 AI 助手可实时总结网页、跨标签页对比信息、自动执行任务。告别繁琐操作,让 AI 成为你的浏览副驾,大幅提升研究与工作效率。支持 Windows、macOS、Android 和 iOS。
下载
关键点:
注册GET接口(如
/download/:filename
)
从请求路径或查询参数中获取文件名
检查文件是否存在,防止路径穿越攻击
设置
Content-Disposition
为attachment,指定下载文件名
使用
http.ServeFile()
直接输出文件内容
示例代码:
服务启动与
路由
配置
使用http包注册处理函数,并启动HTTP服务器。
前端可通过表单或Ajax上传文件,下载时构造对应URL即可。
基本上就这些。核心逻辑清晰,Go的
标准库
已足够支撑基础的文件传输需求,无需引入额外框架。注意做好错误处理和安全校验,比如限制上传大小、检查文件类型、防止路径遍历等。
func uploadHandler(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
http.Error(w, "仅支持POST请求", http.StatusMethodNotAllowed)
return
}
file, header, err := r.FormFile("file")
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
defer file.Close()
outFile, err := os.Create(header.Filename)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
defer outFile.Close()
_, err = io.Copy(outFile, file)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
w.Write([]byte("文件上传成功"))
}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
}
filepath := "./uploads/" + filename
if _, err := os.Stat(filepath); 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)
}func main() {
http.HandleFunc("/upload", uploadHandler)
http.HandleFunc("/download", downloadHandler)
// 创建上传目录
os.MkdirAll("uploads", 0755)
fmt.Println("服务器启动,端口 :8080")
http.ListenAndServe(":8080", nil)
}