re-write with std usage of net/http

middleware/logging.go

@@ -0,0 +1,56 @@
+package middleware
+
+import (
+	"net/http"
+	"time"
+
+	"git.okseby.com/okseby/playground-go/logger"
+)
+
+// ANSI color codes
+const (
+	ColorReset  = "\033[0m"
+	ColorRed    = "\033[31m"
+	ColorGreen  = "\033[32m"
+	ColorYellow = "\033[33m"
+	ColorCyan   = "\033[36m"
+	ColorGrey   = "\033[90m"
+)
+
+type statusRecorder struct {
+	http.ResponseWriter
+	status int
+}
+
+func (r *statusRecorder) WriteHeader(code int) {
+	r.status = code
+	r.ResponseWriter.WriteHeader(code)
+}
+
+func Logger(next http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		start := time.Now()
+
+		recorder := &statusRecorder{ResponseWriter: w, status: 200}
+		next.ServeHTTP(recorder, r)
+
+		duration := time.Since(start)
+
+		statusColor := ColorGreen
+		switch {
+		case recorder.status >= 500:
+			statusColor = ColorRed
+		case recorder.status >= 400:
+			statusColor = ColorRed
+		case recorder.status >= 300:
+			statusColor = ColorYellow
+		}
+
+		logger.Info("%s%-7s%s %s%-30s%s %s%d%s [%s]",
+			ColorCyan, r.Method, ColorReset,
+			ColorGrey, r.URL.Path, ColorReset,
+			statusColor, recorder.status, ColorReset,
+			duration.Round(time.Millisecond),
+		)
+	})
+}