separated logger middleware

internal/server/router/mux.go

@@ -0,0 +1,36 @@
+package router
+
+import (
+	"net/http"
+
+	"ruben/inventory2/internal/server/response"
+)
+
+type (
+	Mux struct {
+		Mux        *http.ServeMux
+		middleware []response.Middleware
+	}
+)
+
+func NewMux(ms ...response.Middleware) *Mux {
+	return &Mux{
+		Mux:        http.NewServeMux(),
+		middleware: ms,
+	}
+}
+
+func (m *Mux) Handle(pattern string, fn response.HandlerFunc) {
+	for _, mw := range m.middleware {
+		prev := fn
+		fn = mw(func(r *http.Request) (response.Response, error) {
+			return prev(r)
+		})
+	}
+
+	m.Mux.Handle(pattern, response.Handler(fn))
+}
+
+func (m *Mux) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	m.Mux.ServeHTTP(w, r)
+}