Как в gorilla mux сделать серверную мидлвару?

Question

[impressive17]

Мне нужно ловить ответ сервера, доставать из него тело ответа и логировать его, допустим в консоль.
Подскажите, как в мидлваре получить доступ к телу ответа?

Comments

[Rerurk]

. Body не?

[impressive17]

Rerurk, можешь пожалуйста пример кода показать?
В примере либы я вижу, что из next.ServeHTTP(w, r) нельзя получить ответ

func loggingMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // Do stuff here
        log.Println(r.RequestURI)
        // Call the next handler, which can be another middleware in the chain, or the final handler.
        next.ServeHTTP(w, r)
    })
}

Answer 1

[Евгений Мамонов]

Есть вариант использовать httptest.ResponseRecorder, я его использовал ранее для написания тестов.

https://pkg.go.dev/net/http/httptest#ResponseRecorder

Пример будет примерно таким

func (m *LoggingMiddleware) LogBody(next http.Handler) http.Handler {
    fn := func(w http.ResponseWriter, r *http.Request) {
        wRecorder := httptest.NewRecorder()
        next.ServeHTTP(wRecorder, r)
        resp := wRecorder.Result()
        body, _ := io.ReadAll(resp.Body)
        // не забыть записать ответ в w
    }

    return http.HandlerFunc(fn)
}

Пример его использования из официальной документации https://pkg.go.dev/net/http/httptest#example-Respo...

Или, можно сделать свой responseWriter, в handler передавать свой writer, а потом, после обработки, записать полученный ответ сервера в оригинальный responseWriter
Что-то типа такого

func (m *LoggingMiddleware) LogBody(next http.Handler) http.Handler {
    fn := func(w http.ResponseWriter, r *http.Request) {
        var buf bytes.Buffer
        logWriter := NewLogToConsoleResponseWriter(&b)
        next.ServeHTTP(logWriter, r)
        // тут у вас в buf будет ответ сервера, с которым можно работать
        // главное потом не забыть записать его в `w`
    }

    return http.HandlerFunc(fn)
}

Comments

[impressive17]

Евгений Мамонов Спасибо за ответ!
Напрямую не получается, так как не соответствует интерфейсу
Header() Header и WriteHeader(statusCode int) не реализованы у *bufio.Writer
Подскажите мне нужно их вручную реализовать?

[Rerurk]

Считывайте body в []byte, и unmarshal или что там у вас

[Евгений Мамонов]

Есть еще один вариант, скорее всего это прям то, что нужно

https://pkg.go.dev/net/http/httptest#ResponseRecorder, я его использовал ранее для написания тестов.

Пример будет примерно таким

func (m *LoggingMiddleware) LogBody(next http.Handler) http.Handler {
    fn := func(w http.ResponseWriter, r *http.Request) {
        wRecorder := httptest.NewRecorder()
        next.ServeHTTP(wRecorder r)
        resp := wRecorder.Result()
        body, _ := io.ReadAll(resp.Body)
        // не забыть записать ответ в w
    }

    return http.HandlerFunc(fn)
}

Пример его использования из официальной документации https://pkg.go.dev/net/http/httptest#example-Respo...

[impressive17]

Евгений Мамонов Это сработало, спасибо!

[Евгений Мамонов]

Рад помочь :)
Допишу этот комментарий в ответ, чтобы если кому то еще понадобится - не искали в комментариях :)