From 5d5f175f520068f5ca55108b42931e52342e83a9 Mon Sep 17 00:00:00 2001 From: Jan-Eric Hellenberg Date: Mon, 17 May 2021 15:38:34 +0200 Subject: [PATCH] Implement http.Hijack interface in HTTPLoggingMiddleware --- logging/logging.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/logging/logging.go b/logging/logging.go index 0e27fc3..d17b9d0 100644 --- a/logging/logging.go +++ b/logging/logging.go @@ -1,7 +1,9 @@ package logging import ( + "bufio" "github.com/sirupsen/logrus" + "net" "net/http" "os" "time" @@ -46,6 +48,10 @@ func (writer *loggingResponseWriter) WriteHeader(code int) { writer.ResponseWriter.WriteHeader(code) } +func (writer *loggingResponseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) { + return writer.ResponseWriter.(http.Hijacker).Hijack() +} + // HTTPLoggingMiddleware returns an http.Handler that logs different information about every request func HTTPLoggingMiddleware(next http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {