中间件的本质来看,就是在执行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
这个包。轮子都造好了。