• 驗(yàn)證文件最大值
• 從請(qǐng)求驗(yàn)證文件和 POST 參數(shù)
• 檢查所提供的文件類型（我們只接受圖像和 PDF）
• 創(chuàng)建一個(gè)隨機(jī)文件名
• 將文件寫入硬盤
• 處理所有錯(cuò)誤，如果一切順利返回成功消息

第一步，我們定義處理程序:

func uploadFileHandler() http.HandlerFunc {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {

然后，我們使用 http.MaxBytesReader 驗(yàn)證文件大小，當(dāng)文件大小大于設(shè)定值時(shí)它將返回一個(gè)錯(cuò)誤。錯(cuò)誤將被一個(gè)助手程序 renderError 進(jìn)行處理，它返回錯(cuò)誤信息及對(duì)應(yīng)的 HTTP 狀態(tài)碼。

r.Body = http.MaxBytesReader(w, r.Body, maxUploadSize)
if err := r.ParseMultipartForm(maxUploadSize); err != nil {
renderError(w, "FILE_TOO_BIG", http.StatusBadRequest)
return
}

如果文件大小驗(yàn)證通過，我們將檢查并解析表單參數(shù)類型和上傳的文件，并讀取文件。在本例中，為了清晰起見，我們不使用花哨的 io.Reader 和 io.Writer 接口，我們只是簡單的將文件讀取到一個(gè)字節(jié)數(shù)組中，這點(diǎn)我們后面會(huì)寫到。

fileType := r.PostFormValue("type")
file, _, err := r.FormFile("uploadFile")
if err != nil {
renderError(w, "INVALID_FILE", http.StatusBadRequest)
return
}
defer file.Close()
fileBytes, err := ioutil.ReadAll(file)
if err != nil {
renderError(w, "INVALID_FILE", http.StatusBadRequest)
return
}

現(xiàn)在我們成功的驗(yàn)證了文件的大小，并且讀取了文件，接下來我們?cè)摍z驗(yàn)文件的類型了。一種廉價(jià)但是并不安全的方式，只檢查文件擴(kuò)展名，并相信用戶沒有改變它，但是對(duì)于一個(gè)正式的項(xiàng)目來講不應(yīng)該這么做。

幸運(yùn)的是，Go 標(biāo)準(zhǔn)庫提供給我們一個(gè) http.DetectContentType 函數(shù)，這個(gè)函數(shù)基于 mimesniff 算法，只需要讀取文件的前 512 個(gè)字節(jié)就能夠判定文件類型。

filetype := http.DetectContentType(fileBytes)
if filetype != "image/jpeg" && filetype != "image/jpg" &&
filetype != "image/gif" && filetype != "image/png" &&
filetype != "application/pdf" {
renderError(w, "INVALID_FILE_TYPE", http.StatusBadRequest)
return
}

在實(shí)際應(yīng)用程序中，我們可能會(huì)使用文件元數(shù)據(jù)做一些事情，例如將其保存到數(shù)據(jù)庫或?qū)⑵渫扑偷酵獠糠?wù)——以任何方式，我們將解析和操作元數(shù)據(jù)。這里我們創(chuàng)建一個(gè)隨機(jī)的新名字（這在實(shí)踐中可能是一個(gè) UUID）并將新文件名記錄下來。

fileName := randToken(12)
fileEndings, err := mime.ExtensionsByType(fileType)
if err != nil {
renderError(w, "CANT_READ_FILE_TYPE", http.StatusInternalServerError)
return
}
newPath := filepath.Join(uploadPath, fileName+fileEndings[0])
fmt.Printf("FileType: %s, File: %s\n", fileType, newPath)

馬上就大功告成了，只剩下一個(gè)關(guān)鍵步驟-寫文件。如上文所提到的，我們只需要復(fù)制讀取的二進(jìn)制文件到一個(gè)新創(chuàng)建的名為 newFile 的文件處理程序里。

如果所有部分都沒問題，我們給用戶返回一個(gè) SUCCESS 信息。

newFile, err := os.Create(newPath)
if err != nil {
renderError(w, "CANT_WRITE_FILE", http.StatusInternalServerError)
return
}
defer newFile.Close()
if _, err := newFile.Write(fileBytes); err != nil {
renderError(w, "CANT_WRITE_FILE", http.StatusInternalServerError)
return
}
w.Write([]byte("SUCCESS"))

這樣可以了. 你可以對(duì)這個(gè)簡單的例子進(jìn)行測試，使用虛擬的文件上傳 HTML 頁面，cURL 或者工具例如 postman [1] 。

這里是完整的代碼示例 這里 [2]

結(jié)論 這是又一個(gè)證明了 Go 如何允許用戶為 Web 編寫簡單而強(qiáng)大的軟件，而不必像處理其他語言和生態(tài)系統(tǒng)中固有的無數(shù)抽象層。

在接下來的篇幅中，我將展示一些在我第一次使用 Go 語言編寫正式的 Web 應(yīng)用中其他細(xì)節(jié)，敬請(qǐng)期待。；）

// 根據(jù) reddit 用戶 lstokeworth 的反饋對(duì)部分代碼進(jìn)行了修改。謝謝:)

到此這篇關(guān)于 Go 語言完成 HTTP 文件上傳和下載的文章就介紹到這了,更多相關(guān) Go HTTP 文件上傳內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！