你想處理一個由用戶上傳的檔案，比如你正在建設一個類似Instagram的網站，你需要存盤用戶拍攝的照片，這種需求該如何實作呢？
要使表單能夠上傳檔案，首先第一步就是要添加form的enctype屬性，enctype屬性有如下三種情況:
application/x-www-form-urlencoded 表示在發送前編碼所有字符（默認）
multipart/form-data 不對字符編碼，在使用包含檔案上傳控制元件的表單時，必須使用該值，
text/plain 空格轉換為 "+" 加號，但不對特殊字符編碼，
所以，表單的html代碼應該類似于:
<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>
在服務器端，我們增加一個handlerFunc:
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)
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呼叫一次之后，后面再次呼叫不會再有效果，
通過上面的實體我們可以看到我們上傳檔案主要三步處理：
- 表單中增加enctype="multipart/form-data"
- 服務端呼叫
r.ParseMultipartForm,把上傳的檔案存盤在記憶體和臨時檔案中 - 使用
r.FormFile獲取檔案句柄，然后對檔案進行存盤等處理，
檔案handler是multipart.FileHeader,里面存盤了如下結構資訊
type FileHeader struct {
Filename string
Header textproto.MIMEHeader
// contains filtered or unexported fields
}
客戶端上傳檔案
我們上面的例子演示了如何通過表單上傳檔案，然后在服務器端處理檔案，其實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方法寫很多其他類似的欄位，
轉載請註明出處，本文鏈接：https://www.uj5u.com/ruanti/258053.html
標籤：其他