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 方法把缓存传到服务器。