如何构建一个高效的中间件

组成部分

中间件通常由两部分组成：

• 第一部分仅在中间件初始化时执行一次。在这里，你可以设置全局对象、配置逻辑等——所有只需在应用生命周期中发生一次的事情。

• 第二部分在每个请求时执行。例如，在数据库中间件中，你会将全局数据库对象注入到请求上下文中。一旦注入，其它中间件和处理函数都可以获取并使用它。

func funcName(params string) gin.HandlerFunc {
  // <---
  // 第一步
  // --->
  // 初始化示例：校验输入参数
  if err := check(params); err != nil {
      panic(err)
  }

  return func(c *gin.Context) {
    // <---
    // 第二步
    // --->
    // 每次请求执行：注入到上下文
    c.Set("TestVar", params)
    c.Next()
  }
}

执行过程

来看下下面的代码示例：

func main() {
  router := gin.Default()

  router.Use(globalMiddleware())

  router.GET("/rest/n/api/*some", mid1(), mid2(), handler)

  router.Run()
}

func globalMiddleware() gin.HandlerFunc {
  fmt.Println("globalMiddleware...1")

  return func(c *gin.Context) {
    fmt.Println("globalMiddleware...2")
    c.Next()
    fmt.Println("globalMiddleware...3")
  }
}

func handler(c *gin.Context) {
  fmt.Println("exec handler.")
}

func mid1() gin.HandlerFunc {
  fmt.Println("mid1...1")

  return func(c *gin.Context) {

    fmt.Println("mid1...2")
    c.Next()
    fmt.Println("mid1...3")
  }
}

func mid2() gin.HandlerFunc {
  fmt.Println("mid2...1")

  return func(c *gin.Context) {
    fmt.Println("mid2...2")
    c.Next()
    fmt.Println("mid2...3")
  }
}

根据上述组成部分，当运行 Gin 进程时，每个中间件的第一步先执行并输出如下信息：

globalMiddleware...1
mid1...1
mid2...1

初始化顺序如下：

globalMiddleware...1
    |
    v
mid1...1
    |
    v
mid2...1

当你发起请求（如 curl -v localhost:8080/rest/n/api/some），每个中间件的第二步按顺序执行，输出如下内容：

globalMiddleware...2
mid1...2
mid2...2
exec handler.
mid2...3
mid1...3
globalMiddleware...3

换句话说，执行顺序如下：

globalMiddleware...2
    |
    v
mid1...2
    |
    v
mid2...2
    |
    v
exec handler.
    |
    v
mid2...3
    |
    v
mid1...3
    |
    v
globalMiddleware...3