go web 处理文件上传

2022年11月5日12:27:11

一 点睛

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

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

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

二 实战

1 前端代码

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

<html>
<head>
    <title>上传文件</title>
</head>
<body>
<form enctype="multipart/form-data" action="/upload" method="post">
  <input type="file" name="uploadfile" />
  <input type="hidden" name="token" value="{{.}}"/>
  <input type="submit" value="upload" />
</form>
</body>
</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 测试结果

三 客户端上传文件

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 上传效果

4 说明

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

  • 作者：chengqiuming
  • 原文链接：https://blog.csdn.net/chengqiuming/article/details/117752133
    更新时间：2022年11月5日12:27:11 ，共 3036 字。