mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-11-04 19:32:17 +08:00
34 lines
720 B
Go
34 lines
720 B
Go
package prommetrics
|
|
|
|
import (
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
"net/http"
|
|
"strconv"
|
|
)
|
|
|
|
const RpcPath = "/metrics"
|
|
|
|
var (
|
|
rpcRegistry = prometheus.NewRegistry()
|
|
rpcCounter = prometheus.NewCounterVec(
|
|
prometheus.CounterOpts{
|
|
Name: "rpc_count",
|
|
Help: "Total number of RPC calls",
|
|
},
|
|
[]string{"name", "path", "code"},
|
|
)
|
|
)
|
|
|
|
func init() {
|
|
rpcRegistry.MustRegister(rpcCounter)
|
|
}
|
|
|
|
func RPCCall(name string, path string, code int) {
|
|
rpcCounter.With(prometheus.Labels{"name": name, "path": path, "code": strconv.Itoa(code)}).Inc()
|
|
}
|
|
|
|
func RPCHandler() http.Handler {
|
|
return promhttp.HandlerFor(rpcRegistry, promhttp.HandlerOpts{})
|
|
}
|