Golang 函数链中如何使用中间件？- 技术经验 -卓越飞翔博客

可以在 go 函数链中使用中间件机制，通过实现 http.handler 接口，在函数执行前后添加代码。中间件包提供了创建和应用中间件的功能，可通过 iris.useglobal 全局应用或 iris.use 应用于特定路由。一个实战示例是通过中间件记录请求处理时间。

Golang 函数链中如何使用中间件？

Go 函数链中的中间件

中间件是一种在函数执行前后执行代码的机制，可用于向函数链添加附加功能。在 Go 中，可以使用专门的中间件包轻松实现此目的。

安装中间件包

立即学习“go语言免费学习笔记（深入）”；

go get github.com/kataras/iris/v12/middleware/recover

创建中间件

func MyMiddleware(next http.Handler) http.Handler {
  return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    // Middleware logic before the next handler
    next.ServeHTTP(w, r)
    // Middleware logic after the next handler
  })
}

应用中间件

中间件可通过 iris.UseGlobal 函数全局应用到所有路由，或通过 iris.Use 函数应用于单个路由。

// 全局应用中间件
iris.UseGlobal(recover.New())

// 应用于单个路由
iris.Use(MyMiddleware)

实战案例

以下是一个使用中间件记录请求处理时间的示例：

func TimeMiddleware(next http.Handler) http.Handler {
  return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    start := time.Now()

    next.ServeHTTP(w, r)

    elapsed := time.Since(start)
    log.Printf("Request %s processed in %s", r.URL.Path, elapsed)
  })
}

func main() {
  iris.Use(TimeMiddleware)
  // ...
}

相关推荐