/// <summary>
/// Adds the needed services for Google Cloud Tracing. Used with <see cref="UseGoogleTrace"/>.
/// </summary>
/// <param name="services">The service collection. Cannot be null.</param>
/// <param name="projectId">Optional if running on Google App Engine or Google Compute Engine.
/// The Google Cloud Platform project ID. If unspecified and running on GAE or GCE the project ID will be
/// detected from the platform.</param>
/// <param name="config">Optional trace configuration, if unset the default will be used.</param>
/// <param name="client">Optional Trace client, if unset the default will be used.</param>
/// <param name="traceFallbackPredicate">Optional function to trace requests. If the trace header is not set
/// then this function will be called to determine if a given request should be traced. This will
/// not override trace headers. If the function returns true the request will be traced, if false
/// is returned the trace will not be traced and if null is returned it will not affect the
/// trace decision.</param>
public static void AddGoogleTrace(
this IServiceCollection services, string projectId = null,
TraceConfiguration config = null, TraceServiceClient client = null,
Func <HttpRequest, bool?> traceFallbackPredicate = null)
{
GaxPreconditions.CheckNotNull(services, nameof(services));
client = client ?? TraceServiceClient.Create();
config = config ?? TraceConfiguration.Create();
projectId = CommonUtils.GetAndCheckProjectId(projectId);
var consumer = ConsumerFactory <TraceProto> .GetConsumer(
new GrpcTraceConsumer(client), MessageSizer <TraceProto> .GetSize, config.BufferOptions);
var tracerFactory = new ManagedTracerFactory(projectId, consumer,
RateLimitingTraceOptionsFactory.Create(config), TraceIdFactory.Create());
services.AddScoped(CreateTraceHeaderContext);
services.AddSingleton <IManagedTracerFactory>(tracerFactory);
services.AddSingleton <IHttpContextAccessor, HttpContextAccessor>();
services.AddSingleton(CreateManagedTracer);
services.AddSingleton(CreateTraceHeaderPropagatingHandler);
services.AddSingleton(new ShouldTraceRequest(traceFallbackPredicate));
}
/// <summary>
/// Adds the needed services for Google Cloud Tracing. Used with <see cref="UseGoogleTrace"/>.
/// </summary>
/// <param name="services">The service collection. Cannot be null.</param>
/// /// <param name="setupAction">Action to set up options. Cannot be null.</param>
public static void AddGoogleTrace(
this IServiceCollection services, Action <TraceServiceOptions> setupAction)
{
GaxPreconditions.CheckNotNull(services, nameof(services));
GaxPreconditions.CheckNotNull(setupAction, nameof(setupAction));
var serviceOptions = new TraceServiceOptions();
setupAction(serviceOptions);
var client = serviceOptions.Client ?? TraceServiceClient.Create();
var options = serviceOptions.Options ?? TraceOptions.Create();
var traceFallbackPredicate = serviceOptions.TraceFallbackPredicate ?? TraceDecisionPredicate.Default;
var projectId = CommonUtils.GetAndCheckProjectId(serviceOptions.ProjectId);
var consumer = ConsumerFactory <TraceProto> .GetConsumer(
new GrpcTraceConsumer(client), MessageSizer <TraceProto> .GetSize,
options.BufferOptions, options.RetryOptions);
var tracerFactory = new ManagedTracerFactory(projectId, consumer,
RateLimitingTraceOptionsFactory.Create(options), TraceIdFactory.Create());
services.AddScoped(CreateTraceHeaderContext);
services.AddSingleton <IManagedTracerFactory>(tracerFactory);
services.AddSingleton <IHttpContextAccessor, HttpContextAccessor>();
services.AddSingleton(CreateManagedTracer);
services.AddSingleton(CreateTraceHeaderPropagatingHandler);
services.AddSingleton(traceFallbackPredicate);
}
/// <summary>
/// Adds the needed services for Google Cloud Tracing. Used with <see cref="UseGoogleTrace"/>.
/// </summary>
/// <param name="services">The service collection. Cannot be null.</param>
/// <param name="projectId">The Google Cloud Platform project ID. Cannot be null.</param>
/// <param name="config">Optional trace configuration, if unset the default will be used.</param>
/// <param name="clientTask">Optional task which produces the Trace client, if
/// unset the default will be used.</param>
public static void AddGoogleTrace(
this IServiceCollection services, string projectId,
TraceConfiguration config = null, Task <TraceServiceClient> clientTask = null)
{
GaxPreconditions.CheckNotNull(services, nameof(services));
GaxPreconditions.CheckNotNull(projectId, nameof(projectId));
clientTask = clientTask ?? TraceServiceClient.CreateAsync();
config = config ?? TraceConfiguration.Create();
IConsumer <TraceProto> consumer = ConsumerFactory <TraceProto> .GetConsumer(
new GrpcTraceConsumer(clientTask), MessageSizer <TraceProto> .GetSize, config.BufferOptions);
var tracerFactory = new ManagedTracerFactory(projectId, consumer,
RateLimitingTraceOptionsFactory.Create(config), TraceIdFactory.Create());
services.AddSingleton <IManagedTracerFactory>(tracerFactory);
services.AddSingleton <IHttpContextAccessor, HttpContextAccessor>();
services.AddScoped(CreateTraceHeaderContext);
services.AddScoped(CreateManagedTracer);
}
