gotsrpc/prometheus/prometheus.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))
	}
}