.jpg)
如何用go實(shí)現(xiàn)文件上傳和下載功能？首先，使用net/http包處理multipart/form-data格式的文件上傳，通過r.parsemultipartform限制內(nèi)存大小，r.formfile獲取文件句柄，并用io.copy保存到磁盤；其次，使用http.fileserver提供靜態(tài)文件服務(wù)或?qū)ttp.servefile用于精細(xì)控制下載行為；最后，注意文件大小限制、文件名安全、mime類型識別等常見問題。具體步驟：1.解析multipart請求并限制上傳大小；2.獲取并保存上傳文件；3.使用http.fileserver或http.servefile實(shí)現(xiàn)下載服務(wù)；4.設(shè)置content-disposition控制瀏覽器下載或預(yù)覽；5.防范文件過大、文件名沖突、并發(fā)寫入等問題。
在Web開發(fā)中，文件上傳和下載是常見的功能需求。golang作為一門性能優(yōu)秀、并發(fā)友好的語言，在構(gòu)建這類服務(wù)時(shí)非常合適。本文將重點(diǎn)介紹如何用Go處理文件上傳（基于multipart表單）和文件下載（靜態(tài)文件服務(wù)），并給出一些實(shí)用建議。
使用multipart/form-data處理文件上傳
HTTP請求中，文件上傳通常使用multipart/form-data格式發(fā)送。Go標(biāo)準(zhǔn)庫中的net/http包已經(jīng)內(nèi)置了對這種格式的支持，我們可以通過以下步驟來處理上傳：
- 解析請求中的multipart數(shù)據(jù)：使用r.ParseMultipartForm()方法，并傳入一個(gè)最大內(nèi)存大小參數(shù)，例如10
- 獲取上傳的文件句柄：通過r.FormFile(“file”)獲取文件對象，其中”file”是前端傳遞的字段名。
- 保存文件到磁盤：使用ioutil.TempDir創(chuàng)建臨時(shí)目錄，或者指定目標(biāo)路徑，然后用io.Copy將上傳的文件寫入本地。
- 注意清理資源：上傳完成后記得關(guān)閉文件句柄，避免資源泄露。
示例代碼片段如下：
立即學(xué)習(xí)“go語言免費(fèi)學(xué)習(xí)筆記（深入）”；
func uploadHandler(w http.ResponseWriter, r *http.Request) { // 限制上傳文件大小為10MB r.ParseMultipartForm(10 << 20) file, handler, err := r.FormFile("file") if err != nil { http.Error(w, "Error retrieving the file", http.StatusBadRequest) return } defer file.Close() // 創(chuàng)建目標(biāo)文件 dst, err := os.Create(handler.Filename) if err != nil { http.Error(w, "Unable to save the file", http.StatusInternalServerError) return } defer dst.Close() // 拷貝上傳內(nèi)容到目標(biāo)文件 if _, err := io.Copy(dst, file); err != nil { http.Error(w, "Error saving the file", http.StatusInternalServerError) return } fmt.Fprintf(w, "File %s uploaded successfully", handler.Filename) }
注意：實(shí)際部署中應(yīng)考慮文件重命名、路徑安全、文件類型檢查等問題。
提供靜態(tài)文件下載服務(wù)
Go也非常適合用來做靜態(tài)文件服務(wù)器，比如提供圖片、文檔等的下載服務(wù)。這可以通過http.FileServer結(jié)合http.Handle或http.ServeFile實(shí)現(xiàn)。
快速搭建靜態(tài)文件服務(wù)器
如果你希望整個(gè)目錄可訪問，可以使用http.FileServer：
http.Handle("/Static/", http.StripPrefix("/static/", http.FileServer(http.Dir("public")))) log.Fatal(http.ListenAndServe(":8080", nil))
這段代碼會把public目錄下的內(nèi)容映射到/static/路徑下，用戶訪問http://localhost:8080/static/filename即可下載對應(yīng)文件。
精細(xì)控制文件下載行為
如果你想對每個(gè)文件下載進(jìn)行更精細(xì)的控制，比如記錄日志、驗(yàn)證權(quán)限、設(shè)置自定義響應(yīng)頭等，可以使用http.ServeFile函數(shù)：
func downloadHandler(w http.ResponseWriter, r *http.Request) { filename := r.URL.Query().Get("file") filePath := filepath.Join("uploads", filename) // 可以在這里添加權(quán)限判斷邏輯 w.Header().Set("Content-Disposition", "attachment") http.ServeFile(w, r, filePath) }
這個(gè)例子中，我們強(qiáng)制瀏覽器下載而不是預(yù)覽文件，通過設(shè)置Content-Disposition: attachment實(shí)現(xiàn)。
小技巧：如果想讓某些文件支持在線預(yù)覽（如PDF），可以省略該頭部，瀏覽器會根據(jù)MIME類型決定是否直接展示。
常見問題與注意事項(xiàng)
- 上傳文件過大導(dǎo)致阻塞：一定要限制上傳文件的大小，否則可能耗盡服務(wù)器資源。
- 文件名沖突或不安全：上傳前應(yīng)對文件名進(jìn)行過濾或重命名，防止出現(xiàn)路徑穿越攻擊（如../../）。
- MIME類型識別錯(cuò)誤：對于下載服務(wù)，確保服務(wù)器能正確識別文件的MIME類型，否則可能導(dǎo)致瀏覽器處理異常。
- 并發(fā)寫入沖突：多個(gè)用戶同時(shí)上傳同名文件時(shí)，需要考慮加鎖或使用唯一標(biāo)識符重命名。
基本上就這些。Go在這方面的標(biāo)準(zhǔn)庫已經(jīng)足夠強(qiáng)大，只要稍作封裝就能滿足大多數(shù)場景的需求。
.png)
