mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-25 09:07:48 +00:00
17 lines
510 B
Go
17 lines
510 B
Go
|
package instrumenting
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/zitadel/zitadel/internal/notification/channels"
|
||
|
|
"github.com/zitadel/zitadel/internal/telemetry/tracing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func traceMessages(ctx context.Context, channel channels.NotificationChannel, spanName string) channels.NotificationChannel {
|
||
|
|
return channels.HandleMessageFunc(func(message channels.Message) (err error) {
|
||
|
|
_, span := tracing.NewNamedSpan(ctx, spanName)
|
||
|
|
defer func() { span.EndWithError(err) }()
|
||
|
|
return channel.HandleMessage(message)
|
||
|
|
})
|
||
|
|
}
|