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