之前有篇文章比較淺顯的分析了一下golang的服務(wù)器如何實(shí)現(xiàn)，還有Handler, DefaultServeMux,HandlerFunc的用處。
我們現(xiàn)在已經(jīng)明白了DefaultServeMux就是存放pattern和handler的地方，我們稱其為路由，那么我們可能會(huì)想，既然golang能夠?qū)崿F(xiàn)這個(gè)路由，我們能否也模仿一個(gè)呢？
首先我們需要一個(gè)能夠保存客戶端的請(qǐng)求的一個(gè)容器(路由)。
創(chuàng)建路由結(jié)構(gòu)體
|
1
2
3
|
type CopyRouter struct { router map[string]map[string]http.HandlerFunc} |
在這里我們創(chuàng)建了一個(gè)像DefaultServeMux的路由。
客戶端請(qǐng)求存入路由
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
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將每一個(gè)URL所對(duì)應(yīng)的handler保存起來。
實(shí)現(xiàn)Handler接口
|
1
2
3
4
5
|
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) }} |
在這里為什么要實(shí)現(xiàn)這個(gè)Handler接口，因?yàn)槲覀儼l(fā)現(xiàn)在ListenAndServe方法中，最后會(huì)調(diào)用h.ServeHTTP(w, r)，那么我們就只需要讓我們定義的路由實(shí)現(xiàn)Handler接口就可以了。
獲取一個(gè)路由
|
1
2
3
|
func NewRouter() *CopyRouter { return new(CopyRouter)} |
到這里，我們自己定義的路由就完成了，我們來看看使用方法。
|
1
2
3
4
5
6
7
8
9
|
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))} |
這樣就完成了一個(gè)高仿版的自定義路由，是不是和golang提供給我們的ServeMux很像，當(dāng)然我們這個(gè)路由是一個(gè)低配版的，還有很多細(xì)節(jié)沒有處理。
現(xiàn)在再看看，我們的main函數(shù)里面的代碼不是很美觀，每一次都要寫get或者post方法，那么我們能否提供一個(gè)比較美觀的方式呢？可以，那么我們?cè)俜庋b一下。
|
1
2
3
4
5
6
7
8
9
|
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)}... |
然后再修改一下調(diào)用方式。
|
1
|
copyRouter.GET("/sayHi",sayHi) |
現(xiàn)在看起來是不是就美觀很多了？是的，很多web框架也是這樣，為什么用起來就感覺很流暢，因?yàn)檫@些大神們就是站在我們開發(fā)者的角度來考慮問題，提供了很方便的一些用法，封裝的很完善。
再考慮一下，我們這個(gè)自定義的路由還能做些什么，如果我們要記錄每一次的訪問請(qǐng)求，該如何處理呢？也很簡單，我們只需要將邏輯寫在ServeHTTP方法中就可以了，稍微修改一下我們的代碼。
|
1
2
3
4
5
6
7
8
9
10
|
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(" 請(qǐng)求 [%s] 開始時(shí)間為 : %v\n", r.URL.String(), start) f.ServeHTTP(w, r) log.Printf(" 請(qǐng)求 [%s] 完成時(shí)間為 : %v\n", r.URL.String(), time.Since(start)) }(f) }} |
這里我們又加入了一個(gè)記錄請(qǐng)求時(shí)間的功能，所以在這個(gè)自定義的路由里面還可以做更多的事情。
還有一點(diǎn)，就是我們?cè)诙x這個(gè)路由結(jié)構(gòu)體的時(shí)候，能否將這個(gè)類型修改為Handler呢？也就是將這個(gè)類型map[string]map[string]http.HandlerFunc修改為map[string]map[string]http.Handler，是可以的，但是我們?cè)谡{(diào)用的時(shí)候就需要在main方法里面做一下修改。
|
1
|
copyRouter.GET("/sayHi",HandlerFunc(sayHi)) |
在這里做一個(gè)強(qiáng)制轉(zhuǎn)換即可，但是這樣也不是很美觀。
看到這里，我們應(yīng)該對(duì)一個(gè)源碼中的類型重點(diǎn)關(guān)注一下，那就是HandlerFunc。
|
1
2
3
4
5
|
type HandlerFunc func(ResponseWriter, *Request)func (f HandlerFunc) ServeHTTP(w ResponseWriter, r *Request) { f(w, r)} |
這里HandlerFunc起到了一個(gè)適配器的作用，這是一個(gè)非常巧妙的設(shè)計(jì)，不得不說golang在接口這方面確實(shí)設(shè)計(jì)的很精妙。
以上就是本文的全部內(nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持服務(wù)器之家。
原文鏈接：https://segmentfault.com/a/1190000015324113
