Web 中间件怎么玩？

阿兵云原生

发布于 2023-10-05 08:21:20

2450

发布于 2023-10-05 08:21:20

文章被收录于专栏：golang云原生newgolang云原生new

本次主要是聊聊关于 web 中间件，分为如下四个方面

什么是 web 框架中间件
为什么要使用 web 中间件
如何使用及其原理
哪些场景需要使用中间件

开门见山 web 中间件是啥

Web 框架中的中间件主要指的是在 web 请求到具体路由之前或者之后，会经过一个或者多个组件进行处理一些必要的公共逻辑（业务相关或者与业务无关的），而处理这些事项的部分，就称为 web 中间件

那是否会有这样的疑问？

明明就能直接请求到具体的路由，为什么要在它之前加一个中间件？

这是在增加程序复杂度？有啥事情不能直接在路由中做的吗？

我们可以带着这个问题继续往下看

为什么要使用中间件

一般很多技术或者很多组件大多是因为现有的工具无法满足日益正常的需求而慢慢出现的

例如

在 web 中需要对多个路由或者业务进行解耦，或者需要在多个路由之前或者之后加上一些统一的逻辑，这个时候就需要中间件来进行处理

又例如

我们的 web 服务需要有限流功能

如果我们 web 框架中只有几个路由，那么很简单，可能你会去对每个路由进行限流，那么如果是达到几十上百个路由你还会这样做吗？

如果已经有几十上百个路由了，需要针对所有路由统计一下程序处理时长，那么，这个时候你会去给这些路由一个一个的去复制粘贴代码吗？

正常人自然是不会的，我们会想办法寻求简单高效且保证质量的方式，明明使用一个中间件就能搞定的事情，何必去做无意义的卷王

多多提高效率去做更多有意义的事情不香吗？

中间件如何使用及其原理

此处咱们使用大名鼎鼎的高性能 web 框架 Gin 框架来举例子，使用 Gin 框架

Gin 中的中间件实际上就是一个 RouterGroup 对应的 handers 调用链 ，我们先来看一个例子，自定义两个最简单的中间件，先写一个 main

func main() {
   log.SetFlags(log.Lshortfile)
   r := gin.New()
   r.Use(Demo2())
   r.Use(Demo1())

   r.GET("/test", func(c *gin.Context) {
      log.Println("----inner test----")
      c.JSON(200, gin.H{
         "message": "demo",
      })
   })

   // 监听8080 端口
   r.Run(":8080")
}

Main 函数中，我们可以看到，开启了一个 web 服务，监听的端口是 8080，其中使用 Use 方法关联了 2 个中间件，分别是 Demo2 和 Demo1

Gin 框架中，先新建一个引擎，然后通过 Use 方法来将中间件和路由关联起来，这些中间件会对于每一个请求形成一个调用链

此处的调用链就是通过 Use 方法中使用 append 来进行追加的

// Use adds middleware to the group, see example code in GitHub.
func (group *RouterGroup) Use(middleware ...HandlerFunc) IRoutes {
   group.Handlers = append(group.Handlers, middleware...)
   return group.returnObj()
}

因此，对于我们自定义的中间件，先关联的中间件就先执行，后关联的中间件就后执行，这里我们简单写了两个自定义中间件

中间件，实际上就是去写一个这样的函数

// HandlerFunc defines the handler used by gin middleware as return value.
type HandlerFunc func(*Context)

对于 Demo1 自定义中间件实现是这样的

代码在 c.Next() 前的会在请求具体接口之前进行运行
在代码 c.Next() 后的代码会在执行完具体接口之后执行

可以看到此处我们简单的添加了打印，以及记录接口执行的时间

func Demo1() gin.HandlerFunc {
   return func(c *gin.Context) {
      // 请求接口之前
      a := time.Now()
      log.Println("--before--demo1----")
      c.Next()
      // 请求接口之后
      log.Println("--after--demo1----")
      dur := time.Since(a)
      log.Println("req Time consuming : ",dur)
   }
}

Demo2 也是类似的逻辑，仅仅是添加一些打印