6a21c96b55
Fixes #9806 go-grpc-prometheus is deprecated. The new location it was moved to also introduced an entirely new api, but afaict this matches what we have at the moment. Signed-off-by: Danny Canter <danny@dcantah.dev>
49 lines
1.4 KiB
Go
49 lines
1.4 KiB
Go
// Copyright (c) The go-grpc-middleware Authors.
|
|
// Licensed under the Apache License 2.0.
|
|
|
|
package prometheus
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
)
|
|
|
|
type exemplarFromCtxFn func(ctx context.Context) prometheus.Labels
|
|
|
|
type serverMetricsConfig struct {
|
|
counterOpts counterOptions
|
|
// serverHandledHistogram can be nil.
|
|
serverHandledHistogram *prometheus.HistogramVec
|
|
}
|
|
|
|
type ServerMetricsOption func(*serverMetricsConfig)
|
|
|
|
func (c *serverMetricsConfig) apply(opts []ServerMetricsOption) {
|
|
for _, o := range opts {
|
|
o(c)
|
|
}
|
|
}
|
|
|
|
// WithServerCounterOptions sets counter options.
|
|
func WithServerCounterOptions(opts ...CounterOption) ServerMetricsOption {
|
|
return func(o *serverMetricsConfig) {
|
|
o.counterOpts = opts
|
|
}
|
|
}
|
|
|
|
// WithServerHandlingTimeHistogram turns on recording of handling time of RPCs.
|
|
// Histogram metrics can be very expensive for Prometheus to retain and query.
|
|
func WithServerHandlingTimeHistogram(opts ...HistogramOption) ServerMetricsOption {
|
|
return func(o *serverMetricsConfig) {
|
|
o.serverHandledHistogram = prometheus.NewHistogramVec(
|
|
histogramOptions(opts).apply(prometheus.HistogramOpts{
|
|
Name: "grpc_server_handling_seconds",
|
|
Help: "Histogram of response latency (seconds) of gRPC that had been application-level handled by the server.",
|
|
Buckets: prometheus.DefBuckets,
|
|
}),
|
|
[]string{"grpc_type", "grpc_service", "grpc_method"},
|
|
)
|
|
}
|
|
}
|