golang保存上传文件

使用Golang保存上传文件 --- 在开发Web应用过程中，经常会遇到需要用户上传文件的需求。Golang作为一门现代的高性能编程语言，提供了简洁、高效的方式来处理文件上传。本文将介绍如何使用Golang保存上传的文件。 ## 文件上传的基本流程 1. 创建一个HTTP路由，指定文件上传的接口和处理函数。 ```go http.HandleFunc("/", uploadHandler) ``` 2. 根据请求的方法，采取不同的处理方式。对于GET请求，返回文件上传的表单页面；对于POST请求，处理上传的文件。 3. 解析POST请求中的表单数据，获取上传的文件。 ```go file, _, err := r.FormFile("uploadFile") ``` 4. 创建一个新的文件，将上传的文件内容写入其中。 ```go newFile, err := os.Create(filepath.Join(uploadDir, header.Filename)) defer newFile.Close() _, err = io.Copy(newFile, file) ``` 5. 返回上传成功信息给客户端。 ```go w.Write([]byte("File uploaded successfully!")) ``` ## 示例代码 下面是一个完整的示例代码，演示了如何使用Golang保存上传的文件。 ```go package main import ( "fmt" "io" "net/http" "os" "path/filepath" ) func main() { http.HandleFunc("/", uploadHandler) http.ListenAndServe(":8080", nil) } func uploadHandler(w http.ResponseWriter, r *http.Request) { switch r.Method { case "GET": showUploadForm(w) case "POST": handleUpload(w, r) default: w.WriteHeader(http.StatusMethodNotAllowed) fmt.Fprintf(w, "Only GET and POST methods are supported.") } } func showUploadForm(w http.ResponseWriter) { html := `

` w.Write([]byte(html)) } func handleUpload(w http.ResponseWriter, r *http.Request) { file, header, err := r.FormFile("uploadFile") if err != nil { w.WriteHeader(http.StatusBadRequest) fmt.Fprintf(w, "Failed to get file from request: %v", err) return } defer file.Close() uploadDir := "./uploads" err = os.MkdirAll(uploadDir, os.ModePerm) if err != nil { w.WriteHeader(http.StatusInternalServerError) fmt.Fprintf(w, "Failed to create upload directory: %v", err) return } newFile, err := os.Create(filepath.Join(uploadDir, header.Filename)) if err != nil { w.WriteHeader(http.StatusInternalServerError) fmt.Fprintf(w, "Failed to create new file: %v", err) return } defer newFile.Close() _, err = io.Copy(newFile, file) if err != nil { w.WriteHeader(http.StatusInternalServerError) fmt.Fprintf(w, "Failed to save file: %v", err) return } w.Write([]byte("File uploaded successfully!")) } ``` 在上面的代码中，我们创建了一个简单的Web应用，当用户通过浏览器访问根路由时，会显示一个文件上传的表单页面。当用户选择并上传了文件之后，文件将保存在指定的目录中，并返回上传成功的信息给客户端。 ## 小结 本文介绍了如何使用Golang保存上传的文件。通过简单的代码实例，展示了处理文件上传的基本流程和实现方式。如今，越来越多的Web应用需要支持文件上传功能，而使用Golang来实现文件上传是一种非常高效和可靠的选择。