go 框架的中间件允许开发人员扩展 Web 应用程序的功能。 最新趋势包括:路由中间件:在处理函数之前拦截请求并执行特定操作。错误处理中间件:捕获错误并提供自定义响应。日志记录中间件:记录有关请求和响应的信息。
Go 框架中间件的最新趋势
在 Go 中,中间件是一个用于扩展 Web 应用程序功能的软件层。中间件可用于执行各种任务,例如身份验证、日志记录和错误处理。
1. 路由中间件
立即学习“go语言免费学习笔记(深入)”;
路由中间件允许开发人员拦截传入的请求,执行特定操作,然后再将其传递到处理函数。这对于需要在每个请求上执行常见操作(例如身份验证或日志记录)的情况非常有用。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
type AuthMiddleware struct {
authClient *AuthClient
}
func (m *AuthMiddleware) WrapHandler(next HTTP.Handler) http.Handler {
return http.HandlerFunc( func (w http.ResponseWriter, r *http.Request) {
ctx := r.Context()
if _, err := m.authClient.Authorize(ctx, r); err != nil {
http. Error (w, "unauthorized" , http.StatusUnauthorized)
return
}
next.ServeHTTP(w, r)
})
}
|
2. 错误处理中间件
错误处理中间件允许开发人员捕获应用程序中发生的错误,并提供自定义响应。这对于防止错误泄露给最终用户非常有用。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
type ErrORMiddleware struct {
logger *zap.Logger
}
func (m *ErrorMiddleware) WrapHandler(next http.Handler) http.Handler {
return http.HandlerFunc( func (w http.ResponseWriter, r *http.Request) {
defer func () {
if err := recover (); err != nil {
m.logger. Error ( "error occurred" , zap. Error (err))
http. Error (w, "internal server error" , http.StatusInternalServerError)
}
}()
next.ServeHTTP(w, r)
})
}
|
3. 日志记录中间件
日志记录中间件允许开发人员记录有关应用程序请求和响应的信息。这对于调试问题和分析应用程序行为非常有用。
1
2
3
4
5
6
7
8
9
10
11
12
|
type LogMiddleware struct {
logger *zap.Logger
}
func (m *LogMiddleware) WrapHandler(next http.Handler) http.Handler {
return http.HandlerFunc( func (w http.ResponseWriter, r *http.Request) {
ctx := r.Context()
m.logger.Info( "request received" , zap. String ( "method" , r.Method), zap. String ( "path" , r.URL.Path))
next.ServeHTTP(w, r)
m.logger.Info( "request completed" , zap. String ( "method" , r.Method), zap. String ( "path" , r.URL.Path), zap. Int ( "status" , w.StatusCode))
})
}
|
实战案例
考虑一个需要用户进行身份验证才能访问资源的 Web 应用程序。我们可以使用路由中间件来拦截传入的请求并执行身份验证检查:
1
2
3
4
5
6
7
8
9
10
11
12
13
|
func main() {
authClient := &AuthClient{}
authMiddleware := &AuthMiddleware{authClient: authClient}
mux := http.NewServeMux()
mux.Handle( "/protected" , authMiddleware.WrapHandler(http.HandlerFunc(protectedHandler)))
http.ListeNANDServe( ":8080" , mux)
}
func protectedHandler(w http.ResponseWriter, r *http.Request) {
fmt.Fprintf(w, "Protected content" )
}
|
使用此中间件,只有经过身份验证的用户才能访问 /protected 路由,否则将返回 401 Unauthorized 错误。