package handler import ( "net/http" "strconv" "time" "github.com/tal-tech/go-zero/core/metric" "github.com/tal-tech/go-zero/core/timex" "github.com/tal-tech/go-zero/rest/internal/security" ) const serverNamespace = "http_server" var ( metricServerReqDur = metric.NewHistogramVec(&metric.HistogramVecOpts{ Namespace: serverNamespace, Subsystem: "requests", Name: "duration_ms", Help: "http server requests duration(ms).", Labels: []string{"path"}, Buckets: []float64{5, 10, 25, 50, 100, 250, 500, 1000}, }) metricServerReqCodeTotal = metric.NewCounterVec(&metric.CounterVecOpts{ Namespace: serverNamespace, Subsystem: "requests", Name: "code_total", Help: "http server requests error count.", Labels: []string{"path", "code"}, }) ) func PromethousHandler(path string) func(http.Handler) http.Handler { return func(next http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { startTime := timex.Now() cw := &security.WithCodeResponseWriter{Writer: w} defer func() { metricServerReqDur.Observe(int64(timex.Since(startTime)/time.Millisecond), path) metricServerReqCodeTotal.Inc(path, strconv.Itoa(cw.Code)) }() next.ServeHTTP(cw, r) }) } }