Echo web 框架绑定 FormFile

Question

我正在将 Echo 框架用于接受表单数据的 post 端点。我使用 Struct 作为绑定模型来提取表单数据。我的绑定模型和上传处理程序代码如下所示。 type FormModel struct {    ID string                `form:"ID"`    FirstName string                `form:"FirstName"`    File      *multipart.FileHeader `form:"myFileName"`}func (cs *handler) uploadForm(c echo.Context) error {s := new(FormModel)if err := c.Bind(s); err != nil {    return nil}fileHandler, err := c.FormFile("myFileName")我可以通过绑定获得 ID 和 FirstName 等表单值。但我无法在绑定期间获取文件。我必须使用fileHandler, err := c.FormFile("myFileName")来获取文件。有没有办法在绑定模型中获取文件信息？

Answer 1

echo 默认不支持绑定 multipart.Form.File 数据，需要重新绑定实现接口。

为 echo Bind 封装了一层额外的 bind FormFile。如果结构指针类型的属性类型为*multipart.FileHeader 或[]*multipart.FileHeader，则该属性将通过反射设置为FormFile 的值。

我大概实现了这个功能。我没有使用过echo，也没有测试过，但想法是正确的。

最后更新：添加示例并修复代码以构建。感谢@vicTROLLA 指出typeMultipartFileHeader类型定义错误。

package main

import (

    "bytes"

    "fmt"

    "github.com/labstack/echo"

    "io/ioutil"

    "mime/multipart"

    "net/http"

    "reflect"

    "strings"

    "time"

)

var (

    typeMultipartFileHeader      = reflect.TypeOf((*multipart.FileHeader)(nil))

    typeMultipartSliceFileHeader = reflect.TypeOf(([]*multipart.FileHeader)(nil))

)

type File struct {

    File  *multipart.FileHeader   `form:"file"`

    Files []*multipart.FileHeader `form:"files"`

}

func main() {

    app := echo.New()

    // warp bind suppet bind FormFile

    app.Binder = NewBindFile(app.Binder)

    app.Any("/file", func(ctx echo.Context) error {

        var file File

        err := ctx.Bind(&file)

        if err != nil {

            return err

        }

        readfile := func(file *multipart.FileHeader) {

            f, err := file.Open()

            if err != nil {

                fmt.Printf("open %s error: %s\n", file.Filename, err.Error())

                return

            }

            body, err := ioutil.ReadAll(f)

            fmt.Printf("read file %s error: %v body: %s\n", file.Filename, err, body)

        }

        readfile(file.File)

        for _, file := range file.Files {

            readfile(file)

        }

        return err

    })

    go func() {

        time.Sleep(200 * time.Millisecond)

        buf := bytes.NewBuffer(nil)

        w := multipart.NewWriter(buf)

        part, _ := w.CreateFormFile("file", "file")

        part.Write([]byte("this one file"))

        part, _ = w.CreateFormFile("files", "files")

        part.Write([]byte("fils part 1"))

        part, _ = w.CreateFormFile("files", "files")

        part.Write([]byte("fils part 2"))

        part, _ = w.CreateFormFile("files", "files")

        part.Write([]byte("fils part 3"))

        part, _ = w.CreateFormFile("files", "files")

        part.Write([]byte("fils part 4"))

        w.Close()

        http.Post("http://localhost:1323/file", w.FormDataContentType(), buf)

    }()

    app.Start(":1323")

}

type BindFunc func(interface{}, echo.Context) error

func (fn BindFunc) Bind(i interface{}, ctx echo.Context) error {

    return fn(i, ctx)

}

func NewBindFile(b echo.Binder) echo.Binder {

    return BindFunc(func(i interface{}, ctx echo.Context) error {

        err := b.Bind(i, ctx)

        if err == nil {

            ctype := ctx.Request().Header.Get(echo.HeaderContentType)

            // if bind form

            if strings.HasPrefix(ctype, echo.MIMEApplicationForm) || strings.HasPrefix(ctype, echo.MIMEMultipartForm) {

                // get form files

                var form *multipart.Form

                form, err = ctx.MultipartForm()

                if err == nil {

                    err = EchoBindFile(i, ctx, form.File)

                }

            }

        }

        return err

    })

}

func EchoBindFile(i interface{}, ctx echo.Context, files map[string][]*multipart.FileHeader) error {

    iValue := reflect.Indirect(reflect.ValueOf(i))

    // check bind type is struct pointer

    if iValue.Kind() != reflect.Struct {

        return fmt.Errorf("BindFile input not is struct pointer, indirect type is %s", iValue.Type().String())

    }

    iType := iValue.Type()

    for i := 0; i < iType.NumField(); i++ {

        fType := iType.Field(i)

        // check canset field

        fValue := iValue.Field(i)

        if !fValue.CanSet() {

            continue

        }

        // revc type must *multipart.FileHeader or []*multipart.FileHeader

        switch fType.Type {

        case typeMultipartFileHeader:

            file := getFiles(files, fType.Name, fType.Tag.Get("form"))

            if len(file) > 0 {

                fValue.Set(reflect.ValueOf(file[0]))

            }

        case typeMultipartSliceFileHeader:

            file := getFiles(files, fType.Name, fType.Tag.Get("form"))

            if len(file) > 0 {

                fValue.Set(reflect.ValueOf(file))

            }

        }

    }

    return nil

}

func getFiles(files map[string][]*multipart.FileHeader, names ...string) []*multipart.FileHeader {

    for _, name := range names {

        file, ok := files[name]

        if ok {

            return file

        }

    }

    return nil

}