Golang中怎么自建一个HTTP中间件

这篇文章给大家介绍Golang中怎么自建一个HTTP中间件，内容非常详细，感兴趣的小伙伴们可以参考借鉴，希望对大家能有所帮助。

自建Http中间件

* 以类型形式* 以函数形式* 追加响应内容* 自定义响应

以类型形式：

package mainimport (        "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 mainimport (        "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中间件就分享到这里了，希望以上内容可以对大家有一定的帮助，可以学到更多知识。如果觉得文章不错，可以把它分享出去让更多的人看到。