使用 MultipartRequest 上传文件
Using MultipartRequest to upload file
我正在使用 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)
})
})
I 运行 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()
并显然获得对图像文件的引用,但我仍然不确定如何将其映射到 [=22= 中的 Bytes
字段] 类型(或者也许我应该声明另一种类型的字段来处理文件??)。
我在 Goa 文档中找不到任何示例。
好吧,我终于明白了multipart.Reader
是如何工作的,我想出了一个解决方案。
首先让我们澄清一下,与 Goa 通常的工作方式不同(将 'automatically' 请求参数映射到 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
}
我正在使用 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)
})
})
I 运行 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()
并显然获得对图像文件的引用,但我仍然不确定如何将其映射到 [=22= 中的 Bytes
字段] 类型(或者也许我应该声明另一种类型的字段来处理文件??)。
我在 Goa 文档中找不到任何示例。
好吧,我终于明白了multipart.Reader
是如何工作的,我想出了一个解决方案。
首先让我们澄清一下,与 Goa 通常的工作方式不同(将 'automatically' 请求参数映射到 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
}