如何使用 go 框架中的中间件实现代码复用?创建一个函数接收 http.handler 并返回新的 http.handler。将中间件注册到路由器或 http.handler 中。举例说明在需要身份验证的 api 控制器中使用中间件处理身份验证。
如何使用 Go 框架中的中间件实现代码复用
简介
在 Go 框架中,中间件是一个功能强大的工具,它允许我们在处理 HTTP 请求或响应之前或之后执行特定操作。这使得代码复用变得更加容易,可以减少重复代码。
立即学习“go语言免费学习笔记(深入)”;
创建中间件
要创建中间件,我们需要定义一个函数,该函数接收一个 http.Handler 并返回一个新的 http.Handler。下面是一个示例:
func MyMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
// 自定义中间件逻辑
next.ServeHTTP(w, r)
})
}
注册中间件
为了在我们的 Go 框架中使用中间件,我们需要将其注册到路由器或 http.Handler 中。我们可以使用 Use() 方法来实现:
router := http.NewServeMux()
router.Use(MyMiddleware)
实战案例
假设我们有一个需要对请求进行身份验证的 API 控制器。我们可以创建一个中间件来处理身份验证,然后将它应用于所有需要身份验证的控制器:
// 身份验证中间件
func AuthMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
// 从请求中获取身份验证标头
token := r.Header.Get("Authorization")
// 验证令牌并检查用户是否已登录
user, err := VerifyToken(token)
if err != nil || user == nil {
http.Error(w, "未经授权", http.StatusUnauthorized)
return
}
// 将用户保存在请求上下文中,以便控制器使用
ctx := context.WithValue(r.Context(), "user", user)
// 继续处理请求
next.ServeHTTP(w, r.WithContext(ctx))
})
}
// 需要身份验证的控制器
func MyProtectedController(w http.ResponseWriter, r *http.Request) {
// 从上下文中获取用户
user := r.Context().Value("user").(*User)
// 执行需要身份验证的操作
fmt.Fprintf(w, "欢迎,%s", user.Username)
}
func main() {
// 创建路由器和注册中间件
router := http.NewServeMux()
router.Use(AuthMiddleware)
// 注册需要身份验证的控制器
router.HandleFunc("/api/protected", MyProtectedController)
// 启动服务器
http.ListenAndServe(":8080", router)
}
通过这种方式,我们能够使用中间件实现代码复用,并简化需要授权的 API 控制器的开发过程。