metrics

README

metrics

The grpc's server-side and client-side metrics can continue to be captured using prometheus.

Example of use

grpc server

import "github.com/zhufuyi/sponge/pkg/grpc/metrics"

func UnaryServerLabels(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
	// set up prometheus custom labels
	tag := grpc_ctxtags.NewTags().
		Set(serverNameLabelKey, serverNameLabelValue).
		Set(envLabelKey, envLabelValue)
	newCtx := grpc_ctxtags.SetInContext(ctx, tag)

	return handler(newCtx, req)
}

func getServerOptions() []grpc.ServerOption {
	var options []grpc.ServerOption

	// metrics interceptor
	option := grpc.ChainUnaryInterceptor(
		//UnaryServerLabels,                  // tag
		metrics.UnaryServerMetrics(
			// metrics.WithCounterMetrics(customizedCounterMetric) // adding custom metrics
		),
	)
	options = append(options, option)

	option = grpc.ChainStreamInterceptor(
		metrics.StreamServerMetrics(), // metrics interceptor for streaming rpc
	)
	options = append(options, option)

	return options
}

func main() {
	rand.Seed(time.Now().UnixNano())

	addr := ":8282"
	fmt.Println("start rpc server", addr)

	list, err := net.Listen("tcp", addr)
	if err != nil {
		panic(err)
	}

	server := grpc.NewServer(getServerOptions()...)
	serverNameV1.RegisterGreeterServer(server, &GreeterServer{})

	// start metrics server, collect grpc metrics by default, turn on, go metrics
	metrics.ServerHTTPService(":8283", server)
	fmt.Println("start metrics server", ":8283")

	err = server.Serve(list)
	if err != nil {
		panic(err)
	}
}

grpc client

import "github.com/zhufuyi/sponge/pkg/grpc/metrics"

func getDialOptions() []grpc.DialOption {
	var options []grpc.DialOption

	// use insecure transfer
	options = append(options, grpc.WithTransportCredentials(insecure.NewCredentials()))

	// Metrics
	options = append(options, grpc.WithUnaryInterceptor(metrics.UnaryClientMetrics()))
	options = append(options, grpc.WithStreamInterceptor(metrics.StreamClientMetrics()))
	return options
}

func main() {
	conn, err := grpc.NewClient("127.0.0.1:8282", getDialOptions()...)

	metrics.ClientHTTPService(":8284")
	fmt.Println("start metrics server", ":8284")

	client := serverNameV1.NewGreeterClient(conn)
	i := 0
	for {
		i++
		time.Sleep(time.Millisecond * 500) // qps is 2
		err = sayHello(client, i)
		if err != nil {
			fmt.Println(err)
		}
	}
}

Documentation

Overview

Package metrics is grpc's server-side and client-side metrics can continue to be captured using prometheus.

Index

• func ClientHTTPService(addr string) *http.Server
• func ClientRegister(mux *http.ServeMux)
• func Register(mux *http.ServeMux, grpcServer *grpc.Server)
• func ServerHTTPService(addr string, grpcServer *grpc.Server) *http.Server
• func SetClientPattern(pattern string)
• func SetServerPattern(pattern string)
• func StreamClientMetrics() grpc.StreamClientInterceptor
• func StreamServerMetrics(opts ...Option) grpc.StreamServerInterceptor
• func UnaryClientMetrics() grpc.UnaryClientInterceptor
• func UnaryServerMetrics(opts ...Option) grpc.UnaryServerInterceptor
• type ConnectionOption
  • func WithConnectionsGauge() ConnectionOption
  • func WithConnectionsLogger(l *zap.Logger) ConnectionOption
• type CustomConn
  • func (c *CustomConn) Close() error
• type CustomListener
  • func NewCustomListener(listener net.Listener, opts ...ConnectionOption) *CustomListener
  • func (l *CustomListener) Accept() (net.Conn, error)
  • func (l *CustomListener) GetActiveConnections() int
• type Option
  • func WithCounterMetrics(metrics ...*prometheus.CounterVec) Option
  • func WithGaugeMetrics(metrics ...*prometheus.GaugeVec) Option
  • func WithHistogramMetrics(metrics ...*prometheus.HistogramVec) Option
  • func WithSummaryMetrics(metrics ...*prometheus.SummaryVec) Option

Functions

func ClientHTTPService

func ClientHTTPService(addr string) *http.Server

ClientHTTPService initialize the client's prometheus exporter service and use http://ip:port/metrics to fetch data

func ClientRegister

func ClientRegister(mux *http.ServeMux)

ClientRegister for http routing and grpc methods

func Register

func Register(mux *http.ServeMux, grpcServer *grpc.Server)

Register for http routing and grpc methods

func ServerHTTPService

func ServerHTTPService(addr string, grpcServer *grpc.Server) *http.Server

ServerHTTPService initialize the prometheus exporter service on the server side and fetch data using http://ip:port/metrics

func SetClientPattern

func SetClientPattern(pattern string)

SetClientPattern set the client pattern

func SetServerPattern

func SetServerPattern(pattern string)

SetServerPattern set the server pattern

func StreamClientMetrics

func StreamClientMetrics() grpc.StreamClientInterceptor

StreamClientMetrics metrics stream interceptor

func StreamServerMetrics

func StreamServerMetrics(opts ...Option) grpc.StreamServerInterceptor

StreamServerMetrics metrics stream interceptor

func UnaryClientMetrics

func UnaryClientMetrics() grpc.UnaryClientInterceptor

UnaryClientMetrics metrics unary interceptor

func UnaryServerMetrics

func UnaryServerMetrics(opts ...Option) grpc.UnaryServerInterceptor

UnaryServerMetrics metrics unary interceptor

Types

type ConnectionOption

type ConnectionOption func(*connectionOptions)

ConnectionOption set connection option

func WithConnectionsGauge

func WithConnectionsGauge() ConnectionOption

WithConnectionsGauge set prometheus gauge for connections

func WithConnectionsLogger

func WithConnectionsLogger(l *zap.Logger) ConnectionOption

WithConnectionsLogger set logger for connection

type CustomConn

type CustomConn struct {
	net.Conn
	// contains filtered or unexported fields
}

CustomConn custom connections, intercept disconnected behavior

func (*CustomConn) Close

func (c *CustomConn) Close() error

Close closes the listener, any blocked except operations will be unblocked and return errors.

type CustomListener

type CustomListener struct {
	net.Listener
	// contains filtered or unexported fields
}

CustomListener custom listener for counting connections

func NewCustomListener

func NewCustomListener(listener net.Listener, opts ...ConnectionOption) *CustomListener

NewCustomListener creates a new custom listener.

func (*CustomListener) Accept

func (l *CustomListener) Accept() (net.Conn, error)

Accept waits for and returns the next connection to the listener.

func (*CustomListener) GetActiveConnections

func (l *CustomListener) GetActiveConnections() int

GetActiveConnections returns the number of active connections.

type Option

type Option func(*options)

Option set metrics

func WithCounterMetrics

func WithCounterMetrics(metrics ...*prometheus.CounterVec) Option

WithCounterMetrics add Counter type indicator

func WithGaugeMetrics

func WithGaugeMetrics(metrics ...*prometheus.GaugeVec) Option

WithGaugeMetrics add Gauge type indicator

func WithHistogramMetrics

func WithHistogramMetrics(metrics ...*prometheus.HistogramVec) Option

WithHistogramMetrics adding Histogram type indicators

func WithSummaryMetrics

func WithSummaryMetrics(metrics ...*prometheus.SummaryVec) Option

WithSummaryMetrics add Summary type indicator