feat(pkg/server/ginserver): add http_server_processing_request metric

boxgo · May 28, 2024 · 6570123 · 6570123
1 parent 7200326
commit 6570123
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/pkg/server/ginserver/mid/ginprom/config.go b/pkg/server/ginserver/mid/ginprom/config.go
@@ -69,5 +69,9 @@ func UrlMapping(c *gin.Context) string {
 		url = strings.Replace(url, "/"+p.Value, "/:"+p.Key, 1)
 	}
 
-	return url
+	if len(url) > 200 {
+		return url[:197] + "..."
+	}
+
+	return url[:200]
 }
diff --git a/pkg/server/ginserver/mid/ginprom/ginprom.go b/pkg/server/ginserver/mid/ginprom/ginprom.go
@@ -11,6 +11,7 @@ import (
 type (
 	GinProm struct {
 		cfg                *Config
+		processingGauge    *metric.GaugeVec
 		reqSizeSummary     *metric.SummaryVec
 		reqBeginCounter    *metric.CounterVec
 		reqFinishCounter   *metric.CounterVec
@@ -22,6 +23,11 @@ type (
 func newGinProm(c *Config) *GinProm {
 	return &GinProm{
 		cfg: c,
+		processingGauge: metric.NewGaugeVec(
+			"http_server_processing_request",
+			"http server processing request",
+			[]string{"method", "url"},
+		),
 		reqSizeSummary: metric.NewSummaryVec(
 			"http_server_request_size_bytes",
 			"The HTTP request sizes in bytes.",
@@ -81,9 +87,12 @@ func (prom *GinProm) Handler() gin.HandlerFunc {
 
 		reqSz := computeApproximateRequestSize(ctx.Request)
 
+		prom.processingGauge.WithLabelValues(labels...).Inc()
 		prom.reqSizeSummary.WithLabelValues(labels...).Observe(reqSz)
 		prom.reqBeginCounter.WithLabelValues(labels...).Inc()
 
+		defer prom.processingGauge.WithLabelValues(labels...).Dec()
+
 		ctx.Next()
 
 		resSz := ctx.Writer.Size()