GO實現(xiàn)文件上傳操作

更新時間：2022年07月26日 08:35:08 作者：Rookie_Hang

這篇文章主要為大家詳細介紹了GO實現(xiàn)文件上傳操作，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下

本文實例為大家分享了GO實現(xiàn)文件上傳操作的具體代碼，供大家參考，具體內(nèi)容如下

由于需求中有文件上傳這一個需求，在這里我們就學習一下go語言如何上傳文件。本文主要通過表單的方式進行文件上傳操作。主要有以下三步：

表單中增加enctype屬性
服務端調(diào)用r.ParseMultipartForm,把上傳的文件存儲在內(nèi)存和臨時文件中
使用r.FormFile獲取文件句柄，然后對文件進行存儲等處理。

1、表單操作

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

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

所以可以創(chuàng)建如下上傳表單：

<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 {
?? ??? ?//設置內(nèi)存大小
?? ??? ?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("./test", os.ModePerm)
?? ??? ?//創(chuàng)建上傳文件
?? ??? ?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()函數(shù)中記得添加http.HandleFunc("/upload", upload)即可。

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

選擇文件之后就會在當前目錄下的test文件夾中成功上傳文件。

3、流程解析

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

ParseMultipartForm函數(shù)如下：

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函數(shù)如下：

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、成功結果

瀏覽器端顯示如下消息。