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)
		//创建上传目录
		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语言文件上传内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

Golang使用panic控制程序错误流程
这篇文章主要介绍了Golang使用panic控制程序错误流程，Golang panic异常处理机制中的一种流程控制方式，用于中断程序流程并触发异常处理
2023-04-04
golang rate令牌桶源码分析实现方式
这篇文章主要介绍了golang rate令牌桶源码分析实现方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2023-07-07
Go 使用xorm操作mysql详情
这篇文章主要介绍了Go 使用xorm操作mysql详情，golang orm 库 xorm 的使用和项目结构。更多详细neural，需要的小伙伴可以参考下面文章内容
2022-01-01
go语言题解LeetCode1275找出井字棋的获胜者示例
这篇文章主要为大家介绍了go语言题解LeetCode1275找出井字棋的获胜者示例，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2023-01-01
Golang实现自己的Redis(pipeline客户端)实例探索
这篇文章主要为大家介绍了Golang实现自己的Redis(pipeline客户端)实例探索,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
2024-01-01
详解Golang中Context的三个常见应用场景
Golang context主要用于定义超时取消，取消后续操作，在不同操作中传递值。本文通过简单易懂的示例进行说明，感兴趣的可以了解一下
2022-12-12
浅谈Go语言中的接口类型
Go语言中接口是一种抽象的类型,本文主要介绍了浅谈Go语言中的接口类型,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
2024-08-08
Go 结构体、数组、字典和 json 字符串的相互转换方法
今天小编就为大家分享一篇Go 结构体、数组、字典和 json 字符串的相互转换方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-08-08
go语言变量定义用法实例
这篇文章主要介绍了go语言变量定义用法,实例分析了go语言变量的定义及使用技巧,具有一定参考借鉴价值,需要的朋友可以参考下
2015-02-02
Go语言中内建函数make的使用
在Go语言编程中,make函数是一个重要的内建函数,它用于创建和初始化切片、映射和通道,握 make 的使用方法,可以帮助我们更高效地管理内存和数据结构
2024-09-09