mirror of
https://github.com/zitadel/zitadel.git
synced 2025-08-11 23:47:33 +00:00
444f682e25
# Which Problems Are Solved The recently introduced notification queue have potential race conditions. # How the Problems Are Solved Current code is refactored to use the queue package, which is safe in regards of concurrency. # Additional Changes - the queue is included in startup - improved code quality of queue # Additional Context - closes https://github.com/zitadel/zitadel/issues/9278
32 lines
1.0 KiB
Go
32 lines
1.0 KiB
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.GetTriggeringEventType())),
|
|
}
|
|
addCountErr := metrics.AddCount(ctx, metricName, 1, labels)
|
|
logging.WithFields("name", metricName, "labels", labels).OnError(addCountErr).Error("incrementing counter metric failed")
|
|
}
|