go 框架中间件是一种可插拔机制,允许开发者自定义请求处理逻辑。通过源码分析,了解中间件的工作原理:net/http 包: 定义 http 请求处理器和中间件函数接口。net/http/httputil 包: 提供了 middlewarefunc 类型,用于为处理器添加中间件。gin 框架: 具有类似于 net/http 包的中间件用法。实战案例验证中间件:验证中间件: 验证请求中的身份验证令牌。计时中间件: 测量请求执行时间。
Go 框架中间件的源码分析
简介
在 Go 框架中,中间件是一种强大且灵活的机制,允许开发者在请求处理管道中插入自定义逻辑。本文将深入分析 Go 框架中间件的源码,了解其内部工作原理。
源码分析
- net/http 包:http.HandlerFunc 类型是 HTTP 请求处理器的基本接口,它接收一个 http.ResponseWriter 和一个 *http.Request 并返回一个 error。
- net/http/httputil 包:MiddlewareFunc 类型是一个函数,它接受一个 Handler 并返回一个新的 Handler。
- github.com/gin-gonic/gin:Gin 是一款流行的 Go 框架,其 HandlerFunc 用法与 http.HandlerFunc 非常相似。
实战案例
验证中间件
创建一个中间件来验证 HTTP 请求中的身份验证令牌:
立即学习“go语言免费学习笔记(深入)”;
import (
"net/http"
"strings"
)
func AuthMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
token := r.Header.Get("Authorization")
if token == "" || !strings.HasPrefix(token, "Bearer ") {
http.Error(w, "Unauthorized", http.StatusUnauthorized)
return
}
// 验证令牌...
// 调用下个处理器
next.ServeHTTP(w, r)
})
}
计时中间件
创建一个中间件来测量 HTTP 请求的执行时间:
import (
"net/http"
"time"
)
func TimerMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
start := time.Now()
next.ServeHTTP(w, r)
end := time.Now()
// 记录持续时间...
})
}
使用 Gin 框架的中间件
Gin 的中间件用法也很简单:
import (
"github.com/gin-gonic/gin"
)
func main() {
router := gin.Default()
// 注册中间件
router.Use(AuthMiddleware)
router.Use(TimerMiddleware)
// 添加路由处理器
router.GET("/", func(c *gin.Context) { /* ... */ })
// 启动服务器
router.Run()
}