GO语言实现文件上传代码示例

本篇文章小编给大家分享一下GO语言实现文件上传代码示例，小编觉得挺不错的，现在分享给大家供大家参考，有需要的小伙伴们可以来看看。

文件上传

表单操作

要使表单能够上传文件，需要添加form的enctype属性enctype="multipart/form-data"，upload.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来存储文件。