mirror of https://github.com/linka-cloud/grpc.git
58 lines
1.6 KiB
Go
58 lines
1.6 KiB
Go
package metrics
|
|
|
|
import (
|
|
grpc_prometheus "github.com/grpc-ecosystem/go-grpc-prometheus"
|
|
"google.golang.org/grpc"
|
|
|
|
"go.linka.cloud/grpc/interceptors"
|
|
)
|
|
|
|
type metrics struct {
|
|
s *grpc_prometheus.ServerMetrics
|
|
c *grpc_prometheus.ClientMetrics
|
|
}
|
|
|
|
func NewInterceptors(opts ...grpc_prometheus.CounterOption) interceptors.Interceptors {
|
|
s := grpc_prometheus.NewServerMetrics(opts...)
|
|
c := grpc_prometheus.NewClientMetrics(opts...)
|
|
return &metrics{s: s, c: c}
|
|
}
|
|
|
|
func NewServerInterceptors(opts ...grpc_prometheus.CounterOption) interceptors.ServerInterceptors {
|
|
s := grpc_prometheus.NewServerMetrics(opts...)
|
|
return &metrics{s: s}
|
|
}
|
|
|
|
func NewClientInterceptors(opts ...grpc_prometheus.CounterOption) interceptors.ClientInterceptors {
|
|
c := grpc_prometheus.NewClientMetrics(opts...)
|
|
return &metrics{c: c}
|
|
}
|
|
|
|
func DefaultInterceptors() interceptors.Interceptors {
|
|
return &metrics{s: grpc_prometheus.DefaultServerMetrics, c: grpc_prometheus.DefaultClientMetrics}
|
|
}
|
|
|
|
func DefaultServerInterceptors() interceptors.ServerInterceptors {
|
|
return &metrics{s: grpc_prometheus.DefaultServerMetrics}
|
|
}
|
|
|
|
func DefaultClientInterceptors() interceptors.ClientInterceptors {
|
|
return &metrics{c: grpc_prometheus.DefaultClientMetrics}
|
|
}
|
|
|
|
func (m *metrics) UnaryServerInterceptor() grpc.UnaryServerInterceptor {
|
|
return m.s.UnaryServerInterceptor()
|
|
}
|
|
|
|
func (m *metrics) StreamServerInterceptor() grpc.StreamServerInterceptor {
|
|
return m.s.StreamServerInterceptor()
|
|
}
|
|
|
|
func (m *metrics) UnaryClientInterceptor() grpc.UnaryClientInterceptor {
|
|
return m.c.UnaryClientInterceptor()
|
|
}
|
|
|
|
func (m *metrics) StreamClientInterceptor() grpc.StreamClientInterceptor {
|
|
return m.c.StreamClientInterceptor()
|
|
}
|