GO語言實現檔案上傳的範例程式碼

2022-08-25 14:01:43

前言

最近在寫一個檔案上傳的功能，現在來進行整理總結一下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)
		//建立上傳目錄
		os.Mkdir("./upload", os.ModePerm)
		//建立上傳檔案
		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，那麼剩下的部分將儲存在系統的臨時檔案中。我們可以通過r.FormFile獲取上面的檔案控制程式碼，然後範例中使用了io.Copy來儲存檔案。

小結

檔案上傳主要有以下三步：

表單中增加enctype屬性
伺服器端呼叫r.ParseMultipartForm,把上傳的檔案儲存在記憶體和臨時檔案中
使用r.FormFile獲取檔案控制程式碼，然後對檔案進行儲存等處理。

到此這篇關於GO語言實現檔案上傳的範例程式碼的文章就介紹到這了,更多相關GO語言檔案上傳內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！

GO語言實現檔案上傳的範例程式碼

目錄

前言

檔案上傳

表單操作

伺服器端操作

流程實現

小結

熱門文章