[每周一更]-(第83期)：Go新项目-Gin中间件的使用和案例（10）-编程知识

在这里插入图片描述

在 Gin 中，中间件是一种用于处理 HTTP 请求和响应的功能强大的机制。中间件是一段位于请求处理链和最终处理器之间的代码，
它可以截获请求、执行预处理操作，修改请求或响应，然后将控制权传递给下一个中间件或最终的请求处理器。

中间价在业务使用中，方便注入一些业务无关的逻辑，并且方便移植到特质的接口中；

下文会聊下：日志、ip限制、API请求速率控制、跨域请求；

以下是 Gin 中间件的概念和使用方式：

概念

执行顺序： Gin 中的中间件按照它们添加的顺序执行。先添加的中间件先执行。

中断处理链：中间件可以选择中断处理链，阻止后续的中间件和处理器执行。

修改请求和响应：中间件可以修改请求的内容，也可以修改响应的内容。这使得中间件非常灵活，可以用于日志记录、认证、跨域处理等多种用途。

1、日志

使用方式

引入 Gin 包：

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

定义中间件函数：

中间件函数是一个接受 *gin.Context 参数的函数。例如，一个简单的日志记录中间件：

func LoggerMiddleware(c *gin.Context) {// 执行处理之前的逻辑fmt.Println("Start Request")// 继续处理链c.Next()// 处理之后的逻辑fmt.Println("End Request")
}

使用中间件：

在路由组或路由上使用中间件。以下是一个简单的使用示例：

func main() {r := gin.Default()// 使用自定义中间件r.Use(LoggerMiddleware)// 路由定义r.GET("/ping", func(c *gin.Context) {c.String(200, "pong")})// 启动服务r.Run(":8080")
}

在上面的例子中，LoggerMiddleware 将在每个请求前后打印日志。

自定义中间件

你可以通过编写函数来创建自定义中间件。以下是一个例子，演示了一个处理请求头的中间件：

func CustomMiddleware() gin.HandlerFunc {return func(c *gin.Context) {// 在请求处理之前执行的逻辑c.Header("Custom-Header", "Custom Value")// 继续处理链c.Next()// 在请求处理之后执行的逻辑}
}

然后，你可以像下面这样使用自定义中间件：

goCopy code
func main() {r := gin.Default()// 使用自定义中间件r.Use(CustomMiddleware())// 路由定义r.GET("/ping", func(c *gin.Context) {c.String(200, "pong")})// 启动服务r.Run(":8080")
}

这个自定义中间件简单地在每个请求中添加了一个自定义的响应头。

注：某些单独接口使用中间件(一般都是全局设置)

r.POST("demo-middleware",IP,Rate,demo.Action)

2、校验IP

调用开源库：GeoLite2-Country/GeoLite2-Country.mmdb，进行比对判断是否为国内IP

package middlewareimport ("net""net/http""github.com/gin-gonic/gin""github.com/oschwald/geoip2-golang"
)// IP 校验ip，非国内ip禁止访问。采用geoip2-lite数据库，免费版本，不保证准确率
func IP() gin.HandlerFunc {return func(c *gin.Context) {db, err := geoip2.Open(config.Env.GetString("app.mmdb_path"))if err != nil {panic(err)}defer db.Close()ip := net.ParseIP(c.GetHeader("X-Real-IP"))ipInfo, err := db.Country(ip)if err == nil {if ipInfo.Country.IsoCode != "CN" && ipInfo.Country.IsoCode != "" {response.CustomErrorJson(c, http.StatusBadRequest, response.NOT_PERMISSION,"禁止非中国用户访问", map[string]any{"errors": "禁止非中国用户访问"})return}}}// 处理请求c.Next() //  处理请求
}

3、API接口速度控制

接口请求速度控制，防止恶意爬虫或者恶意请求数据，导致服务资源紧张；

package middlewareimport ("net/http""github.com/gin-gonic/gin""golang.org/x/time/rate"
)var limiter *rate.Limiterfunc init() {// 令牌桶大小为60，每秒产生1个token，即每分钟最多处理60个请求limiter = rate.NewLimiter(1, 60)
}func Rate() gin.HandlerFunc {return func(ctx *gin.Context) {if !limiter.Allow() {response.CustomErrorJson(ctx, http.StatusBadRequest, response.RATE_LIMIT,"请求过于频繁，请稍后重试", map[string]any{"errors": "请求过于频繁，请稍后重试"})return}// 处理请求c.Next() //  处理请求}
}

4、跨域请求

package middlewareimport ("net/http""github.com/gin-gonic/gin"
)func Cors() gin.HandlerFunc {return func(c *gin.Context) {method := c.Request.Method               // 请求方法origin := c.Request.Header.Get("Origin") // 请求头部var headerKeys []string                  // 声明请求头keysfor k, _ := range c.Request.Header {headerKeys = append(headerKeys, k)}headerStr := strings.Join(headerKeys, ", ")if headerStr != "" {headerStr = fmt.Sprintf("access-control-allow-origin, access-control-allow-headers, %s", headerStr)} else {headerStr = "access-control-allow-origin, access-control-allow-headers"}if origin != "" {c.Writer.Header().Set("Access-Control-Allow-Origin", "*")c.Header("Access-Control-Allow-Origin", "*")                                       // 这是允许访问所有域c.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE,UPDATE") // 服务器支持的所有跨域请求的方法,为了避免浏览次请求的多次'预检'请求//  header的类型c.Header("Access-Control-Allow-Headers", "Authorization, Content-Length, X-CSRF-Token,"+" Token,session,X_Requested_With,Accept, Origin, Host, Connection, Accept-Encoding, "+"Accept-Language,DNT, X-CustomHeader, Keep-Alive, User-Agent, X-Requested-With, If-Modified-Since, "+"Cache-Control, Content-Type, Pragma")// 允许跨域设置         可以返回其他子段c.Header("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, "+"Access-Control-Allow-Headers,Cache-Control,Content-Language,Content-Type,Expires,Last-Modified,Pragma,FooBar") // 跨域关键设置 让浏览器可以解析c.Header("Access-Control-Max-Age", "172800")          // 缓存请求信息 单位为秒c.Header("Access-Control-Allow-Credentials", "false") //  跨域请求是否需要带cookie信息 默认设置为truec.Set("content-type", "application/json")             // 设置返回格式是json}// 放行所有OPTIONS方法if method == "OPTIONS" {c.JSON(http.StatusOK, "Options Request!")}// 处理请求c.Next() //  处理请求}
}