如何使用 Gin 处理多部分/表单数据请求中的 JSON 数据和图像？

Go：使用 Gin 处理表单中的 JSON 数据和图像

Gin 是 Go 中一个流行的 Web 框架，可简化 HTTP 请求的处理。本文解决了使用 Gin 的绑定机制解析来自 multipart/form-data 请求的 JSON 数据和图像的特定问题。

请求处理代码

请求Gin 中的 handler 函数接收并处理 HTTP 请求。在这种情况下，它期望收到包含 JSON 数据和图像文件的多部分请求。

func (h *Handlers) UpdateProfile() gin.HandlerFunc {
    type request struct {
        Username    string `json:"username" binding:"required,min=4,max=20"`
        Description string `json:"description" binding:"required,max=100"`
    }

    return func(c *gin.Context) {
        var updateRequest request

        // Bind JSON data to `updateRequest` struct.
        if err := c.BindJSON(&updateRequest); err != nil {
            // Handle error here...
            return
        }

        // Get the image file from the request.
        avatar, err := c.FormFile("avatar")
        if err != nil {
            // Handle error here...
            return
        }

        // Validate file size and content type.
        if avatar.Size > 3<<20 || !avatar.Header.Get("Content-Type") { // if avatar size more than 3mb
            // Handle error here...
            return
        }

        // Handle image processing and database operations here...
        // Save username, description, and image to a database.

        c.IndentedJSON(http.StatusNoContent, gin.H{"message": "successful update"})
    }
}

测试用例

包含单元测试来验证处理程序的功能。它使用多部分/表单数据主体设置模拟请求。请求包含 JSON 数据和图像。

func TestUser_UpdateProfile(t *testing.T) {
    type testCase struct {
        name               string
        image              io.Reader
        username           string
        description        string
        expectedStatusCode int
    }

    // Set up mock request with multipart/form-data body.
    // ...

    for _, tc := range testCases {
        // ...

        w := httptest.NewRecorder()
        router.ServeHTTP(w, req)

        assert.Equal(t, tc.expectedStatusCode, w.Result().StatusCode)
    }
}

测试期间出错

测试期间，由于请求正文中存在无效字符而发生错误。错误消息是“错误#01：数字文字中的字符'-'无效。”

根本原因

Gin 的 c.BindJSON 函数用于解析 JSON 数据来自请求正文。但是，它假设请求正文以有效的 JSON 开头。在 multipart/form-data 请求的情况下，正文以边界 (--30b24345de...) 开头，这对于 JSON 文字来说是无效字符。

解决方案

为了解决这个问题，我们可以使用 Gin 的 c.ShouldBind 函数和 binding.FormMultipart 来显式绑定到多部分/表单数据体。这使得 Gin 能够正确解析 JSON 数据和图像文件。

// Bind JSON data and image file to `updateRequest` struct.
if err := c.ShouldBindWith(&updateRequest, binding.FormMultipart); err != nil {
    // Handle error here...
    return
}

结论

本文演示了如何同时处理 JSON 数据和图像文件在使用 Gin 的多部分/表单数据请求中。它强调了使用正确的绑定方法（c.ShouldBindWith(..., binding.FormMultipart)）以避免解析错误的重要性。

以上是如何使用 Gin 处理多部分/表单数据请求中的 JSON 数据和图像？的详细内容。更多信息请关注PHP中文网其他相关文章！