Gin 中间件可以处理所有 HTTP 错误而不需要重复的 if 语句吗？-Golang-PHP中文网

Gin 中间件可以处理所有 HTTP 错误而不需要重复的 if 语句吗？

Barbara Streisand

发布： 2024-11-05 03:52:02

原创

401 人浏览过

Can Gin Middleware Handle All HTTP Errors Without Repetitive `if` Statements?

管理 Gin 中间件中的错误

问题

在 Gin 中，是否可以在不为每个路由编写重复的 if 语句的情况下处理每个路由上的所有 HTTP 错误400、404、500 等错误？目标是在专用的中间件函数中捕获和处理错误。

答案

是的，您可以使用中间件来集中错误处理，从而无需在路由处理程序中重复使用 if 语句。具体方法如下：

中间件实现

创建一个名为 ErrorHandler 的中间件函数：

<code class="go">func ErrorHandler(c *gin.Context) {
    // Execute remaining handlers
    c.Next()

    // Retrieve and handle errors
    for _, err := range c.Errors {
        // Log, handle, etc.
    }

    // Default handler if no other error is handled
    c.JSON(http.StatusInternalServerError, "")
}</code>

登录后复制

中间件使用

在 main 函数中注册 ErrorHandler 中间件：

<code class="go">// ...
router := gin.New()
router.Use(middleware.ErrorHandler)
// ...</code>

登录后复制

路由处理程序修改

在路由处理程序中，只需中止出现错误的请求，而不是调用外部错误处理程序：

<code class="go">func (h *Handler) List(c *gin.Context) {
    movies, err := h.service.ListService()

    if err != nil {
        // Abort the request with the error
        c.AbortWithError(http.StatusInternalServerError, err)
        return
    }

    c.JSON(http.StatusOK, movies)
}</code>

登录后复制

中间件中的错误处理

在 ErrorHandler 中间件中，检查上下文中捕获的错误：

<code class="go">for _, ginErr := range c.Errors {
    switch ginErr.Err {
        case ErrNotFound:
            c.JSON(-1, gin.H{"error": ErrNotFound.Error()})
        // ...
    }
}</code>

登录后复制

日志记录和自定义

您还可以合并日志记录或将自定义参数传递给中间件：需要：

<code class="go">func ErrorHandler(logger *zap.Logger) gin.HandlerFunc {
    return func(c *gin.Context) {
        c.Next()
        
        for _, ginErr := range c.Errors {
            logger.Error("whoops", ...)
        }
    }
}</code>

登录后复制

此外，请考虑在调用 c.JSON 时不要使用 -1 作为状态代码来覆盖错误处理程序中显式设置的 HTTP 状态。