mirror of
https://github.com/foomo/gotsrpc.git
synced 2025-10-16 12:35:35 +00:00
52 lines
2.1 KiB
Go
52 lines
2.1 KiB
Go
package prometheus
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
p "github.com/prometheus/client_golang/prometheus"
|
|
|
|
"github.com/foomo/gotsrpc/v2"
|
|
)
|
|
|
|
func InstrumentService(s http.HandlerFunc) (handler http.HandlerFunc) {
|
|
requestDuration := p.NewSummaryVec(p.SummaryOpts{ //nolint:promlinter
|
|
Namespace: "gotsrpc",
|
|
Subsystem: "service",
|
|
Name: "time_nanoseconds",
|
|
Help: "nanoseconds to unmarshal requests, execute a service function and marshal its reponses",
|
|
Objectives: map[float64]float64{0.5: 0.05, 0.9: 0.01, 0.99: 0.001},
|
|
}, []string{"package", "service", "func", "type"})
|
|
requestSize := p.NewSummaryVec(p.SummaryOpts{
|
|
Namespace: "gotsrpc",
|
|
Subsystem: "service",
|
|
Name: "size_bytes",
|
|
Help: "request and response sizes in bytes",
|
|
Objectives: map[float64]float64{0.5: 0.05, 0.9: 0.01, 0.99: 0.001},
|
|
}, []string{"package", "service", "func", "type"})
|
|
|
|
p.MustRegister(requestSize)
|
|
p.MustRegister(requestDuration)
|
|
|
|
return gotsrpc.InstrumentedService(s.ServeHTTP, func(stats *gotsrpc.CallStats) {
|
|
requestSize.WithLabelValues(stats.Package, stats.Service, stats.Func, "request").Observe(float64(stats.RequestSize))
|
|
requestSize.WithLabelValues(stats.Package, stats.Service, stats.Func, "response").Observe(float64(stats.ResponseSize))
|
|
requestDuration.WithLabelValues(stats.Package, stats.Service, stats.Func, "unmarshalling").Observe(float64(stats.Unmarshalling))
|
|
requestDuration.WithLabelValues(stats.Package, stats.Service, stats.Func, "execution").Observe(float64(stats.Execution))
|
|
requestDuration.WithLabelValues(stats.Package, stats.Service, stats.Func, "marshalling").Observe(float64(stats.Marshalling))
|
|
})
|
|
}
|
|
|
|
func InstrumentGoRPCService() gotsrpc.GoRPCCallStatsHandlerFun {
|
|
callsCounter := p.NewSummaryVec(p.SummaryOpts{
|
|
Namespace: "gorpc",
|
|
Subsystem: "service",
|
|
Name: "time_seconds",
|
|
Help: "seconds to execute a service method",
|
|
}, []string{"package", "service", "func", "type"})
|
|
p.MustRegister(callsCounter)
|
|
|
|
return func(stats *gotsrpc.CallStats) {
|
|
callsCounter.WithLabelValues(stats.Package, stats.Service, stats.Func, "execution").Observe(float64(stats.Execution))
|
|
}
|
|
}
|