From 2916082d5ed94ef86ad58bdb7256ae07b214c4f3 Mon Sep 17 00:00:00 2001
From: Alexander NeonXP Kiryukhin <i@neonxp.ru>
Date: Mon, 29 Jul 2024 02:38:17 +0300
Subject: Начальный коммит
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 middleware/logger.go | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 middleware/logger.go

(limited to 'middleware/logger.go')

diff --git a/middleware/logger.go b/middleware/logger.go
new file mode 100644
index 0000000..80117da
--- /dev/null
+++ b/middleware/logger.go
@@ -0,0 +1,48 @@
+package middleware
+
+import (
+	"log/slog"
+	"net/http"
+	"time"
+
+	"go.neonxp.ru/mux"
+)
+
+type wrappedResponse struct {
+	http.ResponseWriter
+	statusCode int
+}
+
+func (w *wrappedResponse) WriteHeader(code int) {
+	w.statusCode = code
+	w.ResponseWriter.WriteHeader(code)
+}
+
+func Logger(logger *slog.Logger) mux.Middleware {
+	return func(next http.Handler) http.Handler {
+		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			requestID := GetRequestID(r)
+			args := []any{
+				slog.String("proto", r.Proto),
+				slog.String("method", r.Method),
+				slog.String("request_uri", r.RequestURI),
+				slog.String("request_id", requestID),
+			}
+			logger.InfoContext(
+				r.Context(),
+				"start request",
+				args...,
+			)
+			t := time.Now()
+			wr := &wrappedResponse{ResponseWriter: w, statusCode: http.StatusOK}
+			next.ServeHTTP(wr, r)
+			args = append(args, slog.String("response_time", time.Since(t).String()))
+			args = append(args, slog.Int("response_status", wr.statusCode))
+			logger.InfoContext(
+				r.Context(),
+				"finish request",
+				args...,
+			)
+		})
+	}
+}
-- 
cgit v1.2.3