Function NewStatsHandler

middleware/grpc_stats.go:17–37

NewStatsHandler creates handler that can be added to gRPC server options to track received and sent message sizes.

(reg prometheus.Registerer, receivedPayloadSize, sentPayloadSize *prometheus.HistogramVec, inflightRequests *prometheus.GaugeVec, collectMaxStreamsByConn bool)

// NewStatsHandler creates handler that can be added to gRPC server options to track received and sent message sizes.
func NewStatsHandler(reg prometheus.Registerer, receivedPayloadSize, sentPayloadSize *prometheus.HistogramVec, inflightRequests *prometheus.GaugeVec, collectMaxStreamsByConn bool) stats.Handler {
	var streamTracker *StreamTracker
	if collectMaxStreamsByConn {
		grpcConcurrentStreamsByConnMax := prometheus.NewDesc(
			"grpc_concurrent_streams_by_conn_max",
			"The current number of concurrent streams in the connection with the most concurrent streams.",
			[]string{},
			prometheus.Labels{},
		)
		streamTracker = NewStreamTracker(grpcConcurrentStreamsByConnMax)
		reg.MustRegister(streamTracker)
	}

	return &grpcStatsHandler{
		receivedPayloadSize: receivedPayloadSize,
		sentPayloadSize:     sentPayloadSize,
		inflightRequests:    inflightRequests,

		grpcConcurrentStreamsTracker: streamTracker,
	}
}

type grpcStatsHandler struct {
	receivedPayloadSize *prometheus.HistogramVec