以下是一个简单的Golang代码示例,实现了一个基于net/http的文件上传API接口:
package main
import (
"io"
"net/http"
"os"
)
func uploadFile(w http.ResponseWriter, r *http.Request) {
if r.Method == "GET" {
http.ServeFile(w, r, "upload.html")
return
}
r.ParseMultipartForm(32 << 20) // 设置最大内存32MB
file, handler, err := r.FormFile("uploadfile")
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
defer file.Close()
f, err := os.OpenFile("./uploads/"+handler.Filename, os.O_WRONLY|os.O_CREATE, 0666)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
defer f.Close()
io.Copy(f, file)
w.WriteHeader(http.StatusOK)
io.WriteString(w, "File uploaded successfully: "+handler.Filename)
}
func main() {
http.HandleFunc("/upload", uploadFile)
http.ListenAndServe(":8080", nil)
}
在这个例子中,我们定义了一个uploadFile
函数来处理文件上传。它首先检查HTTP方法是否为"GET",如果是,它将提供一个上传表单页面。如果方法是"POST",它将解析multipart表单数据,获取上传的文件,并将其保存到服务器的"./uploads"目录下。
确保你有一个名为upload.html
的HTML文件在同一目录下,用于提供上传表单界面。这个HTML文件应该包含一个表单,指定enctype="multipart/form-data"
并有一个文件输入字段<input type="file" name="uploadfile">
。
此外,确保服务器上存在./uploads
目录,并且服务器有权限写入该目录。
请注意,这个代码示例没有进行错误处理的详细检查,也没有添加跨域处理、请求限制等安全性和性能考虑因素。在实际应用中,你应该添加这些功能以保障API的安全性和稳定性。