如何編寫 Go 中間件

女主宣言

大家有沒有寫過中間件呢，它是怎么寫的呢？今天給大家分享一下使用Go，如何編寫中間件，供大家參考學習。

PS：豐富的一線技術、多元化的表現形式，盡在“ HULK一線技術雜談 ”，點關注哦！

引言

一開始，我們認為編寫中間件似乎很容易，但是我們實際編寫的時候也會遇到各種各樣的坑。讓我們來看看一些例子。

1. 讀取請求

在我們的示例中，所有的中間件都將接受http。處理程序作為一個參數，并返回一個http.Handler。這使得人們很容易就能把中間產品串起來。我們所有的中間產品的基本模式是這樣的：

func X(h http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // Something here...
        h.ServeHTTP(w, r)
    })
}

我們想要將所有的請求重定向到一個斜杠——比方說/message/，到它們的非斜杠等效，比如/message。我們可以這樣寫：

func TrailingSlashRedirect(h http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        if r.URL.Path != "/" && r.URL.Path[len(r.URL.Path)-1] == '/' {
            http.Redirect(w, r, r.URL.Path[:len(r.URL.Path)-1], http.StatusMovedPermanently)
            return
        }
        h.ServeHTTP(w, r)
    })
}

有沒有很簡單。

2. 修改請求

比方說，我們想要向請求添加一個標題，或者修改它。http.Handler文檔中指明：

“

除了讀取主體之外，處理程序不應該修改所提供的請求。

Go標準庫復制http.Request。請求對象在將其傳遞到響應鏈之前，我們也應該這樣做。假設我們想在每個請求上設置Request-Id頭，以便內部跟蹤。創建*Request的淺副本，并在代理之前修改標題。

func RequestID(h http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        r2 := new(http.Request)
        *r2 = *r
        r2.Header.Set("X-Request-Id", uuid.NewV4().String())
        h.ServeHTTP(w, r2)
    })
}

3. 編寫響應頭

如果你想設置響應頭，你可以只寫它們，然后代理請求。

func Server(h http.Handler, servername string) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.Header().Set("Server", servername)
        h.ServeHTTP(w, r)
    })
}

上面的問題是，如果內部處理器也設置了服務器頭，那么你的頭將被覆蓋。如果不想公開內部軟件的服務器頭，或者如果想在發送響應給客戶端之前去掉頭部，這可能會帶來問題。

要做到這一點，我們必須自己實現ResponseWriter接口。大多數時候，我們只會代理到底層的ResponseWriter，但是如果用戶試圖寫一個響應，我們就會潛入并添加我們的標題。

type serverWriter struct {
    w            http.ResponseWriter
    name         string
    wroteHeaders bool
}

func (s *serverWriter) Header() http.Header {
    return s.w.Header()
}
func (s *serverWriter) WriteHeader(code int) http.Header {
    if s.wroteHeader == false {
        s.w.Header().Set("Server", s.name)
        s.wroteHeader = true
    }
    s.w.WriteHeader(code)
}
func (s serverWriter) Write(b []byte) (int, error) {
    if s.wroteHeader == false {
        // We hit this case if user never calls WriteHeader (default 200)
        s.w.Header().Set("Server", s.name)
        s.wroteHeader = true
    }   return s.w.Write(b)
}</code></pre> 
  
要在我們的中間件中使用它，我們會寫：
 
  
func Server(h http.Handler, servername string) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r http.Request) {
        sw := &serverWriter{
            w:    w,
            name: servername,
        }
        h.ServeHTTP(sw, r)
    })
}</code></pre> 
  
問題
 
  
如果用戶從不調用Write或WriteHeader呢？例如，有一個200狀態并且是空body，或者對選項請求的響應——我們的攔截函數都不會運行。因此，我們應該在ServeHTTP調用之后再添加校驗。
 
  
func Server(h http.Handler, servername string) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        sw := &serverWriter{
            w:    w,
            name: servername,
        }
        h.ServeHTTP(sw, r)
        if sw.wroteHeaders == false {
            s.w.Header().Set("Server", s.name)
            s.wroteHeader = true
        }
    })
}
 
  
其他ResponseWriter接口
 
  
ResponseWriter接口只需要有三種方法。但在實踐中，它也可以對其他接口作出響應，例如http.Pusher。你的中間件可能會意外地禁用HTTP/2支持，這是不好的。
 
  
// Push implements the http.Pusher interface.
func (s serverWriter) Push(target string, opts http.PushOptions) error {
    if pusher, ok := s.w.(http.Pusher); ok {
        return pusher.Push(target, opts)
    }
    return http.ErrNotSupported
}

// Flush implements the http.Flusher interface.
func (s *serverWriter) Flush() {
    f, ok := s.w.(http.Flusher)
    if ok {
        f.Flush()
    }
}</code></pre> 
  
總結
 
  
通過以上的學習，不知道大家對Go編寫中間件有沒有一個完整的認識。大家也可以嘗試著用Go去編寫一個中間件。如果有什么問題或是建議，可以通過公眾號下方的留言與我們溝通。
 
  
 
 
  
 
 
  
來自：https://mp.weixin.qq.com/s/-nRWwy8SjW1TlqCglL0CAQ