GO實現文件上傳操作
本文實例為大傢分享瞭GO實現文件上傳操作的具體代碼，供大傢參考，具體內容如下
由於需求中有文件上傳這一個需求，在這裡我們就學習一下go語言如何上傳文件。本文主要通過表單的方式進行文件上傳操作。主要有以下三步：
- 表單中增加
enctype屬性 - 服務端調用
r.ParseMultipartForm,把上傳的文件存儲在內存和臨時文件中 - 使用
r.FormFile獲取文件句柄，然後對文件進行存儲等處理。
1、表單操作
要使表單能夠上傳文件，首先第一步就要添加form的enctype屬性進去，enctype屬性有如下三種情況：
application/x-www-form-urlencoded 表示在發送前編碼所有字符（默認）
multipart/form-data 不對字符編碼。在使用包含文件上傳控件的表單時，必須使用該值。
text/plain 空格轉換為 "+" 加號，但不對特殊字符編碼。
所以可以創建如下上傳表單：
<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>
2、服務端操作
在服務端隻需要添加一個handlerFunc並完善相關功能即可：
// 處理/upload 邏輯
func upload(w http.ResponseWriter, r *http.Request) {
//獲取請求的方法
fmt.Println("method:", r.Method)
//GET的處理操作
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)
//創建上傳目錄
os.Mkdir("./test", os.ModePerm)
//創建上傳文件
f, err := os.Create("./test/" + handler.Filename)
//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)
}
}
在main()函數中記得添加http.HandleFunc("/upload", upload)即可。
通過http://127.0.0.1:9999/upload來測試文件上傳。 截圖
選擇文件之後就會在當前目錄下的test文件夾中成功上傳文件。
3、流程解析
通過上面的代碼可以看到，處理文件上傳我們需要調用r.ParseMultipartForm，裡面的參數表示maxMemory，調用ParseMultipartForm之後，上傳的文件存儲在maxMemory大小的內存裡面，如果文件大小超過瞭maxMemory，那麼剩下的部分將存儲在系統的臨時文件中。我們可以通過r.FormFile獲取上面的文件句柄，然後實例中使用瞭io.Copy來存儲文件。我們可以嘗試使用看一下使用的相關原函數：
ParseMultipartForm函數如下：
func (r *Request) ParseMultipartForm(maxMemory int64) error {
if r.MultipartForm == multipartByReader {
return errors.New("http: multipart handled by MultipartReader")
}
if r.Form == nil {
err := r.ParseForm()
if err != nil {
return err
}
}
if r.MultipartForm != nil {
return nil
}
mr, err := r.multipartReader(false)
if err != nil {
return err
}
f, err := mr.ReadForm(maxMemory)
if err != nil {
return err
}
if r.PostForm == nil {
r.PostForm = make(url.Values)
}
for k, v := range f.Value {
r.Form[k] = append(r.Form[k], v...)
// r.PostForm should also be populated. See Issue 9305.
r.PostForm[k] = append(r.PostForm[k], v...)
}
r.MultipartForm = f
return nil
}
FormFile函數如下：
func (r *Request) FormFile(key string) (multipart.File, *multipart.FileHeader, error) {
if r.MultipartForm == multipartByReader {
return nil, nil, errors.New("http: multipart handled by MultipartReader")
}
if r.MultipartForm == nil {
err := r.ParseMultipartForm(defaultMaxMemory)
if err != nil {
return nil, nil, err
}
}
if r.MultipartForm != nil && r.MultipartForm.File != nil {
if fhs := r.MultipartForm.File[key]; len(fhs) > 0 {
f, err := fhs[0].Open()
return f, fhs[0], err
}
}
return nil, nil, ErrMissingFile
}
文件handler是multipart.FileHeader裡面的結構體如下
// A FileHeader describes a file part of a multipart request.
type FileHeader struct {
Filename string
Header textproto.MIMEHeader
Size int64
content []byte
tmpfile string
}
4、成功結果
瀏覽器端顯示如下消息。
以上就是本文的全部內容，希望對大傢的學習有所幫助，也希望大傢多多支持WalkonNet。