使用MultipartRequest上传文件

Question

我正在使用 Goa v3 设计一个端点，允许我通过 POST 请求上传文件（更准确地说是图像）multipart/form-data。我已声明如下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()并获取对图像文件的引用，但我仍然不确定应该如何将其映射到类型Bytes中的字段images.UploadPayload（或者也许我应该声明另一种类型的字段来处理文件？？）。我在果阿文档中找不到任何示例。

Answer 1

好吧，我终于明白了它是如何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

}