如何使用go的net / http或类似方法保存在多部分/表单数据POST请求中收到的文件？

debugcn 发表于 Dev

青年开发者

对于上下文，我正在尝试为程序ShareX（https://getsharex.com/docs/custom-uploader）创建一个自定义图像上传器。我尝试了很多搜索词，但是找不到该问题的任何解决方案。

这是我当前的代码（位于文件main.go中）：

package main

import (
    "fmt"
    "net/http"
)

func handleUpload(w http.ResponseWriter, r *http.Request) {
    r.ParseMultipartForm(16384)
    f, fi, _ := r.FormFile("file_image")
    _ = f
    fmt.Println(fi.Filename, fi.Size)

}

func main() {
    mux := http.NewServeMux()
    mux.HandleFunc("/upload", handleUpload)
    fmt.Println("Starting listener on port 8085")
    http.ListenAndServe(":8085", mux)
}

我的目标是将从请求中接收到的文件（图像）数据保存到本地文件中，但是我无法做到这一点，因为写入文件需要type[]byte且f是type multipart.File。

Moficodes

这是如何处理表单数据的示例。

这里的主要思想是使用io pkg https://golang.org/pkg/io/

r.FormFile返回type的接口File。File工具Reader和Writer方法。因此，我们可以从一个文件复制内容，然后将其写入一个空文件。

package main

import (
    "fmt"
    "io"
    "log"
    "net/http"
    "os"
)

func uploadHandler(w http.ResponseWriter, r *http.Request) {
    switch r.Method {
    case "POST":
        r.ParseMultipartForm(10 << 20) //10 MB
        file, handler, err := r.FormFile("file_image")
        if err != nil {
            log.Println("error retrieving file", err)
            http.Error(w, err.Error(), http.StatusInternalServerError)
            return
        }
        defer file.Close()
        dst, err := os.Create(handler.Filename)
        if err != nil {
            log.Println("error creating file", err)
            http.Error(w, err.Error(), http.StatusInternalServerError)
            return
        }
        defer dst.Close()
        if _, err := io.Copy(dst, file); err != nil {
            http.Error(w, err.Error(), http.StatusInternalServerError)
            return
        }
        fmt.Fprintf(w, "uploaded file")
    }
}

func main() {
    http.HandleFunc("/upload", uploadHandler)

    http.ListenAndServe(":8085", nil)
}

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。