百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术文章 > 正文

gin框架中间件详解(gin的中间件)

ccwgpt 2025-04-05 17:17 31 浏览 0 评论

中间件是一种可以拦截HTTP请求-响应生命周期的特殊函数,可以在请求处理过程中加入用户自己的钩子函数。Gin框架提供了一些默认的中间件,如Logger和Recovery,也允许用户自定义中间件

使用中间件的好处有:

  • 实现了逻辑的解耦和分离,每个中间件只负责各自的功能
  • 可以实现一些通用的功能,如日志输出、错误处理、身份验证等
  • 可以根据不同的路由或分组选择不同的中间件,实现灵活的控制

全局中间件:可以使用r.Use()方法注册一个全局中间件,它会对所有的请求生效。例如:

package main
import (
    "fmt"
    "net/http"
    "github.com/gin-gonic/gin"
)
func middleware1() gin.HandlerFunc {
    return func(c *gin.Context) {
        fmt.Println("这是中间件")
        c.Next()
    }
}
func main() {
    r := gin.Default()
    //注册一个全局中间件
    r.Use(middleware1())
    r.GET("/test", func(c *gin.Context) {
        fmt.Println("陈王昔时宴平乐,斗酒十千恣欢谑")
        c.JSON(http.StatusOK, gin.H{
            "msg": "主人何为言少钱,径须沽取对君酌",
        })
    })
    r.Run(":8080")
}

路由分组中间件:可以使用r.Group()方法创建一个路由分组,并在该分组上注册中间件,它只会对该分组下的请求生效。例如:

package main
import (
    "fmt"
    "github.com/gin-gonic/gin"
)
func middleware1() gin.HandlerFunc {
    return func(c *gin.Context) {
        fmt.Println("这是中间件")
        c.Next()
    }
}
func main() {
    r := gin.Default()
    v1 := r.Group("v1")
    v2 := r.Group("v2")
    //注册一个路由分组中间件
    v1.Use(middleware1())
    v1.GET("/test", func(c *gin.Context) {
        c.JSON(200, gin.H{
            "msg": "这里是v1-test",
        })
    })
    v2.GET("/test", func(c *gin.Context) {
        c.JSON(200, gin.H{
            "msg": "这里是v2-test",
        })
    })
    r.Run(":8080")
}

单个路由中间件:可以在注册单个路由时,传入多个处理函数,第一个处理函数就是中间件,它只会对该路由的请求生效。例如:

package main
import (
    "fmt"
    "net/http"
    "github.com/gin-gonic/gin"
)
func middleware1() gin.HandlerFunc {
    return func(c *gin.Context) {
        fmt.Println("这里是中间件")
        c.Next()
    }
}
func main() {
    r := gin.Default()
    //注册一个单个路由中间件
    r.GET("/", middleware1(), func(c *gin.Context) {
        fmt.Println("古来圣贤皆寂寞")
        c.JSON(http.StatusOK, "惟有饮者留其名")
    })
    r.Run(":8080")
}

在gin框架中,编写自定义的中间件函数的方法是定义一个返回值为gin.HandlerFunc类型的函数,并在该函数中接收一个*gin.Context类型的参数,用于获取请求和响应对象。例如:

// 定义一个自定义的日志记录中间件
func Logger() gin.HandlerFunc {
    return func(c *gin.Context) {
        // 获取请求路径和方法
        path := c.Request.URL.Path
        method := c.Request.Method
        // 打印到控制台
        fmt.Printf("路径: %s, 方法: %s\n", path, method)
        // 调用下一个中间件或路由处理函数
        c.Next()
    }
}

Next()函数:

在gin框架中,next()是一个中间件函数中的一个函数,用于将控制权传递给下一个中间件函数或路由处理程序。当遇到c.Next()函数时,它取出所有的没被执行过的注册的函数都执行一遍,然后再回到本函数中。没遇到c.Next()函数将结束当前中间件程序。

package main
import (
    "fmt"
    "github.com/gin-gonic/gin"
)
func middleware1() gin.HandlerFunc {
    return func(c *gin.Context) {
        fmt.Println("中间件1开始")
        c.Next()
        fmt.Println("中间件1结束")
    }
}
func middleware2() gin.HandlerFunc {
    return func(c *gin.Context) {
        fmt.Println("中间件2开始")
        c.Next()
        fmt.Println("中间件2结束")
    }
}
func middleware3() gin.HandlerFunc {
    return func(c *gin.Context) {
        fmt.Println("中间件3开始")
        c.Next()
        fmt.Println("中间件3结束")
    }
}
func main() {
    r := gin.Default()
    r.Use(middleware1(), middleware2(), middleware3())
    r.GET("/", func(c *gin.Context) {
        fmt.Println("古来圣贤皆寂寞")
        c.JSON(200, gin.H{
            "data": "惟有饮者留其名",
        })
    })
    r.Run(":8080")
}

以上示例中,当中间件1中遇到c.Next(),将控制权传递给中间件2,再次遇到c.Next()函数将控制权传递给中间3,中间件3将控制权传递给路由处理程序;然后再回到中间件3,执行c.Next()后的程序,中间件3结束,后面一次是中间件2、中间件1结束。

备注中间件1中的c.Next()函数,程序执行结果如下:

func middleware1() gin.HandlerFunc {
    return func(c *gin.Context) {
        fmt.Println("中间件1开始")
        // c.Next()
        fmt.Println("中间件1结束")
    }
}

中间件1中没使用c.Next()函数,将直接执行完毕本函数,再依次执行注册的中间件函数。

Abort()函数:

在gin框架中,abort()是Context对象的一个方法,用于立即终止当前请求并返回一个指定的HTTP状态码和错误信息。

func middleware2() gin.HandlerFunc {
    return func(c *gin.Context) {
        fmt.Println("中间件2开始")
        if 3 < 4 {
            c.Abort()
        }
        c.Next()
        fmt.Println("中间件2结束")
    }
}

在中间件2中使用Abort()函数,执行结果如下:

中间件1开始 
中间件2开始 
中间件2结束 
中间件1结束

这个方法函数常用于处理一些非法请求或者权限校验失败的场景。例如:

func AuthMiddleware() gin.HandlerFunc {
    return func(c *gin.Context) {
        // 检查用户是否登录
        if !IsLoggedIn(c) {
            c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"error": "未登录"})
            return
        }
        // 认证通过,继续执行后续的中间件和处理函数
        c.Next()
    }
}

在这个示例中,AuthMiddleware中间件会检查用户是否已登录,如果未登录,则使用abort()方法立即返回一个状态码为401的错误响应,否则继续执行后续的中间件和处理函数。

内置BasicAuth中间件

BasicAuth中间件是一个用于基本认证的中间件,它会对请求进行基本认证,如果认证失败,则返回401 Unauthorized错误响应。

package main

import (
    "net/http"

    "github.com/gin-gonic/gin"
)

var map_data map[string]interface{} = map[string]interface{}{
    "xiaoming": gin.H{"age": 18, "addr": "xxx"},
    "xiaohong": gin.H{"age": 20, "addr": "xxxx"},
}

func main() {
    r := gin.Default()
    // gin.Accounts是gin框架中用于存储用户账号和密码的一种数据结构。
    // 它是一个map[string]string类型的变量,其中键表示用户名,值表示密码。
    // 它通常用于BasicAuth中间件中,用于验证用户的用户名和密码。
    r.GET("/", gin.BasicAuth(gin.Accounts{
        "xiaotu": "123456",
    }), func(c *gin.Context) {
        user_name := c.Query("user_name")
        data, ok := map_data[user_name]
        if ok {
            c.JSON(http.StatusOK, gin.H{
                "user": user_name,
                "data": data,
            })
        } else {
            c.JSON(http.StatusNotFound, gin.H{
                "user": user_name,
                "data": "没有数据",
            })
        }
    })
    r.Run(":8080")
}

使用账户xiaotu和密码123456登录,进行授权认证通过,访问路径中user_name参数为xiaoming和xiaohong的时候,获取对应的信息。

访问路径中user_name参数为xiaohong111的时候,没有对应的数据。

在这个示例中,我们使用了BasicAuth中间件,只有在提供正确的用户名和密码时,才能访问该接口。在这里,我们使用了一个固定的用户名和密码。

相关推荐

一个基于.Net Core遵循Clean Architecture原则开源架构

今天给大家推荐一个遵循CleanArchitecture原则开源架构。项目简介这是基于Asp.netCore6开发的,遵循CleanArchitecture原则,可以高效、快速地构建基于Ra...

AI写代码翻车无数次,我发现只要提前做好这3步,bug立减80%

写十万行全是bug之后终于找到方法了开发"提示词管理助手"新版本那会儿,我差点被bug整崩溃。刚开始两周,全靠AI改代码架构,结果十万行程序漏洞百出。本来以为AI说没问题就稳了,结果...

OneCode低代码平台的事件驱动设计:架构解析与实践

引言:低代码平台的事件驱动范式在现代软件开发中,事件驱动架构(EDA)已成为构建灵活、松耦合系统的核心范式。OneCode低代码平台通过创新性的注解驱动设计,将事件驱动理念深度融入平台架构,实现了业务...

国内大厂AI插件评测:根据UI图生成Vue前端代码

在IDEA中安装大厂的AI插件,打开ruoyi增强项目:yudao-ui-admin-vue31.CodeBuddy插件登录腾讯的CodeBuddy后,大模型选择deepseek-v3,输入提示语:...

AI+低代码技术揭秘(二):核心架构

本文档介绍了为VTJ低代码平台提供支持的基本架构组件,包括Engine编排层、Provider服务系统、数据模型和代码生成管道。有关UI组件库和widget系统的信息,请参阅UI...

GitDiagram用AI把代码库变成可视化架构图

这是一个名为gitdiagram的开源工具,可将GitHub仓库实时转换为交互式架构图,帮助开发者快速理解代码结构。核心功能一键可视化:替换GitHubURL中的"hub...

30天自制操作系统:第六天:代码架构整理与中断处理

1.拆开bootpack.c文件。根据设计模式将对应的功能封装成独立的文件。2.初始化pic:pic(可编程中断控制器):在设计上,cpu单独只能处理一个中断。而pic是将8个中断信号集合成一个中断...

AI写代码越帮越忙?2025年研究揭露惊人真相

近年来,AI工具如雨后春笋般涌现,许多人开始幻想程序员的未来就是“对着AI说几句话”,就能轻松写出完美的代码。然而,2025年的一项最新研究却颠覆了这一期待,揭示了一个令人意外的结果。研究邀请了16位...

一键理解开源项目:两个自动生成GitHub代码架构图与说明书工具

一、GitDiagram可以一键生成github代码仓库的架构图如果想要可视化github开源项目:https://github.com/luler/reflex_ai_fast,也可以直接把域名替换...

5分钟掌握 c# 网络通讯架构及代码示例

以下是C#网络通讯架构的核心要点及代码示例,按协议类型分类整理:一、TCP协议(可靠连接)1.同步通信//服务器端usingSystem.Net.Sockets;usingTcpListene...

从复杂到优雅:用建造者和责任链重塑代码架构

引用设计模式是软件开发中的重要工具,它为解决常见问题提供了标准化的解决方案,提高了代码的可维护性和可扩展性,提升了开发效率,促进了团队协作,提高了软件质量,并帮助开发者更好地适应需求变化。通过学习和应...

低代码开发当道,我还需要学习LangChain这些框架吗?| IT杂谈

专注LLM深度应用,关注我不迷路前两天有位兄弟问了个问题:当然我很能理解这位朋友的担忧:期望效率最大化,时间用在刀刃上,“不要重新发明轮子”嘛。铺天盖地的AI信息轰炸与概念炒作,很容易让人浮躁与迷茫。...

框架设计并不是简单粗暴地写代码,而是要先弄清逻辑

3.框架设计3.框架设计本节我们要开发一个UI框架,底层以白鹭引擎为例。框架设计的第一步并不是直接撸代码,而是先想清楚设计思想,抽象。一个一个的UI窗口是独立的吗?不是的,...

大佬用 Avalonia 框架开发的 C# 代码 IDE

AvalonStudioAvalonStudio是一个开源的跨平台的开发编辑器(IDE),AvalonStudio的目标是成为一个功能齐全,并且可以让开发者快速使用的IDE,提高开发的生产力。A...

轻量级框架Lagent 仅需20行代码即可构建自己的智能代理

站长之家(ChinaZ.com)8月30日消息:Lagent是一个专注于基于LLM模型的代理开发的轻量级框架。它的设计旨在简化和提高这种模型下代理的开发效率。LLM模型是一种强大的工具,可以...

取消回复欢迎 发表评论: