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

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

ccwgpt 2025-04-05 17:17 5 浏览 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中间件,只有在提供正确的用户名和密码时,才能访问该接口。在这里,我们使用了一个固定的用户名和密码。

相关推荐

详解DNFSB2毒王的各种改动以及大概的加点框架

首先附上改动部分,然后逐项分析第一个,毒攻掌握技能意思是力量智力差距超过15%的话差距会被强行缩小到15%,差距不到15%则无效。举例:2000力量,1650智力,2000*0.85=1700,则智力...

通篇干货!纵观 PolarDB-X 并行计算框架

作者:玄弟七锋PolarDB-X面向HTAP的混合执行器一文详细说明了PolarDB-X执行器设计的初衷,其初衷一直是致力于为PolarDB-X注入并行计算的能力,兼顾TP和AP场景,逐渐...

字节新推理模型逆袭DeepSeek,200B参数战胜671B,豆包史诗级加强

梦晨发自凹非寺量子位|公众号QbitAI字节最新深度思考模型,在数学、代码等多项推理任务中超过DeepSeek-R1了?而且参数规模更小。同样是MoE架构,字节新模型Seed-Thinkin...

阿里智能化研发起飞!RTP-LLM 实现 Cursor AI 1000 token/s 推理技术揭秘

作者|赵骁勇阿里巴巴智能引擎事业部审校|刘侃,KittyRTP-LLM是阿里巴巴大模型预测团队开发的高性能LLM推理加速引擎。它在阿里巴巴集团内广泛应用,支撑着淘宝、天猫、高德、饿...

多功能高校校园小程序/校园生活娱乐社交管理小程序/校园系统源码

校园系统通常是为学校、学生和教职工提供便捷的数字化管理工具。综合性社交大学校园小程序源码:同城校园小程序-大学校园圈子创业分享,校园趣事,同校跑腿交友综合性论坛。小程序系统基于TP6+Uni-app...

婚恋交友系统nuiAPP前端解决上传视频模糊的问题

婚恋交友系统-打造您的专属婚恋交友平台系统基于TP6+Uni-app框架开发;客户移动端采用uni-app开发,管理后台TH6开发支持微信公众号端、微信小程序端、H5端、PC端多端账号同步,可快速打包...

已节省数百万GPU小时!字节再砍MoE训练成本,核心代码全开源

COMET团队投稿量子位|公众号QbitAI字节对MoE模型训练成本再砍一刀,成本可节省40%!刚刚,豆包大模型团队在GitHub上开源了叫做COMET的MoE优化技术。COMET已应用于字节...

通用电气完成XA102发动机详细设计审查 将为第六代战斗机提供动力

2025年2月19日,美国通用电气航空航天公司(隶属于通用电气公司)宣布,已经完成了“下一代自适应推进系统”(NGAP)计划下提供的XA102自适应变循环发动机的详细设计审查阶段。XA102是通用电气...

tpxm-19双相钢材质(双相钢f60材质)

TPXM-19双相钢是一种特殊的钢材,其独特的化学成分、机械性能以及广泛的应用场景使其在各行业中占有独特的地位。以下是对TPXM-19双相钢的详细介绍。**化学成分**TPXM-19双相钢的主要化学成...

thinkphp6里怎么给layui数据表格输送数据接口

layui官网已经下架了,但是产品还是可以使用。今天一个朋友问我怎么给layui数据表格发送数据接口,当然他是学前端的,后端不怎么懂,自学了tp框架问我怎么调用。其实官方文档上就有相应的数据格式,js...

完美可用的全媒体广告精准营销服务平台PHP源码

今天测试了一套php开发的企业网站展示平台,还是非常不错的,下面来给大家说一下这套系统。1、系统架构这是一套基于ThinkPHP框架开发的HTML5响应式全媒体广告精准营销服务平台PHP源码。现在基于...

一对一源码开发,九大方面完善基础架构

以往的直播大多数都是一对多进行直播社交,弊端在于不能满足到每个用户的需求,会降低软件的体验感。伴随着用户需求量的增加,一对一直播源码开始出现。一个完整的一对一直播流程即主播发起直播→观看进入房间观看→...

Int J Biol Macromol .|交联酶聚集体在分级共价有机骨架上的固定化:用于卤代醇不对称合成的高稳定酶纳米反应器

大家好,今天推送的文章发表在InternationalJournalofBiologicalMacromolecules上的“Immobilizationofcross-linkeden...

【推荐】一款开源免费的 ChatGPT 聊天管理系统,支持PC、H5等多端

如果您对源码&技术感兴趣,请点赞+收藏+转发+关注,大家的支持是我分享最大的动力!!!项目介绍GPTCMS是一款开源且免费(基于GPL-3.0协议开源)的ChatGPT聊天管理系统,它基于先进的GPT...

高性能计算(HPC)分布式训练:训练框架、混合精度、计算图优化

在深度学习模型愈发庞大的今天,分布式训练、高效计算和资源优化已成为AI开发者的必修课。本文将从数据并行vs模型并行、主流训练框架(如PyTorchDDP、DeepSpeed)、混合精度训练(...

取消回复欢迎 发表评论: