Golang中的路由使用詳解
之前有篇文章比較淺顯的分析了一下golang的服務(wù)器如何實(shí)現(xiàn),還有Handler, DefaultServeMux,HandlerFunc
的用處。
我們現(xiàn)在已經(jīng)明白了DefaultServeMux
就是存放pattern
和handler
的地方,我們稱其為路由,那么我們可能會想,既然golang能夠?qū)崿F(xiàn)這個路由,我們能否也模仿一個呢?
首先我們需要一個能夠保存客戶端的請求的一個容器(路由)。
創(chuàng)建路由結(jié)構(gòu)體
type CopyRouter struct { router map[string]map[string]http.HandlerFunc }
在這里我們創(chuàng)建了一個像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所對應(yīng)的handler保存起來。
實(shí)現(xiàn)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) } }
在這里為什么要實(shí)現(xiàn)這個Handler接口,因?yàn)槲覀儼l(fā)現(xiàn)在ListenAndServe方法中,最后會調(diào)用h.ServeHTTP(w, r),
那么我們就只需要讓我們定義的路由實(shí)現(xiàn)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
很像,當(dāng)然我們這個路由是一個低配版的,還有很多細(xì)節(jié)沒有處理。
現(xiàn)在再看看,我們的main函數(shù)里面的代碼不是很美觀,每一次都要寫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) } ...
然后再修改一下調(diào)用方式。
copyRouter.GET("/sayHi",sayHi)
現(xiàn)在看起來是不是就美觀很多了?是的,很多web框架也是這樣,為什么用起來就感覺很流暢,因?yàn)檫@些大神們就是站在我們開發(fā)者的角度來考慮問題,提供了很方便的一些用法,封裝的很完善。
再考慮一下,我們這個自定義的路由還能做些什么,如果我們要記錄每一次的訪問請求,該如何處理呢?也很簡單,我們只需要將邏輯寫在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) } }
這里我們又加入了一個記錄請求時間的功能,所以在這個自定義的路由里面還可以做更多的事情。
還有一點(diǎn),就是我們在定義這個路由結(jié)構(gòu)體的時候,能否將這個類型修改為Handler呢?也就是將這個類型map[string]map[string]http.HandlerFunc
修改為map[string]map[string]http.Handler
,是可以的,但是我們在調(diào)用的時候就需要在main方法里面做一下修改。
copyRouter.GET("/sayHi",HandlerFunc(sayHi))
在這里做一個強(qiáng)制轉(zhuǎn)換即可,但是這樣也不是很美觀。
看到這里,我們應(yīng)該對一個源碼中的類型重點(diǎn)關(guān)注一下,那就是HandlerFunc。
type HandlerFunc func(ResponseWriter, *Request) func (f HandlerFunc) ServeHTTP(w ResponseWriter, r *Request) { f(w, r) }
這里HandlerFunc起到了一個適配器的作用,這是一個非常巧妙的設(shè)計(jì),不得不說golang在接口這方面確實(shí)設(shè)計(jì)的很精妙。
以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
go語言beego框架jwt身份認(rèn)證實(shí)現(xiàn)示例
這篇文章主要為大家介紹了go語言beego框架jwt身份認(rèn)證實(shí)現(xiàn)示例,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步早日升職加薪2022-04-04Go語言leetcode題解953驗(yàn)證外星語詞典示例詳解
這篇文章主要為大家介紹了Go語言leetcode題解953驗(yàn)證外星語詞典示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2022-12-12Go環(huán)境變量配置,及GOROOT、GOPATH的區(qū)別小結(jié)
本文主要介紹了Go環(huán)境變量配置,及GOROOT、GOPATH的區(qū)別小結(jié),文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2024-09-09Go語言使用buffer讀取文件的實(shí)現(xiàn)示例
本文主要介紹了Go語言使用buffer讀取文件的實(shí)現(xiàn)示例,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-04-04Go語言服務(wù)器開發(fā)實(shí)現(xiàn)最簡單HTTP的GET與POST接口
這篇文章主要介紹了Go語言服務(wù)器開發(fā)實(shí)現(xiàn)最簡單HTTP的GET與POST接口,實(shí)例分析了Go語言http包的使用技巧,需要的朋友可以參考下2015-02-02Go語言字典(map)用法實(shí)例分析【創(chuàng)建,填充,遍歷,查找,修改,刪除】
這篇文章主要介紹了Go語言字典(map)用法,結(jié)合實(shí)例形式較為詳細(xì)的分析了Go語言字典的創(chuàng)建、填充、遍歷、查找、修改、刪除等操作相關(guān)實(shí)現(xiàn)技巧,需要的朋友可以參考下2017-02-02Golang實(shí)現(xiàn)驗(yàn)證一個字符串是否為URL
在實(shí)際開發(fā)過程中,有時候會遇到?URL?的校驗(yàn)問題,Go?語言中有哪些方法去驗(yàn)證一個字符串是否滿足?URL?格式呢?本文就來和大家詳細(xì)講講2023-04-04