本文主要針對Golang的內(nèi)置庫 net/http 做了簡單的擴(kuò)展，通過添加中間件的形式實現(xiàn)了管道(Pipeline)模式，這樣的好處是各模塊之間是低耦合的，符合單一職責(zé)原則，可以很靈活的通過中間件的形式添加一些功能到管道中，一次請求和響應(yīng)在管道中的執(zhí)行過程如下
首先, 我定義了三個測試的中間件 Middleware1,2,3 如下
func Middleware1(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
fmt.Println("M1 in")
next.ServeHTTP(w, r)
fmt.Println("M1 out")
})
}
func Middleware2(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
fmt.Println("M2 in")
next.ServeHTTP(w, r)
fmt.Println("M2 out")
})
}
func Middleware3(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
fmt.Println("M3 in")
next.ServeHTTP(w, r)
fmt.Println("M3 out")
})
}
這里中間件的入?yún)⒑统鰠⒌念愋投际?http.Handler, 然后在 next.ServeHTTP() 的前后分別輸出了 In 和 Out.
接下來，定義一個 Pipeline 的方法，里面使用嵌套的形式, 使用了上面定義的三個測試的中間件.
func Pipeline(next http.Handler) http.Handler {
return Middleware1(Middleware2(Middleware3(next)))
}
然后還需要業(yè)務(wù)代碼，這里我定義了 LoginHandler 和 RegisterHandler 兩個方法
func LoginHandler(w http.ResponseWriter, r *http.Request) {
fmt.Println("Login...")
w.Write([]byte("Login..."))
}
func RegisterHandler(w http.ResponseWriter, r *http.Request) {
fmt.Println("Register...")
w.Write([]byte("Register..."))
}
最后修改程序的 main 函數(shù), 在 Login 接口上使用上面添加過中間件的 Pipeline
func main() {
http.Handle("/Login", Pipeline(http.HandlerFunc(LoginHandler)))
http.Handle("/Register", http.HandlerFunc(RegisterHandler))
http.ListenAndServe(":8080", nil)
}
啟動程序后，訪問 http://localhost:8080/Login, 程序的輸出如下，這和本文最上面的管道的流程圖是一致的，然后訪問 Register 接口, 控制臺沒有輸出信息，當(dāng)然也不會執(zhí)行任何中間件。
現(xiàn)在已經(jīng)實現(xiàn)了中間件的機(jī)制，但是，上面添加中間件是用嵌套的方法，這種方式不能說不太優(yōu)雅，只能說非常的Low，接下來我們需要對管道進(jìn)行優(yōu)化
type Chain struct {
middlewares []func(handler http.Handler) http.Handler
}
func Pipeline(next http.Handler) http.Handler {
//return Middleware1(Middleware2(Middleware3(next)))
return AddMiddlewares(Middleware1,Middleware2,Middleware3).Then(next)
}
func AddMiddlewares(m ...func(handlerFunc http.Handler) http.Handler) Chain {
c := Chain{}
c.middlewares = append(c.middlewares,m...)
return c
}
func (c Chain) Then(next http.Handler) http.Handler {
for i := range c.middlewares {
prev := c.middlewares[len(c.middlewares)-1-i]
next = prev(next)
}
return next
}
首先定義了一個Chain 的struct，用來接收添加到管道中的中間件，在 AddMiddlewares() 函數(shù)中，接收了多個Handle, 然后組裝到 Chain 對象并返回, 接下來調(diào)用 Then() 函數(shù), 把管道中的中間件和業(yè)務(wù)的Handler 關(guān)聯(lián)起來。在中間件的使用方式上， 這兩種方法都是一樣的，只需要調(diào)用 Pipeline() 方法就行了。
本文在go web中簡單的實現(xiàn)了中間件的機(jī)制，這樣帶來的好處也是顯而易見的，當(dāng)然社區(qū)也有一些成熟的 middleware 組件，包括 Gin 一些Web框架中也包含了 middleware 相關(guān)的功能, 希望對您有用.
到此這篇關(guān)于在 Golang 中實現(xiàn)一個簡單的Http中間件的文章就介紹到這了,更多相關(guān)Golang Http中間件內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
標(biāo)簽：宜賓 嘉峪關(guān) 延邊 張掖 武漢 新余 江西 黑龍江
巨人網(wǎng)絡(luò)通訊聲明：本文標(biāo)題《在 Golang 中實現(xiàn)一個簡單的Http中間件過程詳解》，本文關(guān)鍵詞 在,Golang,中,實現(xiàn),一個,簡單,；如發(fā)現(xiàn)本文內(nèi)容存在版權(quán)問題，煩請?zhí)峁┫嚓P(guān)信息告之我們，我們將及時溝通與處理。本站內(nèi)容系統(tǒng)采集于網(wǎng)絡(luò)，涉及言論、版權(quán)與本站無關(guān)。
