Golang中怎么自建一個(gè)HTTP中間件

這篇文章給大家介紹Golang中怎么自建一個(gè)HTTP中間件，內(nèi)容非常詳細(xì)，感興趣的小伙伴們可以參考借鑒，希望對(duì)大家能有所幫助。

自建Http中間件

* 以類型形式
* 以函數(shù)形式
* 追加響應(yīng)內(nèi)容
* 自定義響應(yīng)

以類型形式：

package main

import (
	"net/http"
)

type SingleHost struct {
	handler     http.Handler
	allowedHost string
}

func (this *SingleHost) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	if r.Host == this.allowedHost {
		this.handler.ServeHTTP(w, r)
	} else {
		w.WriteHeader(403)
	}
}

func myHandler(w http.ResponseWriter, r *http.Request) {
	w.Write([]byte("hello world!"))
}

func main() {
	single := &SingleHost{
		handler:     http.HandlerFunc(myHandler),
		allowedHost: "baidu.com",
	}
	http.ListenAndServe(":8080", single)
}

以函數(shù)形式：

package main

import (
	"net/http"
)

func SingleHost(handler http.Handler, allowedHost string) http.Handler {
	fn := func(w http.ResponseWriter, r *http.Request) {
		if r.Host == r.allowedHost {
			handler.ServeHTTP(w, r)
		} else {
			w.WriteHeader(403)
		}
	}
	return http.HandlerFunc(fn)
}

func myHandler(w http.ResponseWriter,r *http.Request){
	w.Write([]byte("hello world!"))
}

func main(){
	single:=SingleHost(http.HandlerFunc(myHandler),"localhost:8080")
	http.ListenAndServe(":8080",single)
}

自定義響應(yīng)： net/http/httptest：響應(yīng)請(qǐng)求時(shí)先進(jìn)記錄，不直接響應(yīng)客戶端，在完全自定義時(shí)，還沒準(zhǔn)備好響應(yīng)給客戶端，先將請(qǐng)求虛擬記錄下來，沒有真正的相應(yīng)出去，都在一個(gè)記錄過程當(dāng)中，在完全處理完后，在將響應(yīng)一整套的響應(yīng)給客戶。

package main

//追加響應(yīng)，傳入handler 提示經(jīng)過中間件處理
import (
	"net/http"
)

type AppendHiddleware struct{
	handler http.Handler
}

func (this *AppendHiddleware)ServeHTTP(w http.ResponseWriter,r *http.Request){

	if r.Host=="ccc"{
		this.handler.ServeHTTP(w,r)
		w.Write([]byte("Hey,this is middleware"))
	}else{
		w.Write([]byte("no host"))
	}
	
}

func myHandle(w http.ResponseWriter,r *http.Request){
	w.Write([]byte("hello myHandle!"))
}

func main(){
	mid:=&AppendHiddleware{http.HandlerFunc(myHandle)}

	http.ListenAndServe(":8080",mid)
}

關(guān)于Golang中怎么自建一個(gè)HTTP中間件就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，可以學(xué)到更多知識(shí)。如果覺得文章不錯(cuò)，可以把它分享出去讓更多的人看到。