使用MultipartRequest上传文件

哈喽！大家好，很高兴又见面了，我是golang学习网的一名作者，今天由我给大家带来一篇《使用MultipartRequest上传文件》，本文主要会讲到等等知识点，希望大家一起学习进步，也欢迎大家关注、点赞、收藏、转发! 下面就一起来看看吧！

我正在使用 goa v3 设计一个端点，允许我使用 multipart/form-data post 请求上传文件（更准确地说是图像）。 我已声明以下 service：

var _ = service("images", func() {
    http(func() {
        path("/images")
    })

    method("upload", func() {  
        http(func() {
            post("/")
            multipartrequest()
        })

        payload(func() {
            description("multipart request payload")
            attribute("file", bytes, "file")
        })

        result(imagelist)
    })
})

我运行 goa gen 和 goa example 命令来生成样板代码。除了 cmd 目录之外，example 代码还生成 images.go 主文件和 multipart.go 文件来声明编码器和解码器逻辑，例如：

func ImagesUploadDecoderFunc(mr *multipart.Reader, p **images.UploadPayload) error {
    // Add multipart request decoder logic here
    return nil
}

我可以使用 mr.nextpart() 并显然获得对图像文件的引用，但我仍然不确定应该如何将其映射到 images.uploadpayload 类型中的 bytes 字段（或者也许我应该声明另一种类型的字段来处理文件？？）。

我在 goa 文档中找不到任何示例。

解决方案

好吧，我终于明白了 multipart.reader 是如何工作的，并且我想出了一个解决方案。

首先让我们澄清一下，与 goa 通常的工作方式不同（使用 payload 字段“自动”映射请求参数），使用 multipartrequest()，我必须自己进行映射，因此 payload 实际上可以有任何结构。

就我而言，我重新定义了 payload 结构，如下所示：

// imageupload single image upload element
var imageupload = type("imageupload", func() {
    description("a single image upload type")
    attribute("type", string)
    attribute("bytes", bytes)
    attribute("name", string)
})

// imageuploadpayload is a list of files
var imageuploadpayload = type("imageuploadpayload", func() {
    description("image upload payload")

    attribute("files", arrayof(imageupload), "collection of uploaded files")
})

简而言之，我想支持上传多个文件，每个文件都有其 mime 类型、文件名和数据。

为了实现这一点，我实现了 multipart.go 解码器函数，如下所示：

func ImagesUploadDecoderFunc(mr *multipart.Reader, p **images.ImageUploadPayload) error {
    res := images.ImageUploadPayload{}

    for {
        p, err := mr.NextPart()
        if err == io.EOF {
            break
        }

        if err != nil {
            fmt.Fprintln(os.Stderr, err)
            return err
        }

        _, params, err := mime.ParseMediaType(p.Header.Get("Content-Disposition"))
        if err != nil {
            // can't process this entry, it probably isn't an image
            continue
        }

        disposition, _, err := mime.ParseMediaType(p.Header.Get("Content-Type"))
        // the disposition can be, for example 'image/jpeg' or 'video/mp4'
        // I want to support only image files!
        if err != nil || !strings.HasPrefix(disposition, "image/") {
            // can't process this entry, it probably isn't an image
            continue
        }

        if params["name"] == "file" {
            bytes, err := ioutil.ReadAll(p)
            if err != nil {
                // can't process this entry, for some reason
                fmt.Fprintln(os.Stderr, err)
                continue
            }
            filename := params["filename"]
            imageUpload := images.ImageUpload{
                Type:  &disposition,
                Bytes: bytes,
                Name:  &filename,
            }
            res.Files = append(res.Files, &imageUpload)
        }
    }
    *p = &res
    return nil
}

好了，本文到此结束，带大家了解了《使用MultipartRequest上传文件》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！