zitadel/internal/notification/channels/instrumenting/metrics.go

package instrumenting

import (
	"context"

	"github.com/zitadel/logging"
	"go.opentelemetry.io/otel/attribute"

	"github.com/zitadel/zitadel/internal/notification/channels"
	"github.com/zitadel/zitadel/internal/telemetry/metrics"
)

func countMessages(ctx context.Context, channel channels.NotificationChannel, successMetricName, errorMetricName string) channels.NotificationChannel {
	return channels.HandleMessageFunc(func(message channels.Message) error {
		err := channel.HandleMessage(message)
		metricName := successMetricName
		if err != nil {
			metricName = errorMetricName
		}
		addCount(ctx, metricName, message)
		return err
	})
}

func addCount(ctx context.Context, metricName string, message channels.Message) {
	labels := map[string]attribute.Value{
		"triggering_event_type": attribute.StringValue(string(message.GetTriggeringEvent().Type())),
	}
	addCountErr := metrics.AddCount(ctx, metricName, 1, labels)
	logging.WithFields("name", metricName, "labels", labels).OnError(addCountErr).Error("incrementing counter metric failed")
}