refactor(metrics): remove external library

Author: TheFox0x7

routers/common/middleware.go

@@ -8,6 +8,7 @@ import (
 	"net/http"
 	"strconv"
 	"strings"
+	"time"
 
 	"code.gitea.io/gitea/modules/cache"
 	"code.gitea.io/gitea/modules/httplib"
@@ -18,7 +19,6 @@ import (
 
 	"gitea.com/go-chi/session"
 	"github.com/chi-middleware/proxy"
-	"github.com/felixge/httpsnoop"
 	"github.com/go-chi/chi/v5"
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/prometheus/client_golang/prometheus/promauto"
@@ -145,13 +145,14 @@ func RouteMetrics() func(h http.Handler) http.Handler {
 			inflight := reqInflightGauge.WithLabelValues(req.Method)
 			inflight.Inc()
 			defer inflight.Dec()
+			start := time.Now()
 
-			m := httpsnoop.CaptureMetrics(next, resp, req)
 			next.ServeHTTP(resp, req)
+			m := context.WrapResponseWriter(resp)
 			route := chi.RouteContext(req.Context()).RoutePattern()
-			code := strconv.Itoa(m.Code)
-			reqDurationHistogram.WithLabelValues(req.Method, code, route).Observe(m.Duration.Seconds())
-			respSizeHistogram.WithLabelValues(req.Method, code, route).Observe(float64(m.Written))
+			code := strconv.Itoa(m.WrittenStatus())
+			reqDurationHistogram.WithLabelValues(req.Method, code, route).Observe(time.Since(start).Seconds())
+			respSizeHistogram.WithLabelValues(req.Method, code, route).Observe(float64(m.Size()))
 			size := req.ContentLength
 			if size < 0 {
 				size = 0

routers/common/middleware_test.go

@@ -0,0 +1,30 @@
+package common_test
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	"time"
+
+	"code.gitea.io/gitea/routers/common"
+
+	"github.com/go-chi/chi/v5"
+	"github.com/stretchr/testify/require"
+)
+
+func TestMetricsMiddlewere(t *testing.T) {
+
+	middleware := common.RouteMetrics()
+	r := chi.NewRouter()
+	r.Use(middleware)
+	r.Get("/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Write([]byte("test"))
+		time.Sleep(5 * time.Millisecond)
+	}))
+
+	testServer := httptest.NewServer(r)
+
+	_, err := http.Get(testServer.URL)
+	require.NoError(t, err)
+
+}