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è)計，不得不說golang在接口這方面確實(shí)設(shè)計的很精妙。
以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。
相關(guān)文章
詳解Go語言如何熱重載和優(yōu)雅地關(guān)閉程序
我們有時會因不同的目的去關(guān)閉服務(wù)，一種關(guān)閉服務(wù)是終止操作系統(tǒng)，一種關(guān)閉服務(wù)是用來更新配置，本文就來和大家簡單講講這兩種方法的實(shí)現(xiàn)吧2023-07-07
golang優(yōu)先級隊(duì)列的實(shí)現(xiàn)全過程
優(yōu)先級隊(duì)列是一種特殊隊(duì)列,下面這篇文章主要給大家介紹了關(guān)于golang優(yōu)先級隊(duì)列的實(shí)現(xiàn)全過程,文中通過示例代碼介紹的非常詳細(xì),需要的朋友可以參考下2022-12-12
在Go語言中涉及 I/O 操作的內(nèi)置庫有很多種，比如： io 庫， os 庫， ioutil 庫， bufio 庫， bytes 庫， strings 庫等等。擁有這么多內(nèi)置庫是好事，但是具體到涉及 I/O 的場景我們應(yīng)該選擇哪個庫呢，帶著這個問題一起通過本文學(xué)習(xí)下吧2021-06-06
從golang的定義來看，unsafe 是類型安全的操作。顧名思義，它應(yīng)該非常謹(jǐn)慎地使用; unsafe可能很危險，但也可能非常有用。例如，當(dāng)使用系統(tǒng)調(diào)用和Go結(jié)構(gòu)必須具有與C結(jié)構(gòu)相同的內(nèi)存布局時，您可能別無選擇，只能使用unsafe2022-10-10
Golang實(shí)現(xiàn)的聊天程序服務(wù)端和客戶端代碼分享
這篇文章主要介紹了Golang實(shí)現(xiàn)的聊天程序服務(wù)端和客戶端代碼分享,本文先是講解了實(shí)現(xiàn)邏輯,然后給出了實(shí)現(xiàn)代碼,需要的朋友可以參考下2014-10-10
在Golang（又稱Go語言）中，map是一種非常有用的數(shù)據(jù)結(jié)構(gòu)，所以這篇文章小編就來帶大家一起深入了解一下map的用法，感興趣的小伙伴可以了解一下2023-06-06
Go緩沖channel和非緩沖channel的區(qū)別說明
這篇文章主要介紹了Go緩沖channel和非緩沖channel的區(qū)別說明，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-04-04
Ticker是一個周期觸發(fā)定時的計時器，它會按照一個時間間隔往channel發(fā)送系統(tǒng)當(dāng)前時間，而channel的接收者可以以固定的時間間隔從channel中讀取事件，這篇文章主要介紹了Golang cron 定時器和定時任務(wù),需要的朋友可以參考下2022-09-09
這篇文章主要介紹了GOLANG版的冒泡排序和快速排序分享,需要的朋友可以參考下2015-03-03
Golang WaitGroup實(shí)現(xiàn)原理解析
WaitGroup是Golang并發(fā)的兩種方式之一,一個是Channel,另一個是WaitGroup,下面這篇文章主要給大家介紹了關(guān)于golang基礎(chǔ)之waitgroup用法以及使用要點(diǎn)的相關(guān)資料,需要的朋友可以參考下2023-02-02