檔案上傳處理

要使表單能夠上傳檔案,首先第一步就是要新增 form 的enctype屬性,enctype屬性有如下三種情況:

application/x-www-form-urlencoded   表示在傳送前編碼所有字元(預設)
multipart/form-data	  不對字元編碼。在使用包含檔案上傳控制元件的表單時,必須使用該值。
text/plain	  空格轉換為 "+" 加號,但不對特殊字元編碼。

client 網頁

<html>
<head>
	<title>上傳檔案</title>
</head>
<body>
<form enctype="multipart/form-data" action="/upload" method="post">
  <input type="file" name="uploadfile" />
  <input type="hidden" name="token" value="{{.}}"/>
  <input type="submit" value="upload" />
</form>
</body>
</html>

Server 端

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 來儲存檔案。

透過上面的範例我們可以看到我們上傳檔案主要三步處理:

1、單中增加 enctype="multipart/form-data"
2、伺服器端呼叫 r.ParseMultipartForm,把上傳的檔案儲存在記憶體和臨時檔案中
3、使用 r.FormFile 取得檔案控制代碼,然後對檔案進行儲存等處理。

檔案 handler 是 multipart.FileHeader,裡面儲存了如下結構資訊

type FileHeader struct {
	Filename string
	Header   textproto.MIMEHeader
	// contains filtered or unexported fields
}

透過程式上傳檔案

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 方法寫很多其他類似的欄位。

相關函數庫