forked from TrueCloudLab/neoneo-go
40 lines
731 B
Go
40 lines
731 B
Go
package rpcsrv
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
)
|
|
|
|
// Metrics used in monitoring service.
|
|
var (
|
|
rpcTimes = map[string]prometheus.Histogram{}
|
|
)
|
|
|
|
func addReqTimeMetric(name string, t time.Duration) {
|
|
hist, ok := rpcTimes[name]
|
|
if ok {
|
|
hist.Observe(t.Seconds())
|
|
}
|
|
}
|
|
|
|
func regCounter(call string) {
|
|
rpcTimes[call] = prometheus.NewHistogram(
|
|
prometheus.HistogramOpts{
|
|
Help: "RPC " + call + " call handling time",
|
|
Name: "rpc_" + strings.ToLower(call) + "_time",
|
|
Namespace: "neogo",
|
|
},
|
|
)
|
|
prometheus.MustRegister(rpcTimes[call])
|
|
}
|
|
|
|
func init() {
|
|
for call := range rpcHandlers {
|
|
regCounter(call)
|
|
}
|
|
for call := range rpcWsHandlers {
|
|
regCounter(call)
|
|
}
|
|
}
|