metrics.go

package p2p

import (
	"context"
	"fmt"

	"go.opentelemetry.io/otel"
	"go.opentelemetry.io/otel/attribute"
	"go.opentelemetry.io/otel/metric"
)

var meter = otel.Meter("shrex/eds")

type status string

const (
	StatusBadRequest  status = "bad_request"
	StatusSendRespErr status = "send_resp_err"
	StatusSendReqErr  status = "send_req_err"
	StatusReadRespErr status = "read_resp_err"
	StatusInternalErr status = "internal_err"
	StatusNotFound    status = "not_found"
	StatusTimeout     status = "timeout"
	StatusSuccess     status = "success"
	StatusRateLimited status = "rate_limited"
)

type Metrics struct {
	totalRequestCounter metric.Int64Counter
}

// ObserveRequests increments the total number of requests sent with the given status as an
// attribute.
func (m *Metrics) ObserveRequests(ctx context.Context, count int64, status status) {
	if m == nil {
		return
	}
	if ctx.Err() != nil {
		ctx = context.Background()
	}
	m.totalRequestCounter.Add(ctx, count,
		metric.WithAttributes(
			attribute.String("status", string(status)),
		))
}

func InitClientMetrics(protocol string) (*Metrics, error) {
	totalRequestCounter, err := meter.Int64Counter(
		fmt.Sprintf("shrex_%s_client_total_requests", protocol),
		metric.WithDescription(fmt.Sprintf("Total count of sent shrex/%s requests", protocol)),
	)
	if err != nil {
		return nil, err
	}

	return &Metrics{
		totalRequestCounter: totalRequestCounter,
	}, nil
}

func InitServerMetrics(protocol string) (*Metrics, error) {
	totalRequestCounter, err := meter.Int64Counter(
		fmt.Sprintf("shrex_%s_server_total_responses", protocol),
		metric.WithDescription(fmt.Sprintf("Total count of sent shrex/%s responses", protocol)),
	)
	if err != nil {
		return nil, err
	}

	return &Metrics{
		totalRequestCounter: totalRequestCounter,
	}, nil
}