如何在 Gin 中間件中擷取並記錄回應正文？

在Gin 中間件中記錄回應正文

在中間件中，您可能需要擷取並記錄回應正文以用於偵錯或審核目的。那麼，我們如何從 Gin 的中間件上下文存取回應主體呢？

Gin 在內部管理回應寫入，這使得直接取得回應主體變得很棘手。為了克服這個問題，我們需要實作自己的 Writer 來攔截寫入操作。以下是逐步方法：

1. 建立自訂編寫器：

type bodyLogWriter struct {
    gin.ResponseWriter
    body *bytes.Buffer
}

func (w bodyLogWriter) Write(b []byte) (int, error) {
    w.body.Write(b)
    return w.ResponseWriter.Write(b)
}

此編寫器在寫入回應正文的同時緩衝響應正文致原作者。

2. 實作中間件：

func ginBodyLogMiddleware(c *gin.Context) {
    blw := &bodyLogWriter{body: bytes.NewBufferString(""), ResponseWriter: c.Writer}
    c.Writer = blw
    c.Next()

     statusCode := c.Writer.Status()
    if statusCode >= 400 {
        fmt.Println("Response body: " + blw.body.String())
    }
}

在中間件中，我們將自訂writer blw 指派給上下文，確保它捕捉所有響應。然後，如果狀態代碼有錯誤，我們會記錄正文。

3. 使用中間件：

router.Use(ginBodyLogMiddleware)

現在，當請求由您的Gin 路由器處理，中間件將攔截並帶有記錄錯誤狀態代碼的請求的回應正文。

請注意，這方法不會記錄靜態檔案的回應主體，但它應該解決大多數動態內容的用例。如果您需要攔截所有文件，則需要在 Gin 引擎周圍使用更複雜的包裝器。

以上是如何在 Gin 中間件中擷取並記錄回應正文？的詳細內容。更多資訊請關注PHP中文網其他相關文章！