Использование MultipartRequest для загрузки файла - PullRequest
Новая
       27

Использование MultipartRequest для загрузки файла

0 голосов
/ 08 октября 2019

Я использую 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() и, по-видимому, получить ссылку на файл изображения, но я все еще не уверен, как мне сопоставить это с полем Bytes в типе images.UploadPayload (или, может быть, мне следует объявить другой тип поля для обработки файлов? ?).

Я не могу найти ни одного примера в документации по Гоа.

1 Ответ

0 голосов
/ 13 октября 2019

Хорошо, я наконец понял, как работает multipart.Reader, и я нашел решение.

Сначала давайте выясним, что отличается от того, как обычно работает Гоа (отображение «автоматически» параметров запросов с помощью 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
}
...