怎么在Golang中實(shí)現(xiàn)一個(gè)簡(jiǎn)單的Http中間件

本篇內(nèi)容主要講解“怎么在Golang中實(shí)現(xiàn)一個(gè)簡(jiǎn)單的Http中間件”，感興趣的朋友不妨來(lái)看看。本文介紹的方法操作簡(jiǎn)單快捷，實(shí)用性強(qiáng)。下面就讓小編來(lái)帶大家學(xué)習(xí)“怎么在Golang中實(shí)現(xiàn)一個(gè)簡(jiǎn)單的Http中間件”吧!

本文主要針對(duì)Golang的內(nèi)置庫(kù) net/http 做了簡(jiǎn)單的擴(kuò)展，通過(guò)添加中間件的形式實(shí)現(xiàn)了管道(Pipeline)模式，這樣的好處是各模塊之間是低耦合的，符合單一職責(zé)原則，可以很靈活的通過(guò)中間件的形式添加一些功能到管道中，一次請(qǐng)求和響應(yīng)在管道中的執(zhí)行過(guò)程如下

首先, 我定義了三個(gè)測(cè)試的中間件 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.

接下來(lái)，定義一個(gè) Pipeline 的方法，里面使用嵌套的形式, 使用了上面定義的三個(gè)測(cè)試的中間件.

func Pipeline(next http.Handler) http.Handler {

	return Middleware1(Middleware2(Middleware3(next)))

}

然后還需要業(yè)務(wù)代碼，這里我定義了 LoginHandler 和 RegisterHandler 兩個(gè)方法

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 接口上使用上面添加過(guò)中間件的 Pipeline

func main() {

	http.Handle("/Login", Pipeline(http.HandlerFunc(LoginHandler)))

	http.Handle("/Register", http.HandlerFunc(RegisterHandler))

	http.ListenAndServe(":8080", nil)

}

啟動(dòng)程序后，訪問 http://localhost:8080/Login, 程序的輸出如下，這和本文最上面的管道的流程圖是一致的，然后訪問 Register 接口, 控制臺(tái)沒有輸出信息，當(dāng)然也不會(huì)執(zhí)行任何中間件。

現(xiàn)在已經(jīng)實(shí)現(xiàn)了中間件的機(jī)制，但是，上面添加中間件是用嵌套的方法，這種方式不能說(shuō)不太優(yōu)雅，只能說(shuō)非常的Low，接下來(lái)我們需要對(duì)管道進(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
}

首先定義了一個(gè)Chain 的struct，用來(lái)接收添加到管道中的中間件，在 AddMiddlewares() 函數(shù)中，接收了多個(gè)Handle, 然后組裝到 Chain 對(duì)象并返回, 接下來(lái)調(diào)用 Then() 函數(shù), 把管道中的中間件和業(yè)務(wù)的Handler 關(guān)聯(lián)起來(lái)。在中間件的使用方式上， 這兩種方法都是一樣的，只需要調(diào)用 Pipeline() 方法就行了。

本文在go web中簡(jiǎn)單的實(shí)現(xiàn)了中間件的機(jī)制，這樣帶來(lái)的好處也是顯而易見的，當(dāng)然社區(qū)也有一些成熟的 middleware 組件，包括 Gin 一些Web框架中也包含了 middleware 相關(guān)的功能, 希望對(duì)您有用.

到此，相信大家對(duì)“怎么在Golang中實(shí)現(xiàn)一個(gè)簡(jiǎn)單的Http中間件”有了更深的了解，不妨來(lái)實(shí)際操作一番吧！這里是億速云網(wǎng)站，更多相關(guān)內(nèi)容可以進(jìn)入相關(guān)頻道進(jìn)行查詢，關(guān)注我們，繼續(xù)學(xué)習(xí)！