move rpc prometheus config to metric service

metric/drpc.go

@@ -6,12 +6,12 @@ import (
 	"time"
 )
 
-type PrometheusDRPC struct {
+type prometheusDRPC struct {
 	drpc.Handler
 	SummaryVec *prometheus.SummaryVec
 }
 
-func (ph *PrometheusDRPC) HandleRPC(stream drpc.Stream, rpc string) (err error) {
+func (ph *prometheusDRPC) HandleRPC(stream drpc.Stream, rpc string) (err error) {
 	st := time.Now()
 	defer func() {
 		ph.SummaryVec.WithLabelValues(rpc).Observe(time.Since(st).Seconds())

metric/metric.go

@@ -3,32 +3,40 @@ package metric
 import (
 	"context"
 	"github.com/anytypeio/any-sync/app"
+	"github.com/anytypeio/any-sync/app/logger"
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/prometheus/client_golang/prometheus/collectors"
 	"github.com/prometheus/client_golang/prometheus/promhttp"
+	"go.uber.org/zap"
 	"net/http"
+	"storj.io/drpc"
 	"time"
 )
 
 const CName = "common.metric"
 
+var log = logger.NewNamed(CName)
+
 func New() Metric {
 	return new(metric)
 }
 
 type Metric interface {
 	Registry() *prometheus.Registry
+	WrapDRPCHandler(h drpc.Handler) drpc.Handler
 	app.ComponentRunnable
 }
 
 type metric struct {
 	registry *prometheus.Registry
+	rpcLog   logger.CtxLogger
 	config   Config
 }
 
 func (m *metric) Init(a *app.App) (err error) {
 	m.registry = prometheus.NewRegistry()
 	m.config = a.MustComponent("config").(configSource).GetMetric()
+	m.rpcLog = logger.NewNamed("rpcLog")
 	return nil
 }
 
@@ -61,6 +69,31 @@ func (m *metric) Registry() *prometheus.Registry {
 	return m.registry
 }
 
+func (m *metric) WrapDRPCHandler(h drpc.Handler) drpc.Handler {
+	if m == nil {
+		return h
+	}
+	histVec := prometheus.NewSummaryVec(prometheus.SummaryOpts{
+		Namespace: "drpc",
+		Subsystem: "server",
+		Name:      "duration_seconds",
+		Objectives: map[float64]float64{
+			0.5:  0.5,
+			0.85: 0.01,
+			0.95: 0.0005,
+			0.99: 0.0001,
+		},
+	}, []string{"rpc"})
+	if err := m.Registry().Register(histVec); err != nil {
+		log.Warn("can't register prometheus drpc metric", zap.Error(err))
+		return h
+	}
+	return &prometheusDRPC{
+		Handler:    h,
+		SummaryVec: histVec,
+	}
+}
+
 func (m *metric) Close(ctx context.Context) (err error) {
 	return
 }