pub

pkg/middleware/logger.go

@@ -0,0 +1,41 @@
+package middleware
+
+import (
+	"log"
+	"net/http"
+	"time"
+)
+
+// responseWriter is a custom wrapper to capture the status code
+type responseWriter struct {
+	http.ResponseWriter
+	status int
+}
+
+func (rw *responseWriter) WriteHeader(code int) {
+	rw.status = code
+	rw.ResponseWriter.WriteHeader(code)
+}
+
+// Logger is a middleware that logs each incoming HTTP request
+func Logger() Middleware {
+	return func(next http.Handler) http.Handler {
+		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			start := time.Now()
+			
+			// Default status is 200 OK
+			rw := &responseWriter{ResponseWriter: w, status: http.StatusOK}
+			
+			next.ServeHTTP(rw, r)
+			
+			log.Printf(
+				"[%s] %s %s | %d | %v",
+				r.Method,
+				r.Host,
+				r.URL.Path,
+				rw.status,
+				time.Since(start),
+			)
+		})
+	}
+}