使用Gin框架上传文件：轻松上手指南

2023-11-14 03:12:45

在 Gin 框架中轻松实现文件上传

在现代 Web 应用程序中，文件上传至关重要。Gin 框架提供了对文件上传的全面支持，使开发者可以轻松实现各种场景下的文件上传功能。

单个文件上传

1. 表单提交

首先，创建 HTML 表单并添加一个 file 类型的 input 元素，该元素用于选择要上传的文件。

<form action="/upload" method="post" enctype="multipart/form-data">
  <input type="file" name="file" />
  <input type="submit" value="上传" />
</form>

2. 路由定义

在 Gin 路由器中，定义一个处理文件上传请求的路由。通过 FormFile() 方法获取上传的文件。

func UploadFile(c *gin.Context) {
  file, err := c.FormFile("file")
  if err != nil {
    c.JSON(http.StatusBadRequest, gin.H{
      "error": err.Error(),
    })
    return
  }

  // 保存文件
}

3. 文件保存

使用 SaveUploadedFile() 方法将文件保存到指定路径。

if err := c.SaveUploadedFile(file, "uploads/"+file.Filename); err != nil {
  c.JSON(http.StatusInternalServerError, gin.H{
    "error": err.Error(),
  })
  return
}

多个文件上传

1. 表单提交

与单个文件上传类似，但使用 multiple 属性允许上传多个文件。

<form action="/upload" method="post" enctype="multipart/form-data">
  <input type="file" name="files" multiple />
  <input type="submit" value="上传" />
</form>

2. 路由定义

使用 FormFiles() 方法获取上传的多个文件。

func UploadFiles(c *gin.Context) {
  files, err := c.FormFiles("files")
  if err != nil {
    c.JSON(http.StatusBadRequest, gin.H{
      "error": err.Error(),
    })
    return
  }

  // 保存文件
}

3. 文件保存

使用 SaveUploadedFile() 方法依次保存每个文件。

for _, file := range files {
  if err := c.SaveUploadedFile(file, "uploads/"+file.Filename); err != nil {
    c.JSON(http.StatusInternalServerError, gin.H{
      "error": err.Error(),
    })
    return
  }
}

表单数据

除了文件上传，还可以通过 Form() 方法获取表单中的其他数据。

func FormData(c *gin.Context) {
  name := c.PostForm("name")
  email := c.PostForm("email")
  // ...
}

文件类型检查

可以通过以下方法检查上传文件类型：

IsDir(): 检查文件是否为目录
Size(): 检查文件大小
Ext(): 检查文件后缀名

func CheckFileType(c *gin.Context) {
  file, err := c.FormFile("file")
  if err != nil {
    c.JSON(http.StatusBadRequest, gin.H{
      "error": err.Error(),
    })
    return
  }

  if file.IsDir() {
    c.JSON(http.StatusBadRequest, gin.H{
      "error": "文件不能是目录",
    })
    return
  }

  if file.Size > 1024*1024 {
    c.JSON(http.StatusBadRequest, gin.H{
      "error": "文件不能大于1MB",
    })
    return
  }

  if file.Ext != ".jpg" && file.Ext != ".png" && file.Ext != ".gif" {
    c.JSON(http.StatusBadRequest, gin.H{
      "error": "文件类型不正确",
    })
    return
  }
}

文件大小限制

可以通过 MaxFileSize() 方法设置文件大小限制。

func LimitFileSize(c *gin.Context) {
  c.MaxFileSize = 1024 * 1024 // 1MB

  file, err := c.FormFile("file")
  if err != nil {
    c.JSON(http.StatusBadRequest, gin.H{
      "error": err.Error(),
    })
    return
  }

  if file.Size > c.MaxFileSize {
    c.JSON(http.StatusBadRequest, gin.H{
      "error": "文件不能大于1MB",
    })
    return
  }
}

文件保存路径

可以使用 SaveUploadedFile() 方法的第二个参数指定文件保存路径。

func SetSavePath(c *gin.Context) {
  file, err := c.FormFile("file")
  if err != nil {
    c.JSON(http.StatusBadRequest, gin.H{
      "error": err.Error(),
    })
    return
  }

  if err := c.SaveUploadedFile(file, "uploads/"+file.Filename); err != nil {
    c.JSON(http.StatusInternalServerError, gin.H{
      "error": err.Error(),
    })
    return
  }
}