golang 框架_HTTP中间件机制实现与原理 - 从零开始写GO-API框架 [重复造轮子 - Golang]...

1、重置HTTP请求路由
2、统一安全限制、信息上报
3、Header操作、http请求认证
4、屏蔽爬虫
5、提供调试信息
6、请求日志记录
 
还有很多，可以自行发掘下

func middlewareLoger() koala.HandlerFunc {
    return func(ctx *koala.Context) {
 
        fmt.Printf("%+v", ctx.Req)
        fmt.Println("middloger 中间件")
        // fmt.Printf("%+V 进入到中间件了n", ctx)
    }
}
 
func main() {
    app := koala.New()
    app.Use(middlewareLoger())
    app.Add("GET", "/profile/xiaoliang", func(ctx *koala.Context) {
        ctx.Text("profile.xiaoliang")
    })
}

[KOALA-DEBUG][启动中]
Add GET /profile/xiaoliang 
Add GET /profile/xiaoliang1 
Add GET /member/:id 
[KOALA-DEBUG]监听端口[:8080]
[KOALA-DEBUG][服务启动成功]
 
此处是中间件输出信息
&{Method:GET URL:/profile/xiaoliang1 Proto:HTTP/1.1 ...... ctx:0xc00005c4c0}middlewareLoger 中间件

type Middleware interface {}
 
// 定义 koala引擎结构
type Koala struct {
    ...代码折叠
 
    // 中间件
    middleware []HandlerFunc
 
}
 
// 注册中间件
func(k *Koala) Use(m ...Middleware) {
    for i := range m {
        if m[i] != nil {
            // 注册中间件放入切片数组
            k.middleware = append(k.middleware, warpMiddleware(m[i]))
        }
    }
}

func warpMiddleware(m Middleware) HandlerFunc {
    // 断言当前函数类型
    switch m:= m.(type) {
    case HandlerFunc:
        return m
    case func (*Context):
        return m
    default:
        fmt.Printf("%+V", m)
        panic("没找到相关中间件")
    }
}

// 实现net/http 需要的servehttp服务
func (k *Koala) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
 
    ...代码折叠
 
    // 执行相关操作
    ctx.Next()
 
    ...代码折叠
}

// 首先处理中间件，然后处理路由句柄
func (c *Context) Next() {
    // http中间件处理
    c.middleware()
 
    // 路由映射处理
    c.koala.router().HandlerRouter(c)
}
 
 
// 执行HTTP中间件
func (c *Context) middleware() {
    for m := range c.koala.middleware{
        // c http上下文传递给注册的中间件函数,这时候已经拿到了用户侧web请求数据，可以进行响应的逻辑操作
        c.koala.middleware[m](c)
    }
}