Go 處理文件上傳

2022-05-13 17:48 更新

你想處理一個(gè)由用戶上傳的文件，比如你正在建設(shè)一個(gè)類似Instagram的網(wǎng)站，你需要存儲(chǔ)用戶拍攝的照片。這種需求該如何實(shí)現(xiàn)呢？

要使表單能夠上傳文件，首先第一步就是要添加form的enctype屬性，enctype屬性有如下三種情況:

application/x-www-form-urlencoded   表示在發(fā)送前編碼所有字符（默認(rèn)）
multipart/form-data   不對(duì)字符編碼。在使用包含文件上傳控件的表單時(shí)，必須使用該值。
text/plain    空格轉(zhuǎn)換為 "+" 加號(hào)，但不對(duì)特殊字符編碼。

所以，表單的html代碼應(yīng)該類似于:

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

在服務(wù)器端，我們增加一個(gè)handlerFunc:

http.HandleFunc("/upload", upload)

// 處理/upload 邏輯
func upload(w http.ResponseWriter, r *http.Request) {
    fmt.Println("method:", r.Method) //獲取請(qǐng)求的方法
    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)
        if err != nil {
            fmt.Println(err)
            return
        }
        defer f.Close()
        io.Copy(f, file)
    }
}

通過上面的代碼可以看到，處理文件上傳我們需要調(diào)用r.ParseMultipartForm，里面的參數(shù)表示maxMemory，調(diào)用ParseMultipartForm之后，上傳的文件存儲(chǔ)在maxMemory大小的內(nèi)存里面，如果文件大小超過了maxMemory，那么剩下的部分將存儲(chǔ)在系統(tǒng)的臨時(shí)文件中。我們可以通過r.FormFile獲取上面的文件句柄，然后實(shí)例中使用了io.Copy來存儲(chǔ)文件。

獲取其他非文件字段信息的時(shí)候就不需要調(diào)用r.ParseForm，因?yàn)樵谛枰臅r(shí)候Go自動(dòng)會(huì)去調(diào)用。而且ParseMultipartForm調(diào)用一次之后，后面再次調(diào)用不會(huì)再有效果。

通過上面的實(shí)例我們可以看到我們上傳文件主要三步處理：

  1. 表單中增加enctype="multipart/form-data"
  2. 服務(wù)端調(diào)用r.ParseMultipartForm,把上傳的文件存儲(chǔ)在內(nèi)存和臨時(shí)文件中
  3. 使用r.FormFile獲取文件句柄，然后對(duì)文件進(jìn)行存儲(chǔ)等處理。

文件handler是multipart.FileHeader,里面存儲(chǔ)了如下結(jié)構(gòu)信息

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

我們通過上面的實(shí)例代碼打印出來上傳文件的信息如下


客戶端上傳文件

我們上面的例子演示了如何通過表單上傳文件，然后在服務(wù)器端處理文件，其實(shí)Go支持模擬客戶端表單功能支持文件上傳，詳細(xì)用法請(qǐng)看如下示例：

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)

	//關(guān)鍵的一步操作
	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)
}

上面的例子詳細(xì)展示了客戶端如何向服務(wù)器上傳一個(gè)文件的例子，客戶端通過multipart.Write把文件的文本流寫入一個(gè)緩存中，然后調(diào)用http的Post方法把緩存?zhèn)鞯椒?wù)器。

如果你還有其他普通字段例如username之類的需要同時(shí)寫入，那么可以調(diào)用multipart的WriteField方法寫很多其他類似的字段。

以上內(nèi)容是否對(duì)您有幫助：
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號(hào)
微信公眾號(hào)

編程獅公眾號(hào)