main
lzf 2025-07-22 17:44:04 +08:00
parent 063fc82338
commit 1eb10eb1af
1 changed files with 3 additions and 1 deletions

View File

@ -7,6 +7,8 @@ import (
"time" "time"
) )
const CtxCollectRequestFrom = "ctx_collect_request_from"
func QPSCollect(svcName string) gin.HandlerFunc { func QPSCollect(svcName string) gin.HandlerFunc {
hs := mymetric.NewQPSHistogram(svcName, []string{ hs := mymetric.NewQPSHistogram(svcName, []string{
"svc", "method", "route", "from", "svc", "method", "route", "from",
@ -21,7 +23,7 @@ func QPSCollect(svcName string) gin.HandlerFunc {
"svc": svcName, "svc": svcName,
"method": ctx.Request.Method, "method": ctx.Request.Method,
"route": ctx.Request.URL.Path, "route": ctx.Request.URL.Path,
"from": "", "from": ctx.GetString(CtxCollectRequestFrom),
}).Observe(float64(time.Since(st).Milliseconds())) }).Observe(float64(time.Since(st).Milliseconds()))
} }