# 4.5 處理檔案上傳 你想處理一個由使用者上傳的檔案,比如你正在建設一個類似 Instagram 的網站,你需要儲存使用者拍攝的照片。這種需求該如何實現呢? 要使表單能夠上傳檔案,首先第一步就是要新增 form 的`enctype`屬性,`enctype`屬性有如下三種情況: ``` application/x-www-form-urlencoded 表示在傳送前編碼所有字元(預設) multipart/form-data 不對字元編碼。在使用包含檔案上傳控制元件的表單時,必須使用該值。 text/plain 空格轉換為 "+" 加號,但不對特殊字元編碼。 ``` 所以,建立新的表單 html 檔案, 命名為 upload.gtpl, html 程式碼應該類似於: ```html 上傳檔案
``` 在伺服器端,我們增加一個 handlerFunc: ```Go http.HandleFunc("/upload", upload) // 處理/upload 邏輯 func upload(w http.ResponseWriter, r *http.Request) { fmt.Println("method:", r.Method) //取得請求的方法 if r.Method == "GET" { crutime := time.Now().Unix() h := md5.New() io.WriteString(h, strconv.FormatInt(crutime, 10)) token := fmt.Sprintf("%x", h.Sum(nil)) t, _ := template.ParseFiles("upload.gtpl") t.Execute(w, token) } else { r.ParseMultipartForm(32 << 20) file, handler, err := r.FormFile("uploadfile") if err != nil { fmt.Println(err) return } defer file.Close() fmt.Fprintf(w, "%v", handler.Header) f, err := os.OpenFile("./test/"+handler.Filename, os.O_WRONLY|os.O_CREATE, 0666) // 此處假設當前目錄下已存在 test 目錄 if err != nil { fmt.Println(err) return } defer f.Close() io.Copy(f, file) } } ``` 透過上面的程式碼可以看到,處理檔案上傳我們需要呼叫 `r.ParseMultipartForm`,裡面的參數表示 `maxMemory`,呼叫 `ParseMultipartForm` 之後,上傳的檔案儲存在 `maxMemory` 大小的記憶體裡面,如果檔案大小超過了 `maxMemory`,那麼剩下的部分將儲存在系統的臨時檔案中。我們可以透過 `r.FormFile` 取得上面的檔案控制代碼,然後範例中使用了 `io.Copy` 來儲存檔案。 >取得其他非檔案欄位資訊的時候就不需要呼叫 `r.ParseForm`,因為在需要的時候 Go 自動會去呼叫。而且 `ParseMultipartForm` 呼叫一次之後,後面再次呼叫不會再有效果。 透過上面的範例我們可以看到我們上傳檔案主要三步處理: 1. 表單中增加 enctype="multipart/form-data" 2. 伺服器端呼叫 `r.ParseMultipartForm`,把上傳的檔案儲存在記憶體和臨時檔案中 3. 使用 `r.FormFile` 取得檔案控制代碼,然後對檔案進行儲存等處理。 檔案 handler 是 multipart.FileHeader,裡面儲存了如下結構資訊 ```Go type FileHeader struct { Filename string Header textproto.MIMEHeader // contains filtered or unexported fields } ``` 我們透過上面的範例程式碼顯示出來上傳檔案的資訊如下 ![](images/4.5.upload2.png) 圖 4.5 列印檔案上傳後伺服器端接受的資訊 ## 客戶端上傳檔案 我們上面的例子示範了如何透過表單上傳檔案,然後在伺服器端處理檔案,其實 Go 支援模擬客戶端表單功能支援檔案上傳,詳細用法請看如下範例: ```Go package main import ( "bytes" "fmt" "io" "io/ioutil" "mime/multipart" "net/http" "os" ) func postFile(filename string, targetUrl string) error { bodyBuf := &bytes.Buffer{} bodyWriter := multipart.NewWriter(bodyBuf) //關鍵的一步操作 fileWriter, err := bodyWriter.CreateFormFile("uploadfile", filename) if err != nil { fmt.Println("error writing to buffer") return err } //開啟檔案控制代碼操作 fh, err := os.Open(filename) if err != nil { fmt.Println("error opening file") return err } defer fh.Close() //iocopy _, err = io.Copy(fileWriter, fh) if err != nil { return err } contentType := bodyWriter.FormDataContentType() bodyWriter.Close() resp, err := http.Post(targetUrl, contentType, bodyBuf) if err != nil { return err } defer resp.Body.Close() resp_body, err := ioutil.ReadAll(resp.Body) if err != nil { return err } fmt.Println(resp.Status) fmt.Println(string(resp_body)) return nil } // sample usage func main() { target_url := "http://localhost:9090/upload" filename := "./astaxie.pdf" postFile(filename, target_url) } ``` 上面的例子詳細展示了客戶端如何向伺服器上傳一個檔案的例子,客戶端透過 multipart.Write 把檔案的文字流寫入一個快取中,然後呼叫 http 的 Post 方法把快取傳到伺服器。 >如果你還有其他普通欄位例如 username 之類別的需要同時寫入,那麼可以呼叫 multipart 的 WriteField 方法寫很多其他類似的欄位。 ## links * [目錄]() * 上一節:[防止多次提交表單](<04.4.md>) * 下一節:[小結](<04.6.md>)