go web: 3 中间件和路由 ## 中间件 中间件的本质来看，就是在执行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`这个包。轮子都造好了。 来自 大脸猫 写于 2017-08-31 22:18 -- 更新于2020-10-19 13:06 -- 0 条评论