Gin框架文件上传

Gin是一款轻量级的Web框架，它基于Golang语言，提供高效的路由处理和中间件，使得构建Web应用程序更加容易。

在这篇文章中，我们将使用Gin框架来实现文件上传功能，具体步骤如下：

1. 创建项目

我们首先需要创建一个Gin项目，可以使用如下命令：

go mod init example.com/upload
go get -u github.com/gin-gonic/gin

然后在项目根目录下，创建一个main.go文件，用于编写Gin的路由处理和中间件。

2. 编写路由处理

Gin框架的路由处理非常简单，我们可以通过以下代码实现文件上传的路由：

func main() {
    r := gin.Default()

    r.POST("/upload", func(c *gin.Context) {
        // TODO: 实现文件上传代码
    })

    r.Run() // listen and serve on 0.0.0.0:8080
}

该代码创建了一个POST方法的/upload路由，我们需要在其中实现文件上传逻辑。

3. 实现文件上传逻辑

我们可以通过Gin的Context对象获取上传的文件，实现文件上传如下：

func main() {
    r := gin.Default()

    r.POST("/upload", func(c *gin.Context) {
        file, err := c.FormFile("file")
        if err != nil {
            c.String(http.StatusBadRequest, fmt.Sprintf("获取上传文件失败 %s", err.Error()))
            return
        }

        // 保存文件到本地
        err = c.SaveUploadedFile(file, file.Filename)
        if err != nil {
            c.String(http.StatusBadRequest, fmt.Sprintf("保存上传文件失败 %s", err.Error()))
            return
        }

        c.String(http.StatusOK, fmt.Sprintf("上传文件 %s 成功", file.Filename))
    })

    r.Run() // listen and serve on 0.0.0.0:8080
}

在以上代码中，我们首先通过FormFile方法获取上传的文件，然后可以对文件进行操作，例如将文件保存到本地。

最后，我们通过String方法返回上传文件的结果。

当然, 你还可以使用更加简单的方式完成这个功能

• 定义MyFile结构体如下

# "mime/multipart"包
type MyFile struct {
   File *multipart.FileHeader `form:"file" binding:"required"`
}

• 定义BindFile函数

func BindFile(c *gin.Context, obj *MyFile) error {
   if err := c.ShouldBind(obj); err != nil {
      return err
   }

   // 文件校验逻辑, 例如文件大小, 文件类型等
   // ...

   return nil
}

• 修改/upload路由函数

r.POST("/upload", func(c *gin.Context) {
   var myFile MyFile
   if err := BindFile(c, &myFile); err != nil {
      c.JSON(500, gin.H{
         "message": err.Error(),
      })
      return
   }
   c.JSON(200, gin.H{
      "message": "ok",
   })

   // 保存文件
   c.SaveUploadedFile(myFile.File, myFile.File.Filename)
})

该方式是通过ShouldBind方法自动绑定数据, 前一章节Gin框架获取请求参数的各种方式详解有提到, 这里不再赘述,

4. 测试文件上传功能

我们可以使用Postman或curl等工具来测试文件上传功能，例如：

curl -X POST http://localhost:8080/upload \
    -H "Content-Type: multipart/form-data" \
    -F "file=@/path/to/local/file"

以上代码会将/path/to/local/file本地文件上传到http://localhost:8080/upload路由中

总结

通过上述步骤，我们可以基于Gin框架实现文件上传，Gin框架具有简单的路由处理和中间件能力，使得构建Web应用程序更加容易。

完整代码

package main

import (
   "github.com/gin-gonic/gin"
   "mime/multipart"
)

type MyFile struct {
   File *multipart.FileHeader `form:"file" binding:"required"`
}

func BindFile(c *gin.Context, obj *MyFile) error {
   if err := c.ShouldBind(obj); err != nil {
      return err
   }

   // 文件校验逻辑, 例如文件大小, 文件类型等
   // ...

   return nil
}

func main() {
   r := gin.Default()
   r.GET("/ping", func(c *gin.Context) {
      c.JSON(200, gin.H{
         "message": "pong",
      })
   })

   r.POST("/upload", func(c *gin.Context) {
      var myFile MyFile
      if err := BindFile(c, &myFile); err != nil {
         c.JSON(500, gin.H{
            "message": err.Error(),
         })
         return
      }
      c.JSON(200, gin.H{
         "message": "ok",
      })

      // 保存文件
      c.SaveUploadedFile(myFile.File, myFile.File.Filename)
   })

   r.Run() // listen and serve on
}