GO語言實現文件上傳的示例代碼

更新時間：2022年08月25日 11:13:14 作者：yi個俗人

這篇文章主要分享一下golang實現文件上傳的流程和具體代碼，供大家參考，感興趣的小伙伴們可以參考借鑒，希望對大家能有所幫助

前言

最近在寫一個文件上傳的功能，現在來進行整理總結一下go語言如何上傳文件的，本文主要分享一下golang實現文件上傳的流程和具體代碼，供大家參考，感興趣的小伙伴們可以參考借鑒，希望對大家能有所幫助。

文件上傳

表單操作

要使表單能夠上傳文件，需要添加form的enctype屬性enctype="multipart/form-data"，upload.html代碼如下：

<html>
<head>
    <title>上傳文件</title>
</head>
<body>
<form enctype="multipart/form-data" action="/upload" method="post">
  <input type="file" name="uploadfile" />
  <input type="submit" value="upload" />
</form>
</body>
</html>

服務端操作

在服務端啟動一個服務，并添加一個handlerFunc并完善相關功能即可：

package main

import (
	"fmt"
	"html/template"
	"io"
	"log"
	"net/http"
	"os"
)

func main()  {  
	http.HandleFunc("/upload", Upload) 
	if err := http.ListenAndServe("127.0.0.1:9090", nil); err != nil {
		log.Fatal(err)
	}

}

func Upload(w http.ResponseWriter, r *http.Request)  {
	//判斷請求方式
	if r.Method == "POST" {
		//設置內存大小
		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)
		//創(chuàng)建上傳目錄
		os.Mkdir("./upload", os.ModePerm)
		//創(chuàng)建上傳文件
		f, err := os.Create("./upload/" + handler.Filename)
		if err != nil {
			fmt.Println(err)
			return
		}
		defer f.Close()
		io.Copy(f, file)
		w.WriteHeader(http.StatusCreated)
		io.WriteString(w, "Uploaded successfully")
	} else {
		//渲染頁面 解析模板文件
		tpl, _ := template.ParseFiles("./upload.html")
		tpl.Execute(w, nil)
	}
}

通過http://127.0.0.1:9090/upload來測試文件上傳。截圖

選擇文件之后就會在當前目錄下的upload文件夾中看到上傳的文件了。

流程實現

通過上面的示例可以看到，處理文件上傳我們需要先調用FormFile，FormFile返回提供的表單鍵的第一個文件, 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
}

在FormFile方法中，會調用r.ParseMultipartForm， 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
}

里面的參數表示maxMemory，調用ParseMultipartForm之后，ParseMultipartForm將請求主體解析為multipart / form-data。解析整個請求體，將上傳的文件部分存儲在maxMemory大小的內存里面，如果文件大小超過了maxMemory，那么剩下的部分將存儲在系統(tǒng)的臨時文件中。我們可以通過r.FormFile獲取上面的文件句柄，然后實例中使用了io.Copy來存儲文件。