大橙子网站建设,新征程启航
为企业提供网站建设、域名注册、服务器等服务
这篇文章给大家介绍Golang中怎么自建一个HTTP中间件,内容非常详细,感兴趣的小伙伴们可以参考借鉴,希望对大家能有所帮助。
成都创新互联公司主要从事网站设计、做网站、网页设计、企业做网站、公司建网站等业务。立足成都服务乐至,十年网站建设经验,价格优惠、服务专业,欢迎来电咨询建站服务:18982081108
自建Http中间件
* 以类型形式 * 以函数形式 * 追加响应内容 * 自定义响应
以类型形式:
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) }
以函数形式:
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) }
自定义响应: net/http/httptest:响应请求时先进记录,不直接响应客户端,在完全自定义时,还没准备好响应给客户端,先将请求虚拟记录下来,没有真正的相应出去,都在一个记录过程当中,在完全处理完后,在将响应一整套的响应给客户。
package main //追加响应,传入handler 提示经过中间件处理 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) }
关于Golang中怎么自建一个HTTP中间件就分享到这里了,希望以上内容可以对大家有一定的帮助,可以学到更多知识。如果觉得文章不错,可以把它分享出去让更多的人看到。