Factory method for building activity tags collection

EventStore · Jan 6, 2025 · 3feb7c7 · 3feb7c7
1 parent 3890596
commit 3feb7c7
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 15 deletions.
diff --git a/src/EventStore.Client.Streams/EventStoreClient.Append.cs b/src/EventStore.Client.Streams/EventStoreClient.Append.cs
@@ -113,13 +113,7 @@ ValueTask<IWriteResult> AppendToStreamInternal(
 			UserCredentials? userCredentials,
 			CancellationToken cancellationToken
 		) {
-			var tags = new ActivityTagsCollection()
-				.WithRequiredTag(TelemetryTags.EventStore.Stream, header.Options.StreamIdentifier.StreamName.ToStringUtf8())
-				.WithGrpcChannelServerTags(Settings, channelInfo)
-				.WithClientSettingsServerTags(Settings)
-				.WithOptionalTag(TelemetryTags.Database.User, userCredentials?.Username ?? Settings.DefaultCredentials?.Username);
-
-			return EventStoreClientDiagnostics.ActivitySource.TraceClientOperation(Operation, TracingConstants.Operations.Append, tags);
+			return EventStoreClientDiagnostics.ActivitySource.TraceClientOperation(Operation, TracingConstants.Operations.Append, AppendTags);
 
 			async ValueTask<IWriteResult> Operation() {
 				using var call = new StreamsClient(channelInfo.CallInvoker)
@@ -157,6 +151,12 @@ await call.RequestStream
 
 				return HandleWrongExpectedRevision(response, header, operationOptions);
 			}
+
+			ActivityTagsCollection AppendTags() => new ActivityTagsCollection()
+				.WithRequiredTag(TelemetryTags.EventStore.Stream, header.Options.StreamIdentifier.StreamName.ToStringUtf8())
+				.WithGrpcChannelServerTags(Settings, channelInfo)
+				.WithClientSettingsServerTags(Settings)
+				.WithOptionalTag(TelemetryTags.Database.User, userCredentials?.Username ?? Settings.DefaultCredentials?.Username);
 		}
 
 		IWriteResult HandleSuccessAppend(AppendResp response, AppendReq header) {
@@ -282,16 +282,10 @@ ValueTask<IWriteResult> AppendInternal(
 				IEnumerable<EventData> events,
 				CancellationToken cancellationToken
 			) {
-				var tags = new ActivityTagsCollection()
-					.WithRequiredTag(TelemetryTags.EventStore.Stream, options.StreamIdentifier.StreamName.ToStringUtf8())
-					.WithGrpcChannelServerTags(_settings, _channelInfo)
-					.WithClientSettingsServerTags(_settings)
-					.WithOptionalTag(TelemetryTags.Database.User, _settings.DefaultCredentials?.Username);
-
 				return EventStoreClientDiagnostics.ActivitySource.TraceClientOperation(
 					Operation,
 					TracingConstants.Operations.Append,
-					tags
+					AppendTags
 				);
 
 				async ValueTask<IWriteResult> Operation() {
@@ -310,6 +304,12 @@ async ValueTask<IWriteResult> Operation() {
 
 					return await complete.Task.ConfigureAwait(false);
 				}
+
+				ActivityTagsCollection AppendTags() => new ActivityTagsCollection()
+					.WithRequiredTag(TelemetryTags.EventStore.Stream, options.StreamIdentifier.StreamName.ToStringUtf8())
+					.WithGrpcChannelServerTags(_settings, _channelInfo)
+					.WithClientSettingsServerTags(_settings)
+					.WithOptionalTag(TelemetryTags.Database.User, _settings.DefaultCredentials?.Username);
 			}
 
 			async Task Duplex(ValueTask<ChannelInfo> channelInfoTask) {

diff --git a/src/EventStore.Client/Common/Diagnostics/ActivitySourceExtensions.cs b/src/EventStore.Client/Common/Diagnostics/ActivitySourceExtensions.cs
@@ -12,11 +12,13 @@ public static async ValueTask<T> TraceClientOperation<T>(
 		this ActivitySource source,
 		Func<ValueTask<T>> tracedOperation,
 		string operationName,
-		ActivityTagsCollection? tags = null
+		Func<ActivityTagsCollection?>? tagsFactory = null
 	) {
 		if (source.HasNoActiveListeners())
 			return await tracedOperation().ConfigureAwait(false);
 
+		var tags = tagsFactory?.Invoke();
+
 		using var activity = StartActivity(source, operationName, ActivityKind.Client, tags, Activity.Current?.Context);
 
 		try {