go web: 3 中间件和路由

最新推荐文章于 2025-05-22 16:41:40 发布

原创最新推荐文章于 2025-05-22 16:41:40 发布 · 2.4k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#中间件 #web #go

go_web 专栏收录该内容

6 篇文章

订阅专栏

本文探讨了Go Web开发中中间件的概念，解释了如何利用Go的鸭子类型特性来创建自定义handler并实现中间件。同时，文章介绍了通过hack方法控制handler调用来轻松实现路由。推荐使用第三方包`gorilla/mux`来简化路由规则的匹配。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

中间件

中间件的本质来看，就是在执行handler的前（后）先执行一个自定义的handler而已。那问题变成，go web中，每个handler是怎么执行的。

答案在mux := http.NewServeMux()中，稍微阅读下源码，我们就能得出，mux对象中有个ServeHTTP(w, r)方法。这就秘密所在。

根据go的鸭子类型特性，我们完全可以实现一个结构，然后让它拥有ServeHTTP(w, r)方法。把这个结构替换掉http.Server对象中的Handler，就能自定义hander的执行。既然都能控制handler运行了，中间件什么的还不是小case。

然而还能更简单，可爱的go语言还在http包中，提供了一个http.HandlerFunc(ourFunc)方法，它能把签名为func(w http.ResponseWriter, r *http.Request)的函数转化为一个handler，没错，就是上面mux相同的类型。
要实现自己的mux，可以只是一个简单的函数：

// myHost 做中间件
func myHost(handler http.Handler) http.Handler {
    ourFunc := func(w http.ResponseWriter, r *http.Request) {
        //记录时间
        start := time.Now()
        handler.ServeHTTP(w, r)
        logger.Infoln(
            fmt.Sprintf("%s %s %s", r.Method, r.URL, time.Now().Sub(start)))

    }
    return http.HandlerFunc(ourFunc)
}

这个“记录时间”的操作，不就是最简单的中间件吗？
main.go中的调用则改成：

mux := http.NewServeMux()
mh := myHost(mux)
server :=
    http.Server{
        Addr:         fmt.Sprintf(":%d", options.GetInt("port")),
        Handler:      mh,
        ReadTimeout:  3 * time.Second,
        WriteTimeout: 5 * time.Second,
    }
// 开始添加路由
mux.HandleFunc("/hi", test.SayHello)
server.ListenAndServe()

路由

既然我们能通过hack ServeHTTP来控制handler的调用，那实现路由还不是顺水推舟。
在server.go中看看mux.ServeHTTP和mux.handler这两个函数源码，这个简单而蛋疼的默认路由就跃然纸上。
对于路由，我们没必要自己写ServeHttp和match规则，因为太麻烦了。
所以，我们google下 httprouter这个包。轮子都造好了。