| | | 1 | | // Copyright (c) ZeroC, Inc. |
| | | 2 | | |
| | | 3 | | using IceRpc.Telemetry; |
| | | 4 | | using System.Diagnostics; |
| | | 5 | | |
| | | 6 | | namespace IceRpc.Extensions.DependencyInjection; |
| | | 7 | | |
| | | 8 | | /// <summary>Provides an extension method for <see cref="IInvokerBuilder" /> to add the telemetry interceptor.</summary> |
| | | 9 | | public static class TelemetryInvokerBuilderExtensions |
| | | 10 | | { |
| | | 11 | | /// <summary>Adds the <see cref="TelemetryInterceptor" /> to the builder. This interceptor relies on the |
| | | 12 | | /// <see cref="ActivitySource" /> service managed by the service provider.</summary> |
| | | 13 | | /// <param name="builder">The builder being configured.</param> |
| | | 14 | | /// <returns>The builder being configured.</returns> |
| | | 15 | | public static IInvokerBuilder UseTelemetry(this IInvokerBuilder builder) => |
| | 0 | 16 | | builder.ServiceProvider.GetService(typeof(ActivitySource)) is ActivitySource activitySource ? |
| | 0 | 17 | | builder.Use(next => new TelemetryInterceptor(next, activitySource)) : |
| | 0 | 18 | | throw new InvalidOperationException( |
| | 0 | 19 | | $"Could not find service of type '{nameof(ActivitySource)}' in the service container."); |
| | | 20 | | } |