登录
首页 >  Golang >  Go教程

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

来源:脚本之家

时间:2022-12-23 09:17:04 475浏览 收藏

IT行业相对于一般传统行业,发展更新速度更快,一旦停止了学习,很快就会被行业所淘汰。所以我们需要踏踏实实的不断学习,精进自己的技术,尤其是初学者。今天golang学习网给大家整理了《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 

通过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语言实现文件上传的示例代码》的内容了。个人认为,基础知识的学习和巩固,是为了更好的将其运用到项目中,欢迎关注golang学习网公众号,带你了解更多关于golang的知识点!

声明:本文转载于:脚本之家 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>
评论列表