linka-cloud/grpc-toolkit
2
0
Fork 0
mirror of https://github.com/linka-cloud/grpc.git synced 2025-10-31 01:22:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

metrics: fix registration

Browse Source 
Signed-off-by: Adphi <philippe.adrien.nousse@gmail.com>
This commit is contained in:
Adphi
2022-03-12 00:24:37 +01:00
parent 2832f13f3c
commit b1988e0458
1 changed files with 30 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
interceptors/metrics/interceptors.go
View File

@@ -5,42 +5,63 @@ import (
"google.golang.org/grpc" "google.golang.org/grpc"
"go.linka.cloud/grpc/interceptors" "go.linka.cloud/grpc/interceptors"
"go.linka.cloud/grpc/service"
) )
var ( type Registerer interface {
Register = grpc_prometheus.Register Register(svc service.Service)
) }
type Interceptors interface {
interceptors.Interceptors
Registerer
}
type ServerInterceptors interface {
interceptors.ServerInterceptors
Registerer
}
type ClientInterceptors interface {
interceptors.ClientInterceptors
}
type metrics struct { type metrics struct {
s *grpc_prometheus.ServerMetrics s *grpc_prometheus.ServerMetrics
c *grpc_prometheus.ClientMetrics c *grpc_prometheus.ClientMetrics
} }
func NewInterceptors(opts ...grpc_prometheus.CounterOption) interceptors.Interceptors { func (m *metrics) Register(svc service.Service) {
if m.s != nil {
m.s.InitializeMetrics(svc)
}
}
func NewInterceptors(opts ...grpc_prometheus.CounterOption) Interceptors {
s := grpc_prometheus.NewServerMetrics(opts...) s := grpc_prometheus.NewServerMetrics(opts...)
c := grpc_prometheus.NewClientMetrics(opts...) c := grpc_prometheus.NewClientMetrics(opts...)
return &metrics{s: s, c: c} return &metrics{s: s, c: c}
} }
func NewServerInterceptors(opts ...grpc_prometheus.CounterOption) interceptors.ServerInterceptors { func NewServerInterceptors(opts ...grpc_prometheus.CounterOption) ServerInterceptors {
s := grpc_prometheus.NewServerMetrics(opts...) s := grpc_prometheus.NewServerMetrics(opts...)
return &metrics{s: s} return &metrics{s: s}
} }
func NewClientInterceptors(opts ...grpc_prometheus.CounterOption) interceptors.ClientInterceptors { func NewClientInterceptors(opts ...grpc_prometheus.CounterOption) ClientInterceptors {
c := grpc_prometheus.NewClientMetrics(opts...) c := grpc_prometheus.NewClientMetrics(opts...)
return &metrics{c: c} return &metrics{c: c}
} }
func DefaultInterceptors() interceptors.Interceptors { func DefaultInterceptors() Interceptors {
return &metrics{s: grpc_prometheus.DefaultServerMetrics, c: grpc_prometheus.DefaultClientMetrics} return &metrics{s: grpc_prometheus.DefaultServerMetrics, c: grpc_prometheus.DefaultClientMetrics}
} }
func DefaultServerInterceptors() interceptors.ServerInterceptors { func DefaultServerInterceptors() ServerInterceptors {
return &metrics{s: grpc_prometheus.DefaultServerMetrics} return &metrics{s: grpc_prometheus.DefaultServerMetrics}
} }
func DefaultClientInterceptors() interceptors.ClientInterceptors { func DefaultClientInterceptors() ClientInterceptors {
return &metrics{c: grpc_prometheus.DefaultClientMetrics} return &metrics{c: grpc_prometheus.DefaultClientMetrics}
} }