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在接口這方面確實設計的很精妙。