怎么使用 go 框架中的中间件完成代码复用？建立一个函数接受 http.handler 并返回新的 http.handler。将中间件注册到路由器或 http.handler 中。举例子当需要身份认证的 api 控制板中应用中间件解决身份认证。
怎么使用 Go 框架中的中间件完成代码复用
介绍
在 Go 框架中，中间件是一个功能强大工具，它允许我们在对待 HTTP 要求或回应之前或以后实行特殊操作。这使得代码复用变得更加容易，能够减少重复代码。
建立中间件
要建立中间件，我们应该界定一个函数，该函数接受一个 http.Handler 并返回一个新的 http.Handler。下面是一个实例：
funcMyMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(whttp.ResponseWriter,rhttp.Request){
//自定中间件逻辑
next.ServeHTTP(w, r)
})
}
注册中间件
为了在大家的 Go 框架中应用中间件，我们需要将其注册到路由器或 http.Handler 中。我们可以使用 Use() 方式来达到：
router:=http.NewServeMux()
router.Use(MyMiddleware)
实战案例
假定我们有一个必须对要求开展身份认证的 API 控制板。大家可以创建一个中间件去处理身份认证，再将它用于所有需要身份认证的控制板：
//身份认证中间件
funcAuthMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(whttp.ResponseWriter,rhttp.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 控制器的研发流程。
之上就是如何通过golang框架中的中间件完成代码复用？的详细内容，大量请关注其他类似文章！