Golang中的路由

以前有篇文章比较浅显的分析了一下golang的服务器如何实现，还有Handler, DefaultServeMux,HandlerFunc的用处。

咱们如今已经明白了DefaultServeMux就是存放pattern和handler的地方，咱们称其为路由，那么咱们可能会想，既然golang可以实现这个路由，咱们可否也模仿一个呢？

首先咱们须要一个可以保存客户端的请求的一个容器(路由)。

建立路由结构体

type CopyRouter struct {
    router map[string]map[string]http.HandlerFunc
}

在这里咱们建立了一个像DefaultServeMux的路由。

客户端请求存入路由

func (c *CopyRouter) HandleFunc(method, pattern string, handle http.HandlerFunc) {
    if method == "" {
        panic("Method can not be null!")
    }

    if pattern == "" {
        panic("Pattern can not be null!")
    }

    if _, ok := c.router[method][pattern]; ok {
        panic("Pattern Exists!")
    }

    if c.router == nil {
        c.router = make(map[string]map[string]http.HandlerFunc)
    }

    if c.router[method] == nil {
        c.router[method] = make(map[string]http.HandlerFunc)
    }
    c.router[method][pattern] = handle
}

这里咱们模仿源码中的ServeMux将每个URL所对应的handler保存起来。

实现Handler接口

func (c *CopyRouter) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    if f, ok := c.router[r.Method][r.URL.String()]; ok {
        f.ServeHTTP(w, r)
    }
}

在这里为何要实现这个Handler接口，由于咱们发如今ListenAndServe方法中，最后会调用h.ServeHTTP(w, r)，那么咱们就只须要让咱们定义的路由实现Handler接口就能够了。

获取一个路由

func NewRouter() *CopyRouter {
    return new(CopyRouter)
}

到这里，咱们本身定义的路由就完成了，咱们来看看使用方法。

func sayHi(w http.ResponseWriter, r *http.Request)  {
    fmt.Fprint(w,"Hi")
}

func main() {
    copyRouter := copyrouter.NewRouter()
    copyRouter.HandleFunc("GET","/sayHi", sayHi)
    log.Fatal(http.ListenAndServe("localhost:8080", copyRouter))
}

这样就完成了一个高仿版的自定义路由，是否是和golang提供给咱们的ServeMux很像，固然咱们这个路由是一个低配版的，还有不少细节没有处理。

如今再看看，咱们的main函数里面的代码不是很美观，每一次都要写get或者post方法，那么咱们可否提供一个比较美观的方式呢？能够，那么咱们再封装一下。

func (c *CopyRouter) GET(pattern string, handler http.HandlerFunc){
    c.HandleFunc("GET", pattern, handler)
}

func (c *CopyRouter) POST(pattern string, handler http.HandlerFunc){
    c.HandleFunc("POST", pattern, handler)
}

...

而后再修改一下调用方式。

copyRouter.GET("/sayHi",sayHi)

如今看起来是否是就美观不少了？是的，不少web框架也是这样，为何用起来就感受很流畅，由于这些大神们就是站在咱们开发者的角度来考虑问题，提供了很方便的一些用法，封装的很完善。

再考虑一下，咱们这个自定义的路由还能作些什么，若是咱们要记录每一次的访问请求，该如何处理呢？也很简单，咱们只须要将逻辑写在ServeHTTP方法中就能够了，稍微修改一下咱们的代码。

func (c *CopyRouter) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    if f, ok := c.router[r.Method][r.URL.String()]; ok {
        func (handler http.Handler){
            start := time.Now()
            log.Printf(" 请求 [%s] 开始时间为 : %v\n", r.URL.String(), start)
            f.ServeHTTP(w, r)
            log.Printf(" 请求 [%s] 完成时间为 : %v\n", r.URL.String(), time.Since(start))
        }(f)
    }
}

这里咱们又加入了一个记录请求时间的功能，因此在这个自定义的路由里面还能够作更多的事情。

还有一点，就是咱们在定义这个路由结构体的时候，可否将这个类型修改成Handler呢？也就是将这个类型map[string]map[string]http.HandlerFunc修改成map[string]map[string]http.Handler，是能够的，可是咱们在调用的时候就须要在main方法里面作一下修改。

copyRouter.GET("/sayHi",HandlerFunc(sayHi))

在这里作一个强制转换便可，可是这样也不是很美观。
看到这里，咱们应该对一个源码中的类型重点关注一下，那就是HandlerFunc。

type HandlerFunc func(ResponseWriter, *Request)

func (f HandlerFunc) ServeHTTP(w ResponseWriter, r *Request) {
    f(w, r)
}

这里HandlerFunc起到了一个适配器的做用，这是一个很是巧妙的设计，不得不说golang在接口这方面确实设计的很精妙。