go web 处理文件上传

不念不忘少年蓝@ 2021-07-25 01:56 456阅读 0赞

# 一 点睛 #

go web中怎样实现文件上传，下面一一道来。

要使表单能够上传文件，第一步就是要添加 form 的 enctype 属性，enctype 属性有如下三种情况。

*  application/x-www-form-urlencoded：表示在发送前编码所有字符（默认）。
 *  multipart/form-data：不对字符编码。在使用包含文件上传控件的表单时，必须使用该值。
 *  text/plain：空格转换为"+"加号，但不对特殊字符编码。

# 二 实战 #

## 1 前端代码 ##

创建新的表单 html 文件，命名为 upload.html

## 2 服务器端代码 ##

package main
    
    import (
       "crypto/md5"
       "fmt"
       "html/template"
       "io"
       "log"
       "net/http"
       "os"
       "strconv"
       "time"
    )
    
    // 上传文件逻辑
    func upload(w http.ResponseWriter, r *http.Request) {
       fmt.Println("method:", r.Method) // 获取请求的方法
       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("src\\goweb\\demo6\\upload.html")
          t.Execute(w, token)
       } else {
          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)
          f, err := os.OpenFile("./test/"+handler.Filename, os.O_WRONLY|os.O_CREATE, 0666) // test目录放在 GOPATH 目录下
          if err != nil {
             fmt.Println(err)
             return
          }
          defer f.Close()
          io.Copy(f, file)
       }
    }
    
    func main() {
       http.HandleFunc("/upload", upload)       // 设置访问的路由
       err := http.ListenAndServe(":9090", nil) // 设置监听的端口
       if err != nil {
          log.Fatal("ListenAndServe: ", err)
       }
    }

## 3 说明 ##

处理文件上传需要调用 r.ParseMultipartForm，它的参数表示 maxMemory，调用 ParseMultipartForm 之后，上传的文件存储在 maxMemory 大小的内存里面，如果文件大小超过了 maxMemory，那么剩下的部分将存储在系统的临时文件中。

通过 r.FormFile 获取文件句柄，

通过 io.Copy 来存储文件。

获取其他非文件字段信息的时候就不需要调用 r.ParseForm，因为在需要的时候 Go 自动会去调用。

ParseMultipartForm 调用一次之后，后面再次调用不会再有效果。

## 4 小结 ##

上传文件三步曲：

**a 前端表单中增加 enctype="multipart/form-data"。**

**b 服务端调用 r.ParseMultipartForm，把上传的文件存储在内存和临时文件中。**

**c 使用 r.FormFile 获取文件句柄，然后对文件进行存储处理。**

文件 handler 是 multipart.FileHeader 结构

> type FileHeader struct \{
> 
>     Filename string
> 
>     Header   textproto.MIMEHeader
> 
>     // contains filtered or unexported fields
> 
> \}

## 5 测试结果 ##

![20210609190724671.png][]

# 三 客户端上传文件 #

Go 支持模拟客户端表单功能支持文件上传。

## 1 代码 ##

package main
    
    import (
       "bytes"
       "fmt"
       "io"
       "io/ioutil"
       "mime/multipart"
       "net/http"
       "os"
    )
    
    // 模拟客户端表单功能支持文件上传
    func postFile(filename string, targetUrl string) error {
       bodyBuf := &bytes.Buffer{}
       bodyWriter := multipart.NewWriter(bodyBuf)
       // 操作源文件
       fileWriter, err := bodyWriter.CreateFormFile("uploadfile", filename)
       if err != nil {
          fmt.Println("error writing to buffer")
          return err
       }
    
       // 打开文件句柄操作
       fh, err := os.Open(filename)
       if err != nil {
          fmt.Println("error opening file")
          return err
       }
       defer fh.Close()
    
       // iocopy
       _, err = io.Copy(fileWriter, fh)
       if err != nil {
          return err
       }
    
       contentType := bodyWriter.FormDataContentType()
       bodyWriter.Close()
    
       resp, err := http.Post(targetUrl, contentType, bodyBuf)
       if err != nil {
          return err
       }
       defer resp.Body.Close()
       resp_body, err := ioutil.ReadAll(resp.Body)
       if err != nil {
          return err
       }
       fmt.Println(resp.Status)
       fmt.Println(string(resp_body))
       return nil
    }
    
    // 测试
    func main() {
       target_url := "http://localhost:9090/upload"
       filename := "a.txt"
       postFile(filename, target_url)
    }

## 2 测试 ##

客户端打印如下：

> 200 OK
> 
> map\[Content-Disposition:\[form-data; name="uploadfile"; filename="a.txt"\] Content-Type:\[application/octet-stream\]\]

## 3 上传效果 ##

![20210609190840952.png][]

## 4 说明 ##

该例子展示了客户端如何向服务器上传一个文件的例子，客户端通过 multipart.Write 把文件的文本流写入一个缓存中，然后调用 http 的 Post 方法把缓存传到服务器。

[20210609190724671.png]: /images/20210724/035af7c96b10496e862efa3efdd773ac.png
[20210609190840952.png]: /images/20210724/5b124aff3e824100852ed2ce46e2f8ef.png