init

internal/adapters/logger/http.go

@@ -0,0 +1,61 @@
+package logger
+
+import (
+	"github.com/rs/zerolog"
+	"net/http"
+	"time"
+)
+
+// responseData contains the data from the response.
+type responseData struct {
+	status int
+}
+
+// loggingResponseWriter is a wrapper around http.ResponseWriter that keeps track of its HTTP status code and check response data errors
+type loggingResponseWriter struct {
+	http.ResponseWriter
+	responseData *responseData
+}
+
+func (r *loggingResponseWriter) WriteHeader(statusCode int) {
+	r.ResponseWriter.WriteHeader(statusCode)
+	r.responseData.status = statusCode
+}
+
+// HTTPLoggerFunc create middleware that get status code and error message from response writer with Adapted logger
+func HTTPLoggerFunc(logger zerolog.Logger, handler http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		start := time.Now()
+		ipAddress := r.Header.Get("X-Real-Ip")
+		if ipAddress == "" {
+			ipAddress = r.Header.Get("X-Forwarded-For")
+		}
+
+		responseData := &responseData{}
+		ww := loggingResponseWriter{
+			ResponseWriter: w,
+			responseData:   responseData,
+		}
+
+		handler.ServeHTTP(&ww, r)
+
+		switch responseData.status {
+
+		case 200, 201, 202, 204:
+			logger.Info().
+				Dur("ms", time.Since(start)).
+				Str("url", r.URL.String()).
+				Str("mtd", r.Method).
+				Int("sts", responseData.status).
+				Str("ip", ipAddress).Send()
+		default:
+			logger.Error().
+				Dur("ms", time.Since(start)).
+				Str("url", r.URL.String()).
+				Int("sts", responseData.status).
+				Str("mtd", r.Method).
+				Str("ip", ipAddress).
+				Str("tkn", r.Header.Get("X-API-Key")).Send()
+		}
+	})
+}