();\n\n services.AddMvc();\n }\n\n public void Configure(IApplicationBuilder app)\n {\n app.UseDeveloperExceptionPage();\n\n app.UseRouting();\n\n app.UseAuthentication();\n app.UseAuthorization();\n\n app.UseEndpoints(endpoints =>\n {\n endpoints.MapDefaultControllerRoute();\n });\n }\n }\n}\n"
},
{
"path": "samples/netcoreapp3.1/FrontendWeb/Views/Home/Index.cshtml",
"content": "\n\nFrontendWeb\nFrontendWeb
\nThis is the beautiful web frontend.
\nRefresh this page a few times and check the Jaeger UI - you should already see traces!
\n\nPlace an order
\n"
},
{
"path": "samples/netcoreapp3.1/FrontendWeb/Views/Home/PlaceOrder.cshtml",
"content": "@model Shared.PlaceOrderCommand\n\n\nFrontendWeb\nFrontendWeb
\nPlace an order
\n\n\n"
},
{
"path": "samples/netcoreapp3.1/FrontendWeb/Views/_ViewImports.cshtml",
"content": "@using Samples.FrontendWeb\n@addTagHelper *, Microsoft.AspNetCore.Mvc.TagHelpers"
},
{
"path": "samples/netcoreapp3.1/FrontendWeb/appsettings.json",
"content": "{\n \"Logging\": {\n \"IncludeScopes\": false,\n \"LogLevel\": {\n \"Default\": \"Debug\",\n \"System\": \"Information\",\n \"Microsoft\": \"Information\",\n \"OpenTracing\": \"Debug\"\n }\n }\n}\n"
},
{
"path": "samples/netcoreapp3.1/OrdersApi/Controllers/OrdersController.cs",
"content": "using System;\nusing System.Collections.Generic;\nusing System.Linq;\nusing System.Net.Http;\nusing System.Threading.Tasks;\nusing Microsoft.AspNetCore.Mvc;\nusing Microsoft.EntityFrameworkCore;\nusing Newtonsoft.Json;\nusing OpenTracing;\nusing OrdersApi.DataStore;\nusing Shared;\n\nnamespace Samples.OrdersApi.Controllers\n{\n [Route(\"orders\")]\n public class OrdersController : Controller\n {\n private readonly OrdersDbContext _dbContext;\n private readonly HttpClient _httpClient;\n private readonly ITracer _tracer;\n\n public OrdersController(OrdersDbContext dbContext, HttpClient httpClient, ITracer tracer)\n {\n _dbContext = dbContext ?? throw new ArgumentNullException(nameof(dbContext));\n _httpClient = httpClient ?? throw new ArgumentNullException(nameof(httpClient));\n _tracer = tracer ?? throw new ArgumentNullException(nameof(tracer));\n }\n\n [HttpGet]\n public async Task Index()\n {\n var orders = await _dbContext.Orders.ToListAsync();\n\n return Ok(orders.Select(x => new { x.OrderId }).ToList());\n }\n\n [HttpPost]\n public async Task Index([FromBody] PlaceOrderCommand cmd)\n {\n var customer = await GetCustomer(cmd.CustomerId.Value);\n\n var order = new Order\n {\n CustomerId = cmd.CustomerId.Value,\n ItemNumber = cmd.ItemNumber,\n Quantity = cmd.Quantity\n };\n\n _dbContext.Orders.Add(order);\n\n await _dbContext.SaveChangesAsync();\n\n _tracer.ActiveSpan?.Log(new Dictionary {\n { \"event\", \"OrderPlaced\" },\n { \"orderId\", order.OrderId },\n { \"customer\", order.CustomerId },\n { \"customer_name\", customer.Name },\n { \"item_number\", order.ItemNumber },\n { \"quantity\", order.Quantity }\n });\n\n return Ok();\n }\n\n private async Task GetCustomer(int customerId)\n {\n var request = new HttpRequestMessage\n {\n Method = HttpMethod.Get,\n RequestUri = new Uri(Constants.CustomersUrl + \"customers/\" + customerId)\n };\n\n var response = await _httpClient.SendAsync(request);\n\n response.EnsureSuccessStatusCode();\n\n var body = await response.Content.ReadAsStringAsync();\n\n return JsonConvert.DeserializeObject(body);\n }\n }\n}\n"
},
{
"path": "samples/netcoreapp3.1/OrdersApi/DataStore/Order.cs",
"content": "using System;\nusing System.ComponentModel.DataAnnotations;\n\nnamespace OrdersApi.DataStore\n{\n public class Order\n {\n [Key]\n public int OrderId { get; set; }\n \n public int CustomerId { get; set; }\n\n [Required, StringLength(10)]\n public string ItemNumber { get; set; }\n\n [Required, Range(1, 100)]\n public int Quantity { get; set; }\n }\n}\n"
},
{
"path": "samples/netcoreapp3.1/OrdersApi/DataStore/OrdersDbContext.cs",
"content": "using Microsoft.EntityFrameworkCore;\n\nnamespace OrdersApi.DataStore\n{\n public class OrdersDbContext : DbContext\n {\n public OrdersDbContext(DbContextOptions options)\n : base(options)\n {\n }\n\n public DbSet Orders { get; set; }\n\n public void Seed()\n {\n if (Database.EnsureCreated())\n {\n Database.Migrate();\n\n SaveChanges();\n }\n }\n }\n}\n"
},
{
"path": "samples/netcoreapp3.1/OrdersApi/OrdersApi.csproj",
"content": "\n\n \n netcoreapp3.1\n \n\n \n \n \n \n\n \n \n \n \n\n\n"
},
{
"path": "samples/netcoreapp3.1/OrdersApi/Program.cs",
"content": "using Microsoft.AspNetCore.Hosting;\nusing Microsoft.Extensions.DependencyInjection;\nusing Microsoft.Extensions.Hosting;\nusing Shared;\n\nnamespace Samples.OrdersApi\n{\n public class Program\n {\n public static void Main(string[] args)\n {\n CreateHostBuilder(args).Build().Run();\n }\n\n public static IHostBuilder CreateHostBuilder(string[] args)\n {\n return Host.CreateDefaultBuilder(args)\n .ConfigureWebHostDefaults(webBuilder =>\n {\n webBuilder\n .UseStartup()\n .UseUrls(Constants.OrdersUrl);\n })\n .ConfigureServices(services =>\n {\n // Registers and starts Jaeger (see Shared.JaegerServiceCollectionExtensions)\n services.AddJaeger();\n\n // Enables OpenTracing instrumentation for ASP.NET Core, CoreFx, EF Core\n services.AddOpenTracing(builder =>\n {\n builder.ConfigureAspNetCore(options =>\n {\n // We don't need any tracing data for our health endpoint.\n options.Hosting.IgnorePatterns.Add(ctx => ctx.Request.Path == \"/health\");\n });\n });\n });\n }\n }\n}\n"
},
{
"path": "samples/netcoreapp3.1/OrdersApi/Properties/launchSettings.json",
"content": "{\n \"profiles\": {\n \"OrdersApi\": {\n \"commandName\": \"Project\",\n \"launchBrowser\": false,\n \"launchUrl\": \"http://localhost:5002\",\n \"environmentVariables\": {\n \"ASPNETCORE_ENVIRONMENT\": \"Development\"\n }\n }\n }\n}\n"
},
{
"path": "samples/netcoreapp3.1/OrdersApi/Startup.cs",
"content": "using System;\nusing System.Net.Http;\nusing Microsoft.AspNetCore.Builder;\nusing Microsoft.EntityFrameworkCore;\nusing Microsoft.Extensions.DependencyInjection;\nusing OrdersApi.DataStore;\n\nnamespace Samples.OrdersApi\n{\n public class Startup\n {\n public void ConfigureServices(IServiceCollection services)\n {\n // Adds a SqlServer DB to show EFCore traces.\n services\n .AddDbContext(options =>\n {\n options.UseSqlServer(\"Server=(localdb)\\\\mssqllocaldb;Database=Orders-netcoreapp31;Trusted_Connection=True;MultipleActiveResultSets=true\");\n });\n\n services.AddSingleton();\n\n services.AddMvc();\n\n services.AddHealthChecks()\n .AddDbContextCheck();\n }\n\n public void Configure(IApplicationBuilder app)\n {\n // Load some dummy data into the db.\n BootstrapDataStore(app.ApplicationServices);\n\n app.UseDeveloperExceptionPage();\n\n app.UseRouting();\n\n app.UseAuthentication();\n app.UseAuthorization();\n\n app.UseEndpoints(endpoints =>\n {\n endpoints.MapDefaultControllerRoute();\n endpoints.MapHealthChecks(\"/health\");\n });\n }\n\n private void BootstrapDataStore(IServiceProvider serviceProvider)\n {\n using (var scope = serviceProvider.CreateScope())\n {\n var dbContext = scope.ServiceProvider.GetRequiredService();\n dbContext.Seed();\n }\n }\n }\n}\n"
},
{
"path": "samples/netcoreapp3.1/OrdersApi/appsettings.json",
"content": "{\n \"Logging\": {\n \"IncludeScopes\": false,\n \"LogLevel\": {\n \"Default\": \"Debug\",\n \"System\": \"Information\",\n \"Microsoft\": \"Information\"\n }\n }\n}\n"
},
{
"path": "samples/netcoreapp3.1/Shared/Constants.cs",
"content": "namespace Shared\n{\n public class Constants\n {\n public const string FrontendUrl = \"http://localhost:5000/\";\n\n public const string CustomersUrl = \"http://localhost:5001/\";\n\n public const string OrdersUrl = \"http://localhost:5002/\";\n }\n}\n"
},
{
"path": "samples/netcoreapp3.1/Shared/Customer.cs",
"content": "namespace Shared\n{\n public class Customer\n {\n public int CustomerId { get; set; }\n public string Name { get; set; }\n\n public Customer()\n {\n }\n\n public Customer(int customerId, string name)\n {\n CustomerId = customerId;\n Name = name;\n }\n }\n}\n"
},
{
"path": "samples/netcoreapp3.1/Shared/JaegerServiceCollectionExtensions.cs",
"content": "using System;\nusing System.Reflection;\nusing Jaeger;\nusing Jaeger.Reporters;\nusing Jaeger.Samplers;\nusing Jaeger.Senders.Thrift;\nusing Microsoft.Extensions.Logging;\nusing OpenTracing;\nusing OpenTracing.Contrib.NetCore.Configuration;\nusing OpenTracing.Util;\n\nnamespace Microsoft.Extensions.DependencyInjection\n{\n public static class JaegerServiceCollectionExtensions\n {\n private static readonly Uri _jaegerUri = new Uri(\"http://localhost:14268/api/traces\");\n\n public static IServiceCollection AddJaeger(this IServiceCollection services)\n {\n if (services == null)\n throw new ArgumentNullException(nameof(services));\n\n services.AddSingleton(serviceProvider =>\n {\n string serviceName = Assembly.GetEntryAssembly().GetName().Name;\n\n ILoggerFactory loggerFactory = serviceProvider.GetRequiredService();\n\n ISampler sampler = new ConstSampler(sample: true);\n\n IReporter reporter = new RemoteReporter.Builder()\n .WithSender(new HttpSender.Builder(_jaegerUri.ToString()).Build())\n .Build();\n\n ITracer tracer = new Tracer.Builder(serviceName)\n .WithLoggerFactory(loggerFactory)\n .WithSampler(sampler)\n .WithReporter(reporter)\n .Build();\n\n GlobalTracer.Register(tracer);\n\n return tracer;\n });\n\n // Prevent endless loops when OpenTracing is tracking HTTP requests to Jaeger.\n services.Configure(options =>\n {\n options.IgnorePatterns.Add(request => _jaegerUri.IsBaseOf(request.RequestUri));\n });\n\n return services;\n }\n }\n}\n"
},
{
"path": "samples/netcoreapp3.1/Shared/PlaceOrderCommand.cs",
"content": "using System.ComponentModel.DataAnnotations;\n\nnamespace Shared\n{\n public class PlaceOrderCommand\n {\n [Required]\n public int? CustomerId { get; set; }\n\n [Required, StringLength(10)]\n public string ItemNumber { get; set; }\n\n [Required, Range(1, 100)]\n public int Quantity { get; set; }\n }\n}\n"
},
{
"path": "samples/netcoreapp3.1/Shared/Shared.csproj",
"content": "\n\n \n netcoreapp3.1\n \n\n \n \n \n\n \n \n \n \n \n \n \n\n\n"
},
{
"path": "samples/netcoreapp3.1/TrafficGenerator/Program.cs",
"content": "using Microsoft.Extensions.DependencyInjection;\nusing Microsoft.Extensions.Hosting;\n\nnamespace TrafficGenerator\n{\n class Program\n {\n public static void Main(string[] args)\n {\n CreateHostBuilder(args).Build().Run();\n }\n\n public static IHostBuilder CreateHostBuilder(string[] args) =>\n Host.CreateDefaultBuilder(args)\n .ConfigureServices(services =>\n {\n // Registers and starts Jaeger (see Shared.JaegerServiceCollectionExtensions)\n services.AddJaeger();\n\n services.AddOpenTracing();\n\n services.AddHostedService();\n });\n }\n}\n"
},
{
"path": "samples/netcoreapp3.1/TrafficGenerator/TrafficGenerator.csproj",
"content": "\n\n \n netcoreapp3.1\n \n\n \n \n \n \n\n\n"
},
{
"path": "samples/netcoreapp3.1/TrafficGenerator/Worker.cs",
"content": "using System;\nusing System.Net.Http;\nusing System.Threading;\nusing System.Threading.Tasks;\nusing Microsoft.Extensions.Hosting;\nusing Microsoft.Extensions.Logging;\nusing Shared;\n\nnamespace TrafficGenerator\n{\n public class Worker : BackgroundService\n {\n private readonly ILogger _logger;\n\n public Worker(ILogger logger)\n {\n _logger = logger ?? throw new ArgumentNullException(nameof(logger));\n }\n\n protected override async Task ExecuteAsync(CancellationToken stoppingToken)\n {\n try\n {\n HttpClient customersHttpClient = new HttpClient();\n customersHttpClient.BaseAddress = new Uri(Constants.CustomersUrl);\n\n HttpClient ordersHttpClient = new HttpClient();\n ordersHttpClient.BaseAddress = new Uri(Constants.OrdersUrl);\n\n\n while (!stoppingToken.IsCancellationRequested)\n {\n HttpResponseMessage ordershealthResponse = await ordersHttpClient.GetAsync(\"health\");\n _logger.LogInformation($\"Health of 'orders'-endpoint: '{ordershealthResponse.StatusCode}'\");\n\n HttpResponseMessage customersHealthResponse = await customersHttpClient.GetAsync(\"health\");\n _logger.LogInformation($\"Health of 'customers'-endpoint: '{customersHealthResponse.StatusCode}'\");\n\n _logger.LogInformation(\"Requesting customers\");\n\n HttpResponseMessage response = await customersHttpClient.GetAsync(\"customers\");\n\n _logger.LogInformation($\"Response was '{response.StatusCode}'\");\n\n await Task.Delay(TimeSpan.FromSeconds(5), stoppingToken);\n }\n }\n catch (TaskCanceledException)\n {\n /* Application should be stopped -> no-op */\n }\n catch (Exception ex)\n {\n _logger.LogError(ex, \"Unhandled exception\");\n }\n }\n }\n}\n"
},
{
"path": "samples/netcoreapp3.1/TrafficGenerator/appsettings.json",
"content": "{\n \"Logging\": {\n \"LogLevel\": {\n \"Default\": \"Debug\",\n \"System\": \"Information\",\n \"Microsoft\": \"Warning\",\n \"Microsoft.AspNetCore.Hosting\": \"Information\"\n }\n }\n}\n"
},
{
"path": "src/Directory.Build.props",
"content": "\n \n\n \n true\n Christian Weiss\n package-icon.png\n https://avatars0.githubusercontent.com/u/15482765\n https://github.com/opentracing-contrib/csharp-netcore\n Apache-2.0\n https://github.com/opentracing-contrib/csharp-netcore/releases/tag/v$(Version)\n\n $(NoWarn);CS1591\n true\n\n \n true\n true\n snupkg\n \n\n \n \n \n\n \n \n \n\n\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/AspNetCore/AspNetCoreDiagnosticOptions.cs",
"content": "using OpenTracing.Contrib.NetCore.AspNetCore;\n\nnamespace OpenTracing.Contrib.NetCore.Configuration\n{\n public class AspNetCoreDiagnosticOptions : DiagnosticOptions\n {\n public HostingOptions Hosting { get; } = new HostingOptions();\n\n public AspNetCoreDiagnosticOptions()\n {\n // We create separate spans for MVC actions & results so we don't need these additional events by default.\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.BeforeOnResourceExecuting\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.BeforeOnActionExecution\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.BeforeOnActionExecuting\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.AfterOnActionExecuting\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.BeforeActionMethod\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.BeforeControllerActionMethod\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.AfterControllerActionMethod\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.AfterActionMethod\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.BeforeOnActionExecuted\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.AfterOnActionExecuted\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.AfterOnActionExecution\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.BeforeOnActionExecuted\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.AfterOnActionExecuted\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.AfterOnActionExecution\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.BeforeOnResultExecuting\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.AfterOnResultExecuting\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.BeforeOnResultExecuted\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.AfterOnResultExecuted\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.BeforeOnResourceExecuted\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.AfterOnResourceExecuted\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.AfterOnResourceExecuting\");\n\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.Razor.BeginInstrumentationContext\");\n IgnoredEvents.Add(\"Microsoft.AspNetCore.Mvc.Razor.EndInstrumentationContext\");\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/AspNetCore/AspNetCoreDiagnostics.cs",
"content": "using System;\nusing System.Collections.Generic;\nusing Microsoft.AspNetCore.Http;\nusing Microsoft.AspNetCore.Http.Extensions;\nusing Microsoft.AspNetCore.Mvc;\nusing Microsoft.AspNetCore.Mvc.Abstractions;\nusing Microsoft.AspNetCore.Mvc.Controllers;\nusing Microsoft.Extensions.Logging;\nusing Microsoft.Extensions.Options;\nusing OpenTracing.Contrib.NetCore.Configuration;\nusing OpenTracing.Contrib.NetCore.Internal;\nusing OpenTracing.Propagation;\nusing OpenTracing.Tag;\n\nnamespace OpenTracing.Contrib.NetCore.AspNetCore\n{\n /// \n /// Instruments ASP.NET Core.\n /// \n /// Unfortunately, ASP.NET Core only uses one instance\n /// for everything so we also only create one observer to ensure best performance.\n /// Hosting events: https://github.com/aspnet/Hosting/blob/dev/src/Microsoft.AspNetCore.Hosting/Internal/HostingApplicationDiagnostics.cs\n /// \n internal sealed class AspNetCoreDiagnostics : DiagnosticEventObserver\n {\n public const string DiagnosticListenerName = \"Microsoft.AspNetCore\";\n\n private const string HostingScopeItemsKey = \"ot-HttpRequestIn\";\n private const string ActionScopeItemsKey = \"ot-MvcAction\";\n private const string ActionResultScopeItemsKey = \"ot-MvcActionResult\";\n\n private const string ActionComponent = \"AspNetCore.MvcAction\";\n private const string ActionTagActionName = \"action\";\n private const string ActionTagControllerName = \"controller\";\n\n private const string ResultComponent = \"AspNetCore.MvcResult\";\n private const string ResultTagType = \"result.type\";\n\n private static readonly PropertyFetcher _httpRequestIn_start_HttpContextFetcher = new PropertyFetcher(\"HttpContext\");\n private static readonly PropertyFetcher _httpRequestIn_stop_HttpContextFetcher = new PropertyFetcher(\"HttpContext\");\n private static readonly PropertyFetcher _unhandledException_HttpContextFetcher = new PropertyFetcher(\"httpContext\");\n private static readonly PropertyFetcher _unhandledException_ExceptionFetcher = new PropertyFetcher(\"exception\");\n private static readonly PropertyFetcher _beforeAction_httpContextFetcher = new PropertyFetcher(\"httpContext\");\n private static readonly PropertyFetcher _beforeAction_ActionDescriptorFetcher = new PropertyFetcher(\"actionDescriptor\");\n private static readonly PropertyFetcher _afterAction_httpContextFetcher = new PropertyFetcher(\"httpContext\");\n private static readonly PropertyFetcher _beforeActionResult_actionContextFetcher = new PropertyFetcher(\"actionContext\");\n private static readonly PropertyFetcher _beforeActionResult_ResultFetcher = new PropertyFetcher(\"result\");\n private static readonly PropertyFetcher _afterActionResult_actionContextFetcher = new PropertyFetcher(\"actionContext\");\n\n internal static readonly string NoHostSpecified = string.Empty;\n\n private readonly AspNetCoreDiagnosticOptions _options;\n\n public AspNetCoreDiagnostics(ILoggerFactory loggerFactory, ITracer tracer, IOptions options)\n : base(loggerFactory, tracer, options?.Value)\n {\n _options = options?.Value ?? throw new ArgumentNullException(nameof(options));\n }\n\n protected override string GetListenerName() => DiagnosticListenerName;\n\n protected override bool IsSupportedEvent(string eventName)\n {\n return eventName switch\n {\n // We don't want to get the old deprecated Hosting events.\n \"Microsoft.AspNetCore.Hosting.BeginRequest\" => false,\n \"Microsoft.AspNetCore.Hosting.EndRequest\" => false,\n _ => true,\n };\n }\n\n protected override IEnumerable HandledEventNames()\n {\n yield return \"Microsoft.AspNetCore.Hosting.HttpRequestIn.Start\";\n yield return \"Microsoft.AspNetCore.Hosting.UnhandledException\";\n yield return \"Microsoft.AspNetCore.Hosting.HttpRequestIn.Stop\";\n yield return \"Microsoft.AspNetCore.Mvc.BeforeAction\";\n yield return \"Microsoft.AspNetCore.Mvc.AfterAction\";\n yield return \"Microsoft.AspNetCore.Mvc.BeforeActionResult\";\n yield return \"Microsoft.AspNetCore.Mvc.AfterActionResult\";\n }\n\n protected override void HandleEvent(string eventName, object untypedArg)\n {\n switch (eventName)\n {\n case \"Microsoft.AspNetCore.Hosting.HttpRequestIn.Start\":\n {\n var httpContext = (HttpContext)_httpRequestIn_start_HttpContextFetcher.Fetch(untypedArg);\n\n if (Tracer.ActiveSpan == null && !_options.StartRootSpans)\n {\n if (IsLogLevelTraceEnabled)\n {\n Logger.LogTrace(\"Ignoring request due to missing parent span\");\n }\n return;\n }\n\n if (ShouldIgnore(httpContext))\n {\n if (IsLogLevelTraceEnabled)\n {\n Logger.LogTrace(\"Ignoring request\");\n }\n }\n else\n {\n var request = httpContext.Request;\n\n ISpanContext extractedSpanContext = null;\n\n if (_options.Hosting.ExtractEnabled?.Invoke(httpContext) ?? true)\n {\n extractedSpanContext = Tracer.Extract(BuiltinFormats.HttpHeaders, new RequestHeadersExtractAdapter(request.Headers));\n }\n\n string operationName = _options.Hosting.OperationNameResolver(httpContext);\n\n IScope scope = Tracer.BuildSpan(operationName)\n .AsChildOf(extractedSpanContext)\n .WithTag(Tags.Component, _options.Hosting.ComponentName)\n .WithTag(Tags.SpanKind, Tags.SpanKindServer)\n .WithTag(Tags.HttpMethod, request.Method)\n .WithTag(Tags.HttpUrl, GetDisplayUrl(request))\n .StartActive();\n\n _options.Hosting.OnRequest?.Invoke(scope.Span, httpContext);\n\n httpContext.Items[HostingScopeItemsKey] = scope;\n }\n }\n break;\n\n case \"Microsoft.AspNetCore.Hosting.UnhandledException\":\n {\n var httpContext = (HttpContext)_unhandledException_HttpContextFetcher.Fetch(untypedArg);\n var exception = (Exception)_unhandledException_ExceptionFetcher.Fetch(untypedArg);\n\n var scope = httpContext.Items[HostingScopeItemsKey] as IScope;\n if (scope != null)\n {\n var span = scope.Span;\n\n span.SetException(exception);\n\n _options.Hosting.OnError?.Invoke(span, exception, httpContext);\n }\n }\n break;\n\n case \"Microsoft.AspNetCore.Hosting.HttpRequestIn.Stop\":\n {\n var httpContext = (HttpContext)_httpRequestIn_stop_HttpContextFetcher.Fetch(untypedArg);\n\n var scope = httpContext.Items[HostingScopeItemsKey] as IScope;\n if (scope != null)\n {\n httpContext.Items.Remove(HostingScopeItemsKey);\n\n scope.Span.SetTag(Tags.HttpStatus, httpContext.Response.StatusCode);\n scope.Dispose();\n }\n\n httpContext.Items.Remove(ActionResultScopeItemsKey);\n httpContext.Items.Remove(ActionScopeItemsKey);\n }\n break;\n\n case \"Microsoft.AspNetCore.Mvc.BeforeAction\":\n {\n var httpContext = (HttpContext)_beforeAction_httpContextFetcher.Fetch(untypedArg);\n\n // We only create this span if the entire request should be traced.\n var scope = httpContext.Items[HostingScopeItemsKey] as IScope;\n if (scope != null)\n {\n // NOTE: This event is the start of the action pipeline. The action has been selected, the route\n // has been selected but no filters have run and model binding hasn't occured.\n\n var actionDescriptor = (ActionDescriptor)_beforeAction_ActionDescriptorFetcher.Fetch(untypedArg);\n var controllerActionDescriptor = actionDescriptor as ControllerActionDescriptor;\n\n string operationName = controllerActionDescriptor != null\n ? $\"Action {controllerActionDescriptor.ControllerTypeInfo.FullName}/{controllerActionDescriptor.ActionName}\"\n : $\"Action {actionDescriptor.DisplayName}\";\n\n var actionScope = Tracer.BuildSpan(operationName)\n .AsChildOf(scope.Span)\n .WithTag(Tags.Component, ActionComponent)\n .WithTag(ActionTagControllerName, controllerActionDescriptor?.ControllerTypeInfo.FullName)\n .WithTag(ActionTagActionName, controllerActionDescriptor?.ActionName)\n .StartActive();\n\n httpContext.Items[ActionScopeItemsKey] = actionScope;\n }\n }\n break;\n\n case \"Microsoft.AspNetCore.Mvc.AfterAction\":\n {\n var httpContext = (HttpContext)_afterAction_httpContextFetcher.Fetch(untypedArg);\n var scope = httpContext.Items[ActionScopeItemsKey] as IScope;\n if (scope != null)\n {\n httpContext.Items.Remove(ActionScopeItemsKey);\n scope.Dispose();\n } \n }\n break;\n\n case \"Microsoft.AspNetCore.Mvc.BeforeActionResult\":\n {\n var httpContext = ((ActionContext)_beforeActionResult_actionContextFetcher.Fetch(untypedArg)).HttpContext;\n\n // We only create this span if the entire request should be traced.\n var scope = httpContext.Items[HostingScopeItemsKey] as IScope;\n if (scope != null)\n {\n // NOTE: This event is the start of the result pipeline. The action has been executed, but\n // we haven't yet determined which view (if any) will handle the request\n\n object result = _beforeActionResult_ResultFetcher.Fetch(untypedArg);\n\n string resultType = result.GetType().Name;\n string operationName = $\"Result {resultType}\";\n\n var actionResultScope = Tracer.BuildSpan(operationName)\n .AsChildOf(scope.Span)\n .WithTag(Tags.Component, ResultComponent)\n .WithTag(ResultTagType, resultType)\n .StartActive();\n\n httpContext.Items[ActionResultScopeItemsKey] = actionResultScope;\n }\n }\n break;\n\n case \"Microsoft.AspNetCore.Mvc.AfterActionResult\":\n {\n var httpContext = ((ActionContext)_afterActionResult_actionContextFetcher.Fetch(untypedArg)).HttpContext;\n var scope = httpContext.Items[ActionResultScopeItemsKey] as IScope;\n if (scope != null)\n {\n httpContext.Items.Remove(ActionResultScopeItemsKey);\n scope.Dispose();\n }\n\n }\n break;\n\n default:\n HandleUnknownEvent(eventName, untypedArg);\n break;\n }\n }\n\n private static string GetDisplayUrl(HttpRequest request)\n {\n if (request.Host.HasValue)\n {\n return request.GetDisplayUrl();\n }\n\n // HTTP 1.0 requests are not required to provide a Host to be valid\n // Since this is just for display, we can provide a string that is\n // not an actual Uri with only the fields that are specified.\n // request.GetDisplayUrl(), used above, will throw an exception\n // if request.Host is null.\n return $\"{request.Scheme}://{NoHostSpecified}{request.PathBase.Value}{request.Path.Value}{request.QueryString.Value}\";\n }\n\n private bool ShouldIgnore(HttpContext httpContext)\n {\n foreach (Func ignore in _options.Hosting.IgnorePatterns)\n {\n if (ignore(httpContext))\n return true;\n }\n\n return false;\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/AspNetCore/HostingOptions.cs",
"content": "using System;\nusing System.Collections.Generic;\nusing Microsoft.AspNetCore.Http;\n\nnamespace OpenTracing.Contrib.NetCore.AspNetCore\n{\n public class HostingOptions\n {\n public const string DefaultComponent = \"HttpIn\";\n\n // Variables are lazily instantiated to prevent the app from crashing if the required assemblies are not referenced.\n\n private string _componentName = DefaultComponent;\n private List> _ignorePatterns;\n private Func _operationNameResolver;\n\n\n /// \n /// Allows changing the \"component\" tag of created spans.\n /// \n public string ComponentName\n {\n get => _componentName;\n set => _componentName = value ?? throw new ArgumentNullException(nameof(ComponentName));\n }\n\n /// \n /// A list of delegates that define whether or not a given request should be ignored.\n /// \n /// If any delegate in the list returns true, the request will be ignored.\n /// \n public List> IgnorePatterns\n {\n get\n {\n if (_ignorePatterns == null)\n {\n _ignorePatterns = new List>();\n }\n return _ignorePatterns;\n }\n }\n\n /// \n /// A delegates that defines from which requests tracing headers are extracted.\n /// \n public Func ExtractEnabled { get; set; }\n\n /// \n /// A delegate that returns the OpenTracing \"operation name\" for the given request.\n /// \n public Func OperationNameResolver\n {\n get\n {\n if (_operationNameResolver == null)\n {\n _operationNameResolver = (httpContext) => \"HTTP \" + httpContext.Request.Method;\n }\n return _operationNameResolver;\n }\n set => _operationNameResolver = value ?? throw new ArgumentNullException(nameof(OperationNameResolver));\n }\n\n /// \n /// Allows the modification of the created span to e.g. add further tags.\n /// \n public Action OnRequest { get; set; }\n\n /// \n /// Allows the modification of the created span when error occured to e.g. add further tags.\n /// \n public Action OnError { get; set; }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/AspNetCore/RequestHeadersExtractAdapter.cs",
"content": "using System;\nusing System.Collections;\nusing System.Collections.Generic;\nusing Microsoft.AspNetCore.Http;\nusing OpenTracing.Propagation;\n\nnamespace OpenTracing.Contrib.NetCore.AspNetCore\n{\n internal sealed class RequestHeadersExtractAdapter : ITextMap\n {\n private readonly IHeaderDictionary _headers;\n\n public RequestHeadersExtractAdapter(IHeaderDictionary headers)\n {\n _headers = headers ?? throw new ArgumentNullException(nameof(headers));\n }\n\n public void Set(string key, string value)\n {\n throw new NotSupportedException(\"This class should only be used with ITracer.Extract\");\n }\n\n public IEnumerator> GetEnumerator()\n {\n foreach (var kvp in _headers)\n {\n yield return new KeyValuePair(kvp.Key, kvp.Value);\n }\n }\n\n IEnumerator IEnumerable.GetEnumerator()\n {\n return GetEnumerator();\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/Configuration/DiagnosticOptions.cs",
"content": "using System.Collections.Generic;\n\nnamespace OpenTracing.Contrib.NetCore.Configuration\n{\n public abstract class DiagnosticOptions\n {\n /// \n /// Defines whether or not generic events from this DiagnostSource should be logged as events.\n /// \n public bool LogEvents { get; set; } = true;\n\n /// \n /// Defines specific event names that should NOT be logged as events. Set to `false` if you don't want any events to be logged.\n /// \n public HashSet IgnoredEvents { get; } = new HashSet();\n\n /// \n /// Defines whether or not a span should be created if there is no parent span.\n /// \n public bool StartRootSpans { get; set; } = true;\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/Configuration/IOpenTracingBuilder.cs",
"content": "namespace Microsoft.Extensions.DependencyInjection\n{\n /// \n /// An interface for configuring OpenTracing services.\n /// \n public interface IOpenTracingBuilder\n {\n /// \n /// Gets the where OpenTracing services are configured.\n /// \n IServiceCollection Services { get; }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/Configuration/OpenTracingBuilder.cs",
"content": "using System;\nusing Microsoft.Extensions.DependencyInjection;\n\nnamespace OpenTracing.Contrib.NetCore.Configuration\n{\n internal class OpenTracingBuilder : IOpenTracingBuilder\n {\n public IServiceCollection Services { get; }\n\n public OpenTracingBuilder(IServiceCollection services)\n {\n Services = services ?? throw new ArgumentNullException(nameof(services));\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/Configuration/OpenTracingBuilderExtensions.cs",
"content": "using System;\nusing Microsoft.Extensions.DependencyInjection.Extensions;\nusing Microsoft.Extensions.Logging;\nusing OpenTracing.Contrib.NetCore.AspNetCore;\nusing OpenTracing.Contrib.NetCore.Configuration;\nusing OpenTracing.Contrib.NetCore.EntityFrameworkCore;\nusing OpenTracing.Contrib.NetCore.GenericListeners;\nusing OpenTracing.Contrib.NetCore.HttpHandler;\nusing OpenTracing.Contrib.NetCore.Internal;\nusing OpenTracing.Contrib.NetCore.Logging;\nusing OpenTracing.Contrib.NetCore.MicrosoftSqlClient;\nusing OpenTracing.Contrib.NetCore.SystemSqlClient;\n\nnamespace Microsoft.Extensions.DependencyInjection\n{\n public static class OpenTracingBuilderExtensions\n {\n internal static IOpenTracingBuilder AddDiagnosticSubscriber(this IOpenTracingBuilder builder)\n where TDiagnosticSubscriber : DiagnosticObserver\n {\n if (builder == null)\n throw new ArgumentNullException(nameof(builder));\n\n builder.Services.TryAddEnumerable(ServiceDescriptor.Singleton());\n\n return builder;\n }\n\n /// \n /// Adds instrumentation for ASP.NET Core.\n /// \n public static IOpenTracingBuilder AddAspNetCore(this IOpenTracingBuilder builder, Action options = null)\n {\n if (builder == null)\n throw new ArgumentNullException(nameof(builder));\n\n builder.AddDiagnosticSubscriber();\n builder.ConfigureGenericDiagnostics(genericOptions => genericOptions.IgnoredListenerNames.Add(AspNetCoreDiagnostics.DiagnosticListenerName));\n\n // Our default behavior for ASP.NET is that we only want spans if the request itself is traced\n builder.ConfigureEntityFrameworkCore(opt => opt.StartRootSpans = false);\n builder.ConfigureHttpHandler(opt => opt.StartRootSpans = false);\n builder.ConfigureMicrosoftSqlClient(opt => opt.StartRootSpans = false);\n builder.ConfigureSystemSqlClient(opt => opt.StartRootSpans = false);\n\n return ConfigureAspNetCore(builder, options);\n }\n\n public static IOpenTracingBuilder ConfigureAspNetCore(this IOpenTracingBuilder builder, Action options)\n {\n if (builder == null)\n throw new ArgumentNullException(nameof(builder));\n\n if (options != null)\n {\n builder.Services.Configure(options);\n }\n\n return builder;\n }\n\n /// \n /// Adds instrumentation for System.Net.Http.\n /// \n public static IOpenTracingBuilder AddHttpHandler(this IOpenTracingBuilder builder, Action options = null)\n {\n if (builder == null)\n throw new ArgumentNullException(nameof(builder));\n\n builder.AddDiagnosticSubscriber();\n builder.ConfigureGenericDiagnostics(options => options.IgnoredListenerNames.Add(HttpHandlerDiagnostics.DiagnosticListenerName));\n\n return ConfigureHttpHandler(builder, options);\n }\n\n public static IOpenTracingBuilder ConfigureHttpHandler(this IOpenTracingBuilder builder, Action options)\n {\n if (builder == null)\n throw new ArgumentNullException(nameof(builder));\n\n if (options != null)\n {\n builder.Services.Configure(options);\n }\n\n return builder;\n }\n\n /// \n /// Adds instrumentation for Entity Framework Core.\n /// \n public static IOpenTracingBuilder AddEntityFrameworkCore(this IOpenTracingBuilder builder, Action options = null)\n {\n if (builder == null)\n throw new ArgumentNullException(nameof(builder));\n\n builder.AddDiagnosticSubscriber();\n builder.ConfigureGenericDiagnostics(genericOptions => genericOptions.IgnoredListenerNames.Add(EntityFrameworkCoreDiagnostics.DiagnosticListenerName));\n\n return ConfigureEntityFrameworkCore(builder, options);\n }\n\n /// \n /// Configuration options for the instrumentation of Entity Framework Core.\n /// \n public static IOpenTracingBuilder ConfigureEntityFrameworkCore(this IOpenTracingBuilder builder, Action options)\n {\n if (builder == null)\n throw new ArgumentNullException(nameof(builder));\n\n if (options != null)\n {\n builder.Services.Configure(options);\n }\n\n return builder;\n }\n\n\n /// \n /// Adds instrumentation for generic DiagnosticListeners.\n /// \n public static IOpenTracingBuilder AddGenericDiagnostics(this IOpenTracingBuilder builder, Action options = null)\n {\n builder.AddDiagnosticSubscriber();\n\n return ConfigureGenericDiagnostics(builder, options);\n }\n\n public static IOpenTracingBuilder ConfigureGenericDiagnostics(this IOpenTracingBuilder builder, Action options)\n {\n if (builder == null)\n throw new ArgumentNullException(nameof(builder));\n\n if (options != null)\n {\n builder.Services.Configure(options);\n }\n\n return builder;\n }\n\n /// \n /// Disables tracing for all diagnostic listeners that don't have an explicit implementation.\n /// \n public static IOpenTracingBuilder RemoveGenericDiagnostics(this IOpenTracingBuilder builder)\n {\n if (builder == null)\n throw new ArgumentNullException(nameof(builder));\n\n builder.Services.RemoveAll();\n\n return builder;\n }\n\n /// \n /// Adds instrumentation for Microsoft.Data.SqlClient.\n /// \n public static IOpenTracingBuilder AddMicrosoftSqlClient(this IOpenTracingBuilder builder, Action options = null)\n {\n if (builder == null)\n throw new ArgumentNullException(nameof(builder));\n\n builder.AddDiagnosticSubscriber();\n builder.ConfigureGenericDiagnostics(genericOptions => genericOptions.IgnoredListenerNames.Add(MicrosoftSqlClientDiagnostics.DiagnosticListenerName));\n\n return ConfigureMicrosoftSqlClient(builder, options);\n }\n\n /// \n /// Configuration options for the instrumentation of Microsoft.Data.SqlClient.\n /// \n public static IOpenTracingBuilder ConfigureMicrosoftSqlClient(this IOpenTracingBuilder builder, Action options)\n {\n if (builder == null)\n throw new ArgumentNullException(nameof(builder));\n\n if (options != null)\n {\n builder.Services.Configure(options);\n }\n\n return builder;\n }\n\n /// \n /// Adds instrumentation for System.Data.SqlClient.\n /// \n public static IOpenTracingBuilder AddSystemSqlClient(this IOpenTracingBuilder builder, Action options = null)\n {\n if (builder == null)\n throw new ArgumentNullException(nameof(builder));\n\n builder.AddDiagnosticSubscriber();\n builder.ConfigureGenericDiagnostics(options => options.IgnoredListenerNames.Add(SqlClientDiagnostics.DiagnosticListenerName));\n\n return ConfigureSystemSqlClient(builder, options);\n }\n\n public static IOpenTracingBuilder ConfigureSystemSqlClient(this IOpenTracingBuilder builder, Action options)\n {\n if (builder == null)\n throw new ArgumentNullException(nameof(builder));\n\n if (options != null)\n {\n builder.Services.Configure(options);\n }\n\n return builder;\n }\n\n public static IOpenTracingBuilder AddLoggerProvider(this IOpenTracingBuilder builder)\n {\n if (builder == null)\n throw new ArgumentNullException(nameof(builder));\n\n builder.Services.TryAddEnumerable(ServiceDescriptor.Singleton());\n builder.Services.Configure(options =>\n {\n // All interesting request-specific logs are instrumented via DiagnosticSource.\n options.AddFilter(\"Microsoft.AspNetCore.Hosting\", LogLevel.None);\n\n // The \"Information\"-level in ASP.NET Core is too verbose.\n options.AddFilter(\"Microsoft.AspNetCore\", LogLevel.Warning);\n\n // EF Core is sending everything to DiagnosticSource AND ILogger so we completely disable the category.\n options.AddFilter(\"Microsoft.EntityFrameworkCore\", LogLevel.None);\n });\n\n return builder;\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/Configuration/ServiceCollectionExtensions.cs",
"content": "using System;\nusing Microsoft.Extensions.DependencyInjection.Extensions;\nusing Microsoft.Extensions.Hosting;\nusing OpenTracing;\nusing OpenTracing.Contrib.NetCore;\nusing OpenTracing.Contrib.NetCore.Configuration;\nusing OpenTracing.Contrib.NetCore.Internal;\nusing OpenTracing.Util;\n\nnamespace Microsoft.Extensions.DependencyInjection\n{\n public static class ServiceCollectionExtensions\n {\n /// \n /// Adds OpenTracing instrumentation for ASP.NET Core, CoreFx (BCL), Entity Framework Core.\n /// \n public static IServiceCollection AddOpenTracing(this IServiceCollection services, Action builder = null)\n {\n if (services == null)\n throw new ArgumentNullException(nameof(services));\n\n return services.AddOpenTracingCoreServices(otBuilder =>\n {\n otBuilder.AddLoggerProvider();\n otBuilder.AddEntityFrameworkCore();\n otBuilder.AddGenericDiagnostics();\n otBuilder.AddHttpHandler();\n otBuilder.AddMicrosoftSqlClient();\n otBuilder.AddSystemSqlClient();\n\n if (AssemblyExists(\"Microsoft.AspNetCore.Hosting\"))\n {\n otBuilder.AddAspNetCore();\n }\n\n builder?.Invoke(otBuilder);\n });\n }\n\n /// \n /// Adds the core services required for OpenTracing without any actual instrumentations.\n /// \n public static IServiceCollection AddOpenTracingCoreServices(this IServiceCollection services, Action builder = null)\n {\n if (services == null)\n throw new ArgumentNullException(nameof(services));\n\n services.TryAddSingleton(GlobalTracer.Instance);\n services.TryAddSingleton();\n\n services.TryAddSingleton();\n services.TryAddEnumerable(ServiceDescriptor.Singleton());\n\n var builderInstance = new OpenTracingBuilder(services);\n\n builder?.Invoke(builderInstance);\n\n return services;\n }\n\n private static bool AssemblyExists(string assemblyName)\n {\n var assemblies = AppDomain.CurrentDomain.GetAssemblies();\n foreach (var assembly in assemblies)\n {\n if (assembly.FullName.StartsWith(assemblyName))\n return true;\n }\n return false;\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/EntityFrameworkCore/EntityFrameworkCoreDiagnosticOptions.cs",
"content": "using System;\nusing System.Collections.Generic;\nusing Microsoft.EntityFrameworkCore.Diagnostics;\n\nnamespace OpenTracing.Contrib.NetCore.Configuration\n{\n public class EntityFrameworkCoreDiagnosticOptions : DiagnosticOptions\n {\n // NOTE: Everything here that references any EFCore types MUST NOT be initialized in the constructor as that would throw on applications that don't reference EFCore.\n\n public const string DefaultComponent = \"EFCore\";\n\n private string _componentName = DefaultComponent;\n private List> _ignorePatterns;\n private Func _operationNameResolver;\n\n public EntityFrameworkCoreDiagnosticOptions()\n {\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.ChangeTracking.StartedTracking\");\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.ChangeTracking.DetectChangesStarting\");\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.ChangeTracking.DetectChangesCompleted\");\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.ChangeTracking.ForeignKeyChangeDetected\");\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.ChangeTracking.StateChanged\");\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.ChangeTracking.ValueGenerated\");\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.Database.Command.CommandCreating\");\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.Database.Command.CommandCreated\");\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.Database.Command.DataReaderDisposing\");\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.Database.Connection.ConnectionOpening\");\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.Database.Connection.ConnectionOpened\");\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.Database.Connection.ConnectionClosing\");\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.Database.Connection.ConnectionClosed\");\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.Database.Transaction.TransactionStarting\");\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.Database.Transaction.TransactionStarted\");\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.Database.Transaction.TransactionCommitting\");\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.Database.Transaction.TransactionCommitted\");\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.Database.Transaction.TransactionDisposed\");\n IgnoredEvents.Add(\"Microsoft.EntityFrameworkCore.Infrastructure.ContextDisposed\");\n }\n\n /// \n /// A list of delegates that define whether or not a given EF Core command should be ignored.\n /// \n /// If any delegate in the list returns true, the EF Core command will be ignored.\n /// \n public List> IgnorePatterns => _ignorePatterns ??= new List>();\n\n /// \n /// Allows changing the \"component\" tag of created spans.\n /// \n public string ComponentName\n {\n get => _componentName;\n set => _componentName = value ?? throw new ArgumentNullException(nameof(ComponentName));\n }\n\n /// \n /// A delegate that returns the OpenTracing \"operation name\" for the given command.\n /// \n public Func OperationNameResolver\n {\n get\n {\n if (_operationNameResolver == null)\n {\n // Default value may not be set in the constructor because this would fail\n // if the target application does not reference EFCore.\n _operationNameResolver = (data) => \"DB \" + data.ExecuteMethod.ToString();\n }\n return _operationNameResolver;\n }\n set => _operationNameResolver = value ?? throw new ArgumentNullException(nameof(OperationNameResolver));\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/EntityFrameworkCore/EntityFrameworkCoreDiagnostics.cs",
"content": "using System;\nusing System.Collections.Concurrent;\nusing System.Collections.Generic;\nusing Microsoft.EntityFrameworkCore.Diagnostics;\nusing Microsoft.Extensions.Logging;\nusing Microsoft.Extensions.Options;\nusing OpenTracing.Contrib.NetCore.Configuration;\nusing OpenTracing.Contrib.NetCore.Internal;\nusing OpenTracing.Tag;\n\nnamespace OpenTracing.Contrib.NetCore.EntityFrameworkCore\n{\n internal sealed class EntityFrameworkCoreDiagnostics : DiagnosticEventObserver\n {\n // https://github.com/aspnet/EntityFrameworkCore/blob/dev/src/EFCore/DbLoggerCategory.cs\n public const string DiagnosticListenerName = \"Microsoft.EntityFrameworkCore\";\n\n private const string TagMethod = \"db.method\";\n private const string TagIsAsync = \"db.async\";\n\n private readonly EntityFrameworkCoreDiagnosticOptions _options;\n private readonly ConcurrentDictionary _scopeStorage;\n\n public EntityFrameworkCoreDiagnostics(ILoggerFactory loggerFactory, ITracer tracer,\n IOptions options)\n : base(loggerFactory, tracer, options?.Value)\n {\n _options = options?.Value ?? throw new ArgumentNullException(nameof(options));\n _scopeStorage = new ConcurrentDictionary();\n }\n\n protected override string GetListenerName() => DiagnosticListenerName;\n\n protected override IEnumerable HandledEventNames()\n {\n yield return \"Microsoft.EntityFrameworkCore.Database.Command.CommandExecuting\";\n yield return \"Microsoft.EntityFrameworkCore.Database.Command.CommandExecuted\";\n yield return \"Microsoft.EntityFrameworkCore.Database.Command.CommandError\";\n }\n\n protected override void HandleEvent(string eventName, object untypedArg)\n {\n switch (eventName)\n {\n case \"Microsoft.EntityFrameworkCore.Database.Command.CommandExecuting\":\n {\n CommandEventData args = (CommandEventData)untypedArg;\n\n var activeSpan = Tracer.ActiveSpan;\n\n if (activeSpan == null && !_options.StartRootSpans)\n {\n if (IsLogLevelTraceEnabled)\n {\n Logger.LogTrace(\"Ignoring EF command due to missing parent span\");\n }\n return;\n }\n\n if (IgnoreEvent(args))\n {\n if (IsLogLevelTraceEnabled)\n {\n Logger.LogTrace(\"Ignoring EF command due to IgnorePatterns\");\n }\n return;\n }\n\n string operationName = _options.OperationNameResolver(args);\n\n var scope = Tracer.BuildSpan(operationName)\n .AsChildOf(activeSpan)\n .WithTag(Tags.SpanKind, Tags.SpanKindClient)\n .WithTag(Tags.Component, _options.ComponentName)\n .WithTag(Tags.DbInstance, args.Command.Connection.Database)\n .WithTag(Tags.DbStatement, args.Command.CommandText)\n .WithTag(TagMethod, args.ExecuteMethod.ToString())\n .WithTag(TagIsAsync, args.IsAsync)\n .StartActive();\n\n _scopeStorage.TryAdd(args.CommandId, scope);\n }\n break;\n\n case \"Microsoft.EntityFrameworkCore.Database.Command.CommandExecuted\":\n {\n CommandExecutedEventData args = (CommandExecutedEventData)untypedArg;\n\n if (_scopeStorage.TryRemove(args.CommandId, out var scope))\n {\n scope.Dispose();\n }\n }\n break;\n\n case \"Microsoft.EntityFrameworkCore.Database.Command.CommandError\":\n {\n CommandErrorEventData args = (CommandErrorEventData)untypedArg;\n\n // The \"CommandExecuted\" event is NOT called in case of an exception,\n // so we have to dispose the scope here as well!\n if (_scopeStorage.TryRemove(args.CommandId, out var scope))\n {\n scope.Span.SetException(args.Exception);\n scope.Dispose();\n }\n }\n break;\n\n default:\n {\n Dictionary tags = null;\n if (untypedArg is EventData eventArgs)\n {\n tags = new Dictionary\n {\n { \"level\", eventArgs.LogLevel.ToString() },\n };\n }\n\n HandleUnknownEvent(eventName, untypedArg, tags);\n break;\n }\n \n }\n }\n\n private bool IgnoreEvent(CommandEventData eventData)\n {\n foreach (Func ignore in _options.IgnorePatterns)\n {\n if (ignore(eventData))\n return true;\n }\n\n return false;\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/GenericListeners/GenericDiagnosticOptions.cs",
"content": "using System.Collections.Generic;\n\nnamespace OpenTracing.Contrib.NetCore.Configuration\n{\n public sealed class GenericDiagnosticOptions\n {\n public HashSet IgnoredListenerNames { get; } = new HashSet();\n\n public Dictionary> IgnoredEvents { get; } = new Dictionary>();\n\n public void IgnoreEvent(string diagnosticListenerName, string eventName)\n {\n if (diagnosticListenerName == null || eventName == null)\n return;\n\n HashSet ignoredListenerEvents;\n\n if (!IgnoredEvents.TryGetValue(diagnosticListenerName, out ignoredListenerEvents))\n {\n ignoredListenerEvents = new HashSet();\n IgnoredEvents.Add(diagnosticListenerName, ignoredListenerEvents);\n }\n\n ignoredListenerEvents.Add(eventName);\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/GenericListeners/GenericDiagnostics.cs",
"content": "using System;\nusing System.Collections.Generic;\nusing System.Diagnostics;\nusing Microsoft.Extensions.Logging;\nusing Microsoft.Extensions.Options;\nusing OpenTracing.Contrib.NetCore.Configuration;\nusing OpenTracing.Contrib.NetCore.Internal;\n\nnamespace OpenTracing.Contrib.NetCore.GenericListeners\n{\n /// \n /// A subscriber that logs ALL events to .\n /// \n internal sealed class GenericDiagnostics : DiagnosticObserver\n {\n private readonly GenericDiagnosticOptions _options;\n\n public GenericDiagnostics(ILoggerFactory loggerFactory, ITracer tracer, IOptions options)\n : base(loggerFactory, tracer)\n {\n _options = options?.Value ?? throw new ArgumentNullException(nameof(options));\n }\n\n public override IDisposable SubscribeIfMatch(DiagnosticListener diagnosticListener)\n {\n if (_options.IgnoredListenerNames.Contains(diagnosticListener.Name))\n {\n return null;\n }\n\n _options.IgnoredEvents.TryGetValue(diagnosticListener.Name, out var ignoredListenerEvents);\n\n return new GenericDiagnosticsSubscription(this, diagnosticListener, ignoredListenerEvents);\n }\n\n private class GenericDiagnosticsSubscription : IObserver>, IDisposable\n {\n private readonly GenericDiagnostics _subscriber;\n private readonly string _listenerName;\n private readonly HashSet _ignoredEvents;\n private readonly GenericEventProcessor _genericEventProcessor;\n\n private readonly IDisposable _subscription;\n\n\n public GenericDiagnosticsSubscription(GenericDiagnostics subscriber, DiagnosticListener diagnosticListener,\n HashSet ignoredEvents)\n {\n _subscriber = subscriber;\n _ignoredEvents = ignoredEvents;\n _listenerName = diagnosticListener.Name;\n\n _genericEventProcessor = new GenericEventProcessor(_listenerName, _subscriber.Tracer, subscriber.Logger);\n\n _subscription = diagnosticListener.Subscribe(this, IsEnabled);\n }\n\n public void Dispose()\n {\n _subscription.Dispose();\n }\n\n private bool IsEnabled(string eventName)\n {\n if (_ignoredEvents != null && _ignoredEvents.Contains(eventName))\n {\n if (_subscriber.IsLogLevelTraceEnabled)\n {\n _subscriber.Logger.LogTrace(\"Ignoring event '{ListenerName}/{Event}'\", _listenerName, eventName);\n }\n\n return false;\n }\n\n return true;\n }\n\n public void OnCompleted()\n {\n }\n\n public void OnError(Exception error)\n {\n }\n\n public void OnNext(KeyValuePair value)\n {\n string eventName = value.Key;\n object untypedArg = value.Value;\n\n try\n {\n // We have to check this twice because EVERY subscriber is called\n // if ANY subscriber returns IsEnabled=true.\n if (!IsEnabled(eventName))\n return;\n\n _genericEventProcessor?.ProcessEvent(eventName, untypedArg);\n }\n catch (Exception ex)\n {\n _subscriber.Logger.LogWarning(ex, \"Event-Exception: {ListenerName}/{Event}\", _listenerName, value.Key);\n }\n }\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/HttpHandler/HttpHandlerDiagnosticOptions.cs",
"content": "using System;\nusing System.Collections.Generic;\nusing System.Net.Http;\n\nnamespace OpenTracing.Contrib.NetCore.Configuration\n{\n public class HttpHandlerDiagnosticOptions : DiagnosticOptions\n {\n public const string PropertyIgnore = \"ot-ignore\";\n\n public const string DefaultComponent = \"HttpOut\";\n\n private string _componentName;\n private Func _operationNameResolver;\n\n /// \n /// Allows changing the \"component\" tag of created spans.\n /// \n public string ComponentName\n {\n get => _componentName;\n set => _componentName = value ?? throw new ArgumentNullException(nameof(ComponentName));\n }\n\n /// \n /// A list of delegates that define whether or not a given request should be ignored.\n /// \n /// If any delegate in the list returns true, the request will be ignored.\n /// \n public List> IgnorePatterns { get; } = new List>();\n\n /// \n /// A delegates that defines on what requests tracing headers are propagated.\n /// \n public Func InjectEnabled { get; set; }\n\n /// \n /// A delegate that returns the OpenTracing \"operation name\" for the given request.\n /// \n public Func OperationNameResolver\n {\n get => _operationNameResolver;\n set => _operationNameResolver = value ?? throw new ArgumentNullException(nameof(OperationNameResolver));\n }\n\n /// \n /// Allows the modification of the created span to e.g. add further tags.\n /// \n public Action OnRequest { get; set; }\n\n /// \n /// Allows the modification of the created span when error occured to e.g. add further tags.\n /// \n public Action OnError { get; set; }\n\n public HttpHandlerDiagnosticOptions()\n {\n // Default settings\n\n ComponentName = DefaultComponent;\n\n IgnorePatterns.Add((request) =>\n {\n IDictionary requestOptions;\n\n#if NETCOREAPP3_1\n requestOptions = request.Properties;\n#else \n requestOptions = request.Options;\n#endif\n\n return requestOptions.ContainsKey(PropertyIgnore);\n });\n\n OperationNameResolver = (request) =>\n {\n return \"HTTP \" + request.Method.Method;\n };\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/HttpHandler/HttpHandlerDiagnostics.cs",
"content": "using System;\nusing System.Collections.Generic;\nusing System.Net.Http;\nusing System.Threading.Tasks;\nusing Microsoft.Extensions.Logging;\nusing Microsoft.Extensions.Options;\nusing OpenTracing.Contrib.NetCore.Configuration;\nusing OpenTracing.Contrib.NetCore.Internal;\nusing OpenTracing.Propagation;\nusing OpenTracing.Tag;\n\nnamespace OpenTracing.Contrib.NetCore.HttpHandler\n{\n /// \n /// Instruments outgoing HTTP calls that use .\n /// See https://github.com/dotnet/corefx/blob/master/src/System.Net.Http/src/System/Net/Http/DiagnosticsHandler.cs\n /// and https://github.com/dotnet/corefx/blob/master/src/System.Net.Http/src/System/Net/Http/DiagnosticsHandlerLoggingStrings.cs\n /// \n internal sealed class HttpHandlerDiagnostics : DiagnosticEventObserver\n {\n public const string DiagnosticListenerName = \"HttpHandlerDiagnosticListener\";\n\n private const string PropertiesKey = \"ot-Span\";\n\n private static readonly PropertyFetcher _activityStart_RequestFetcher = new PropertyFetcher(\"Request\");\n private static readonly PropertyFetcher _activityStop_RequestFetcher = new PropertyFetcher(\"Request\");\n private static readonly PropertyFetcher _activityStop_ResponseFetcher = new PropertyFetcher(\"Response\");\n private static readonly PropertyFetcher _activityStop_RequestTaskStatusFetcher = new PropertyFetcher(\"RequestTaskStatus\");\n private static readonly PropertyFetcher _exception_RequestFetcher = new PropertyFetcher(\"Request\");\n private static readonly PropertyFetcher _exception_ExceptionFetcher = new PropertyFetcher(\"Exception\");\n\n private readonly HttpHandlerDiagnosticOptions _options;\n\n public HttpHandlerDiagnostics(ILoggerFactory loggerFactory, ITracer tracer,\n IOptions options)\n : base(loggerFactory, tracer, options?.Value)\n {\n _options = options?.Value ?? throw new ArgumentNullException(nameof(options));\n }\n\n protected override string GetListenerName() => DiagnosticListenerName;\n\n protected override bool IsSupportedEvent(string eventName)\n {\n return eventName switch\n {\n // We don't want to get the old deprecated events.\n // https://github.com/dotnet/runtime/blob/master/src/libraries/System.Net.Http/src/System/Net/Http/DiagnosticsHandlerLoggingStrings.cs\n \"System.Net.Http.Request\" => false,\n \"System.Net.Http.Response\" => false,\n _ => true,\n };\n }\n\n protected override IEnumerable HandledEventNames()\n {\n yield return \"System.Net.Http.HttpRequestOut.Start\";\n yield return \"System.Net.Http.Exception\";\n yield return \"System.Net.Http.HttpRequestOut.Stop\";\n }\n\n protected override void HandleEvent(string eventName, object untypedArg)\n {\n switch (eventName)\n {\n case \"System.Net.Http.HttpRequestOut.Start\":\n {\n var request = (HttpRequestMessage)_activityStart_RequestFetcher.Fetch(untypedArg);\n\n var activeSpan = Tracer.ActiveSpan;\n\n if (activeSpan == null && !_options.StartRootSpans)\n {\n if (IsLogLevelTraceEnabled)\n {\n Logger.LogTrace(\"Ignoring Request due to missing parent span\");\n }\n return;\n }\n\n if (IgnoreRequest(request))\n {\n if (IsLogLevelTraceEnabled)\n {\n Logger.LogTrace(\"Ignoring Request {RequestUri}\", request.RequestUri);\n }\n return;\n }\n\n string operationName = _options.OperationNameResolver(request);\n\n ISpan span = Tracer.BuildSpan(operationName)\n .AsChildOf(activeSpan)\n .WithTag(Tags.SpanKind, Tags.SpanKindClient)\n .WithTag(Tags.Component, _options.ComponentName)\n .WithTag(Tags.HttpMethod, request.Method.ToString())\n .WithTag(Tags.HttpUrl, request.RequestUri.ToString())\n .WithTag(Tags.PeerHostname, request.RequestUri.Host)\n .WithTag(Tags.PeerPort, request.RequestUri.Port)\n .Start();\n\n _options.OnRequest?.Invoke(span, request);\n\n if (_options.InjectEnabled?.Invoke(request) ?? true)\n {\n Tracer.Inject(span.Context, BuiltinFormats.HttpHeaders, new HttpHeadersInjectAdapter(request.Headers));\n }\n\n var requestOptions = GetRequestOptions(request);\n requestOptions[PropertiesKey] = span;\n }\n break;\n\n case \"System.Net.Http.Exception\":\n {\n var request = (HttpRequestMessage)_exception_RequestFetcher.Fetch(untypedArg);\n var requestOptions = GetRequestOptions(request);\n\n if (requestOptions.TryGetValue(PropertiesKey, out object objSpan) && objSpan is ISpan span)\n {\n var exception = (Exception)_exception_ExceptionFetcher.Fetch(untypedArg);\n\n span.SetException(exception);\n\n _options.OnError?.Invoke(span, exception, request);\n }\n }\n break;\n\n case \"System.Net.Http.HttpRequestOut.Stop\":\n {\n var request = (HttpRequestMessage)_activityStop_RequestFetcher.Fetch(untypedArg);\n var requestOptions = GetRequestOptions(request);\n\n if (requestOptions.TryGetValue(PropertiesKey, out object objSpan) && objSpan is ISpan span)\n {\n requestOptions.Remove(PropertiesKey);\n\n var response = (HttpResponseMessage)_activityStop_ResponseFetcher.Fetch(untypedArg);\n var requestTaskStatus = (TaskStatus)_activityStop_RequestTaskStatusFetcher.Fetch(untypedArg);\n\n if (response != null)\n {\n span.SetTag(Tags.HttpStatus, (int)response.StatusCode);\n }\n\n if (requestTaskStatus == TaskStatus.Canceled || requestTaskStatus == TaskStatus.Faulted)\n {\n span.SetTag(Tags.Error, true);\n }\n\n span.Finish();\n }\n }\n break;\n\n default:\n HandleUnknownEvent(eventName, untypedArg);\n break;\n }\n }\n\n private bool IgnoreRequest(HttpRequestMessage request)\n {\n foreach (Func ignore in _options.IgnorePatterns)\n {\n if (ignore(request))\n return true;\n }\n\n return false;\n }\n\n private IDictionary GetRequestOptions(HttpRequestMessage request)\n {\n IDictionary requestOptions;\n\n#if NETCOREAPP3_1\n requestOptions = request.Properties;\n#else \n requestOptions = request.Options;\n#endif\n\n return requestOptions;\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/HttpHandler/HttpHeadersInjectAdapter.cs",
"content": "using System;\nusing System.Collections;\nusing System.Collections.Generic;\nusing System.Net.Http.Headers;\nusing OpenTracing.Propagation;\n\nnamespace OpenTracing.Contrib.NetCore.HttpHandler\n{\n internal sealed class HttpHeadersInjectAdapter : ITextMap\n {\n private readonly HttpHeaders _headers;\n\n public HttpHeadersInjectAdapter(HttpHeaders headers)\n {\n _headers = headers ?? throw new ArgumentNullException(nameof(headers));\n }\n\n public void Set(string key, string value)\n {\n if (_headers.Contains(key))\n {\n _headers.Remove(key);\n }\n\n _headers.Add(key, value);\n }\n\n public IEnumerator> GetEnumerator()\n {\n throw new NotSupportedException(\"This class should only be used with ITracer.Inject\");\n }\n\n IEnumerator IEnumerable.GetEnumerator()\n {\n return GetEnumerator();\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/InstrumentationService.cs",
"content": "using System;\nusing System.Threading;\nusing System.Threading.Tasks;\nusing Microsoft.Extensions.Hosting;\nusing OpenTracing.Contrib.NetCore.Internal;\n\nnamespace OpenTracing.Contrib.NetCore\n{\n /// \n /// Starts and stops all OpenTracing instrumentation components.\n /// \n internal class InstrumentationService : IHostedService\n {\n private readonly DiagnosticManager _diagnosticsManager;\n\n public InstrumentationService(DiagnosticManager diagnosticManager)\n {\n _diagnosticsManager = diagnosticManager ?? throw new ArgumentNullException(nameof(diagnosticManager));\n }\n\n public Task StartAsync(CancellationToken cancellationToken)\n {\n _diagnosticsManager.Start();\n\n return Task.CompletedTask;\n }\n\n public Task StopAsync(CancellationToken cancellationToken)\n {\n _diagnosticsManager.Stop();\n\n return Task.CompletedTask;\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/Internal/DiagnosticEventObserver.cs",
"content": "using System;\nusing System.Collections.Generic;\nusing System.Diagnostics;\nusing Microsoft.Extensions.Logging;\nusing OpenTracing.Contrib.NetCore.Configuration;\n\nnamespace OpenTracing.Contrib.NetCore.Internal\n{\n /// \n /// Base class that allows handling events from a single .\n /// \n internal abstract class DiagnosticEventObserver\n : DiagnosticObserver, IObserver>\n {\n private readonly DiagnosticOptions _options;\n private readonly GenericEventProcessor _genericEventProcessor;\n\n protected DiagnosticEventObserver(ILoggerFactory loggerFactory, ITracer tracer, DiagnosticOptions options)\n : base(loggerFactory, tracer)\n {\n _options = options;\n\n if (options.LogEvents)\n {\n _genericEventProcessor = new GenericEventProcessor(GetListenerName(), Tracer, Logger);\n }\n }\n\n public override IDisposable SubscribeIfMatch(DiagnosticListener diagnosticListener)\n {\n if (diagnosticListener.Name == GetListenerName())\n {\n return diagnosticListener.Subscribe(this, IsEnabled);\n }\n\n return null;\n }\n\n void IObserver>.OnCompleted()\n {\n }\n\n void IObserver>.OnError(Exception error)\n {\n }\n\n void IObserver>.OnNext(KeyValuePair value)\n {\n try\n {\n if (IsEnabled(value.Key))\n {\n HandleEvent(value.Key, value.Value);\n }\n }\n catch (Exception ex)\n {\n Logger.LogWarning(ex, \"Event-Exception: {Event}\", value.Key);\n }\n }\n\n /// \n /// The name of the that should be instrumented.\n /// \n protected abstract string GetListenerName();\n\n protected virtual bool IsSupportedEvent(string eventName) => true;\n\n protected abstract IEnumerable HandledEventNames();\n\n private bool IsEnabled(string eventName)\n {\n if (!IsSupportedEvent(eventName))\n return false;\n\n foreach (var handledEventName in HandledEventNames())\n {\n if (handledEventName == eventName)\n return true;\n }\n\n if (!_options.LogEvents || _options.IgnoredEvents.Contains(eventName))\n return false;\n \n return true;\n }\n\n protected abstract void HandleEvent(string eventName, object untypedArg);\n\n protected void HandleUnknownEvent(string eventName, object untypedArg, IEnumerable> tags = null)\n {\n _genericEventProcessor?.ProcessEvent(eventName, untypedArg, tags);\n }\n\n protected void DisposeActiveScope(bool isScopeRequired, Exception exception = null)\n {\n IScope scope = Tracer.ScopeManager.Active;\n\n if (scope != null)\n {\n if (exception != null)\n {\n scope.Span.SetException(exception);\n }\n\n scope.Dispose();\n }\n else if (isScopeRequired)\n {\n Logger.LogWarning(\"Scope not found\");\n }\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/Internal/DiagnosticManager.cs",
"content": "using System;\nusing System.Collections.Generic;\nusing System.Diagnostics;\nusing System.Linq;\nusing Microsoft.Extensions.Logging;\nusing Microsoft.Extensions.Options;\n\nnamespace OpenTracing.Contrib.NetCore.Internal\n{\n /// \n /// Subscribes to and forwards events to individual instances.\n /// \n internal sealed class DiagnosticManager : IObserver, IDisposable\n {\n private readonly ILogger _logger;\n private readonly ITracer _tracer;\n private readonly IEnumerable _diagnosticSubscribers;\n private readonly DiagnosticManagerOptions _options;\n\n private readonly List _subscriptions = new List();\n private IDisposable _allListenersSubscription;\n\n public bool IsRunning => _allListenersSubscription != null;\n\n public DiagnosticManager(\n ILoggerFactory loggerFactory,\n ITracer tracer,\n IEnumerable diagnosticSubscribers,\n IOptions options)\n {\n if (loggerFactory == null)\n throw new ArgumentNullException(nameof(loggerFactory));\n\n if (diagnosticSubscribers == null)\n throw new ArgumentNullException(nameof(diagnosticSubscribers));\n\n _tracer = tracer ?? throw new ArgumentNullException(nameof(tracer));\n _options = options?.Value ?? throw new ArgumentNullException(nameof(options));\n\n _logger = loggerFactory.CreateLogger();\n\n _diagnosticSubscribers = diagnosticSubscribers;\n }\n\n public void Start()\n {\n if (_allListenersSubscription == null)\n {\n if (_tracer.IsNoopTracer() && !_options.StartInstrumentationForNoopTracer)\n {\n _logger.LogWarning(\"Instrumentation has not been started because no tracer was registered.\");\n }\n else\n {\n _logger.LogTrace(\"Starting AllListeners subscription\");\n _allListenersSubscription = DiagnosticListener.AllListeners.Subscribe(this);\n }\n }\n }\n\n void IObserver.OnCompleted()\n {\n }\n\n void IObserver.OnError(Exception error)\n {\n }\n\n void IObserver.OnNext(DiagnosticListener listener)\n {\n foreach (var subscriber in _diagnosticSubscribers)\n {\n IDisposable subscription = subscriber.SubscribeIfMatch(listener);\n if (subscription != null)\n {\n _logger.LogTrace($\"Subscriber '{subscriber.GetType().Name}' returned subscription for '{listener.Name}'\");\n _subscriptions.Add(subscription);\n }\n }\n }\n\n public void Stop()\n {\n if (_allListenersSubscription != null)\n {\n _logger.LogTrace(\"Stopping AllListeners subscription\");\n\n _allListenersSubscription.Dispose();\n _allListenersSubscription = null;\n\n foreach (var subscription in _subscriptions)\n {\n subscription.Dispose();\n }\n\n _subscriptions.Clear();\n }\n }\n\n public void Dispose()\n {\n Stop();\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/Internal/DiagnosticManagerOptions.cs",
"content": "namespace OpenTracing.Contrib.NetCore.Internal\n{\n public class DiagnosticManagerOptions\n {\n public bool StartInstrumentationForNoopTracer { get; set; }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/Internal/DiagnosticObserver.cs",
"content": "using System;\nusing System.Diagnostics;\nusing Microsoft.Extensions.Logging;\n\nnamespace OpenTracing.Contrib.NetCore.Internal\n{\n internal abstract class DiagnosticObserver\n {\n protected ILogger Logger { get; }\n\n protected ITracer Tracer { get; }\n\n protected bool IsLogLevelTraceEnabled { get; }\n\n protected DiagnosticObserver(ILoggerFactory loggerFactory, ITracer tracer)\n {\n if (loggerFactory == null)\n throw new ArgumentNullException(nameof(loggerFactory));\n\n if (tracer == null)\n throw new ArgumentNullException(nameof(tracer));\n\n Logger = loggerFactory.CreateLogger(GetType());\n Tracer = tracer;\n\n IsLogLevelTraceEnabled = Logger.IsEnabled(LogLevel.Trace);\n }\n\n public abstract IDisposable SubscribeIfMatch(DiagnosticListener diagnosticListener);\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/Internal/GenericEventProcessor.cs",
"content": "using System;\nusing System.Collections.Generic;\nusing System.Diagnostics;\nusing Microsoft.Extensions.Logging;\nusing OpenTracing.Tag;\n\nnamespace OpenTracing.Contrib.NetCore.Internal\n{\n internal class GenericEventProcessor\n {\n private readonly string _listenerName;\n private readonly ITracer _tracer;\n private readonly ILogger _logger;\n private readonly bool _isLogLevelTraceEnabled;\n\n public GenericEventProcessor(string listenerName, ITracer tracer, ILogger logger)\n {\n _listenerName = listenerName ?? throw new ArgumentNullException(nameof(listenerName));\n _tracer = tracer ?? throw new ArgumentNullException(nameof(tracer));\n _logger = logger ?? throw new ArgumentNullException(nameof(logger));\n\n _isLogLevelTraceEnabled = _logger.IsEnabled(LogLevel.Trace);\n }\n\n public void ProcessEvent(string eventName, object untypedArg, IEnumerable> tags = null)\n {\n Activity activity = Activity.Current;\n\n if (activity != null && eventName.EndsWith(\".Start\", StringComparison.Ordinal))\n {\n HandleActivityStart(eventName, activity, untypedArg, tags);\n }\n else if (activity != null && eventName.EndsWith(\".Stop\", StringComparison.Ordinal))\n {\n HandleActivityStop(eventName, activity);\n }\n else\n {\n HandleRegularEvent(eventName, untypedArg, tags);\n }\n }\n\n private void HandleActivityStart(string eventName, Activity activity, object untypedArg, IEnumerable> tags)\n {\n ISpanBuilder spanBuilder = _tracer.BuildSpan(activity.OperationName)\n .WithTag(Tags.Component, _listenerName);\n\n foreach (var tag in activity.Tags)\n {\n spanBuilder.WithTag(tag.Key, tag.Value);\n }\n\n if (tags != null)\n {\n foreach (var tag in tags)\n {\n spanBuilder.WithTag(tag.Key, tag.Value);\n }\n } \n\n spanBuilder.StartActive();\n }\n\n private void HandleActivityStop(string eventName, Activity activity)\n {\n IScope scope = _tracer.ScopeManager.Active;\n if (scope != null)\n {\n scope.Dispose();\n }\n else\n {\n _logger.LogWarning(\"No scope found. Event: {ListenerName}/{Event}\", _listenerName, eventName);\n }\n }\n\n private void HandleRegularEvent(string eventName, object untypedArg, IEnumerable> tags)\n {\n var span = _tracer.ActiveSpan;\n\n if (span != null)\n {\n span.Log(GetLogFields(eventName, untypedArg, tags));\n }\n else if (_isLogLevelTraceEnabled)\n {\n _logger.LogTrace(\"No ActiveSpan. Event: {ListenerName}/{Event}\", _listenerName, eventName);\n }\n }\n\n private Dictionary GetLogFields(string eventName, object arg, IEnumerable> tags)\n {\n var fields = new Dictionary\n {\n { LogFields.Event, eventName },\n { Tags.Component.Key, _listenerName }\n };\n\n if (tags != null)\n {\n foreach (var tag in tags)\n {\n fields[tag.Key] = tag.Value;\n }\n }\n\n // TODO improve the hell out of this... :)\n\n if (arg != null)\n {\n Type argType = arg.GetType();\n\n if (argType.IsPrimitive)\n {\n fields.Add(\"arg\", arg);\n }\n else if (argType.Namespace == null)\n {\n // Anonymous types usually contain complex objects so their output is not really useful.\n // Ignoring them for now.\n }\n else\n {\n fields.Add(\"arg\", arg.ToString());\n\n if (_isLogLevelTraceEnabled)\n {\n _logger.LogTrace(\"Can not extract value for argument type '{Type}'. Using ToString()\", argType);\n }\n }\n }\n\n return fields;\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/Internal/GlobalTracerAccessor.cs",
"content": "using OpenTracing.Util;\n\nnamespace OpenTracing.Contrib.NetCore.Internal\n{\n public class GlobalTracerAccessor : IGlobalTracerAccessor\n {\n public ITracer GetGlobalTracer()\n {\n return GlobalTracer.Instance;\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/Internal/IGlobalTracerAccessor.cs",
"content": "namespace OpenTracing.Contrib.NetCore.Internal\n{\n /// \n /// Helper interface which allows unit tests to mock the .\n /// \n public interface IGlobalTracerAccessor\n {\n ITracer GetGlobalTracer();\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/Internal/PropertyFetcher.cs",
"content": "// From https://github.com/dotnet/corefx/blob/master/src/System.Diagnostics.DiagnosticSource/src/System/Diagnostics/DiagnosticSourceEventSource.cs\n\nusing System;\nusing System.Linq;\nusing System.Reflection;\n\nnamespace OpenTracing.Contrib.NetCore.Internal\n{\n internal class PropertyFetcher\n {\n private readonly string _propertyName;\n private Type _expectedType;\n private PropertyFetch _fetchForExpectedType;\n\n /// \n /// Make a new PropertyFetcher for a property named 'propertyName'.\n /// \n public PropertyFetcher(string propertyName)\n {\n _propertyName = propertyName;\n }\n\n /// \n /// Given an object fetch the property that this PropertySpec represents.\n /// \n public object Fetch(object obj)\n {\n Type objType = obj.GetType();\n if (objType != _expectedType)\n {\n TypeInfo typeInfo = objType.GetTypeInfo();\n var propertyInfo = typeInfo.DeclaredProperties.FirstOrDefault(p => string.Equals(p.Name, _propertyName, StringComparison.InvariantCultureIgnoreCase));\n _fetchForExpectedType = PropertyFetch.FetcherForProperty(propertyInfo);\n _expectedType = objType;\n }\n return _fetchForExpectedType.Fetch(obj);\n }\n\n\n /// \n /// PropertyFetch is a helper class. It takes a PropertyInfo and then knows how\n /// to efficiently fetch that property from a .NET object (See Fetch method).\n /// It hides some slightly complex generic code. \n /// \n private class PropertyFetch\n {\n /// \n /// Create a property fetcher from a .NET Reflection PropertyInfo class that\n /// represents a property of a particular type.\n /// \n public static PropertyFetch FetcherForProperty(PropertyInfo propertyInfo)\n {\n if (propertyInfo == null)\n return new PropertyFetch(); // returns null on any fetch.\n\n Type typedPropertyFetcher = typeof(TypedFetchProperty<,>);\n Type instantiatedTypedPropertyFetcher = typedPropertyFetcher.GetTypeInfo().MakeGenericType(\n propertyInfo.DeclaringType, propertyInfo.PropertyType);\n\n return (PropertyFetch)Activator.CreateInstance(instantiatedTypedPropertyFetcher, propertyInfo);\n }\n\n /// \n /// Given an object, fetch the property that this propertyFech represents.\n /// \n public virtual object Fetch(object obj)\n {\n return null;\n }\n\n private class TypedFetchProperty : PropertyFetch\n {\n public TypedFetchProperty(PropertyInfo property)\n {\n _propertyFetch = (Func)property.GetMethod.CreateDelegate(typeof(Func));\n }\n public override object Fetch(object obj)\n {\n return _propertyFetch((TObject)obj);\n }\n private readonly Func _propertyFetch;\n }\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/Internal/SpanExtensions.cs",
"content": "using System;\nusing System.Collections.Generic;\nusing OpenTracing.Tag;\n\nnamespace OpenTracing.Contrib.NetCore.Internal\n{\n internal static class SpanExtensions\n {\n /// \n /// Sets the tag and adds information about the \n /// to the given .\n /// \n public static void SetException(this ISpan span, Exception exception)\n {\n if (span == null || exception == null)\n return;\n\n span.SetTag(Tags.Error, true);\n\n span.Log(new Dictionary(3)\n {\n { LogFields.Event, Tags.Error.Key },\n { LogFields.ErrorKind, exception.GetType().Name },\n { LogFields.ErrorObject, exception }\n });\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/Internal/TracerExtensions.cs",
"content": "using OpenTracing.Noop;\nusing OpenTracing.Util;\n\nnamespace OpenTracing.Contrib.NetCore.Internal\n{\n internal static class TracerExtensions\n {\n public static bool IsNoopTracer(this ITracer tracer)\n {\n if (tracer is NoopTracer)\n return true;\n\n // There's no way to check the underlying tracer on the instance so we have to check the static method.\n if (tracer is GlobalTracer && !GlobalTracer.IsRegistered())\n return true;\n\n return false;\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/Logging/OpenTracingLogger.cs",
"content": "using System;\nusing System.Collections.Generic;\nusing Microsoft.Extensions.Logging;\nusing OpenTracing.Contrib.NetCore.Internal;\nusing OpenTracing.Noop;\nusing OpenTracing.Util;\n\nnamespace OpenTracing.Contrib.NetCore.Logging\n{\n internal class OpenTracingLogger : ILogger\n {\n private const string OriginalFormatPropertyName = \"{OriginalFormat}\";\n\n private readonly string _categoryName;\n private readonly IGlobalTracerAccessor _globalTracerAccessor;\n\n public OpenTracingLogger(IGlobalTracerAccessor globalTracerAccessor, string categoryName)\n {\n _globalTracerAccessor = globalTracerAccessor;\n _categoryName = categoryName;\n }\n\n public IDisposable BeginScope(TState state)\n {\n return NoopDisposable.Instance;\n }\n\n public bool IsEnabled(LogLevel logLevel)\n {\n // Filtering should be done via the general Logging filtering feature.\n ITracer tracer = _globalTracerAccessor.GetGlobalTracer();\n return !(\n (tracer is NoopTracer) ||\n (tracer is GlobalTracer && !GlobalTracer.IsRegistered()));\n }\n\n public void Log(LogLevel logLevel, EventId eventId, TState state, Exception exception, Func formatter)\n {\n if (formatter == null)\n {\n // This throws an Exception e.g. in Microsoft's DebugLogger but we don't want the app to crash if the logger has an issue.\n return;\n }\n\n ITracer tracer = _globalTracerAccessor.GetGlobalTracer();\n ISpan span = tracer.ActiveSpan;\n\n if (span == null)\n {\n // Creating a new span for a log message seems brutal so we ignore messages if we can't attach it to an active span.\n return;\n }\n\n if (!IsEnabled(logLevel))\n {\n return;\n }\n\n var fields = new Dictionary\n {\n { \"component\", _categoryName },\n { \"level\", logLevel.ToString() }\n };\n\n try\n {\n if (eventId.Id != 0)\n {\n fields[\"eventId\"] = eventId.Id;\n }\n\n try\n {\n // This throws if the argument count (message format vs. actual args) doesn't match.\n // e.g. LogInformation(\"Foo {Arg1} {Arg2}\", arg1);\n // We want to preserve as much as possible from the original log message so we just continue without this information.\n string message = formatter(state, exception);\n fields[LogFields.Message] = message;\n }\n catch (Exception)\n {\n /* no-op */\n }\n\n if (exception != null)\n {\n fields[LogFields.ErrorKind] = exception.GetType().FullName;\n fields[LogFields.ErrorObject] = exception;\n }\n\n bool eventAdded = false;\n\n var structure = state as IEnumerable>;\n if (structure != null)\n {\n try\n {\n // The enumerator throws if the argument count (message format vs. actual args) doesn't match.\n // We want to preserve as much as possible from the original log message so we just ignore\n // this error and take as many properties as possible.\n foreach (var property in structure)\n {\n if (string.Equals(property.Key, OriginalFormatPropertyName, StringComparison.Ordinal)\n && property.Value is string messageTemplateString)\n {\n fields[LogFields.Event] = messageTemplateString;\n eventAdded = true;\n }\n else\n {\n fields[property.Key] = property.Value;\n }\n }\n }\n catch (IndexOutOfRangeException)\n {\n /* no-op */\n }\n }\n\n if (!eventAdded)\n {\n fields[LogFields.Event] = \"log\";\n }\n }\n catch (Exception logException)\n {\n fields[\"opentracing.contrib.netcore.error\"] = logException.ToString();\n }\n\n span.Log(fields);\n }\n\n private class NoopDisposable : IDisposable\n {\n public static NoopDisposable Instance = new NoopDisposable();\n\n public void Dispose()\n {\n }\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/Logging/OpenTracingLoggerProvider.cs",
"content": "using System;\nusing Microsoft.Extensions.Logging;\nusing OpenTracing.Contrib.NetCore.Internal;\n\nnamespace OpenTracing.Contrib.NetCore.Logging\n{\n /// \n /// The provider for the .\n /// \n [ProviderAlias(\"OpenTracing\")]\n internal class OpenTracingLoggerProvider : ILoggerProvider\n {\n private readonly IGlobalTracerAccessor _globalTracerAccessor;\n\n public OpenTracingLoggerProvider(IGlobalTracerAccessor globalTracerAccessor)\n {\n _globalTracerAccessor = globalTracerAccessor ?? throw new ArgumentNullException(nameof(globalTracerAccessor));\n }\n\n /// \n public ILogger CreateLogger(string categoryName)\n {\n return new OpenTracingLogger(_globalTracerAccessor, categoryName);\n }\n\n public void Dispose()\n {\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/MicrosoftSqlClient/MicrosoftSqlClientDiagnosticOptions.cs",
"content": "using System;\nusing System.Collections.Generic;\nusing System.Linq;\nusing Microsoft.Data.SqlClient;\n\nnamespace OpenTracing.Contrib.NetCore.Configuration\n{\n public class MicrosoftSqlClientDiagnosticOptions : DiagnosticOptions\n {\n public static class EventNames\n {\n // https://github.com/dotnet/SqlClient/blob/master/src/Microsoft.Data.SqlClient/netcore/src/Microsoft/Data/SqlClient/SqlClientDiagnosticListenerExtensions.cs\n\n private const string SqlClientPrefix = \"Microsoft.Data.SqlClient.\";\n\n public const string WriteCommandBefore = SqlClientPrefix + nameof(WriteCommandBefore);\n public const string WriteCommandAfter = SqlClientPrefix + nameof(WriteCommandAfter);\n public const string WriteCommandError = SqlClientPrefix + nameof(WriteCommandError);\n\n public const string WriteConnectionOpenBefore = SqlClientPrefix + nameof(WriteConnectionOpenBefore);\n public const string WriteConnectionOpenAfter = SqlClientPrefix + nameof(WriteConnectionOpenAfter);\n public const string WriteConnectionOpenError = SqlClientPrefix + nameof(WriteConnectionOpenError);\n\n public const string WriteConnectionCloseBefore = SqlClientPrefix + nameof(WriteConnectionCloseBefore);\n public const string WriteConnectionCloseAfter = SqlClientPrefix + nameof(WriteConnectionCloseAfter);\n public const string WriteConnectionCloseError = SqlClientPrefix + nameof(WriteConnectionCloseError);\n\n public const string WriteTransactionCommitBefore = SqlClientPrefix + nameof(WriteTransactionCommitBefore);\n public const string WriteTransactionCommitAfter = SqlClientPrefix + nameof(WriteTransactionCommitAfter);\n public const string WriteTransactionCommitError = SqlClientPrefix + nameof(WriteTransactionCommitError);\n\n public const string WriteTransactionRollbackBefore = SqlClientPrefix + nameof(WriteTransactionRollbackBefore);\n public const string WriteTransactionRollbackAfter = SqlClientPrefix + nameof(WriteTransactionRollbackAfter);\n public const string WriteTransactionRollbackError = SqlClientPrefix + nameof(WriteTransactionRollbackError);\n }\n\n public const string DefaultComponent = \"SqlClient\";\n public const string SqlClientPrefix = \"sqlClient \";\n\n private string _componentName = DefaultComponent;\n private List> _ignorePatterns;\n private Func _operationNameResolver;\n\n /// \n /// A list of delegates that define whether or not a given SQL command should be ignored.\n /// \n /// If any delegate in the list returns true, the SQL command will be ignored.\n /// \n public List> IgnorePatterns => _ignorePatterns ??= new List>();\n\n /// \n /// Allows changing the \"component\" tag of created spans.\n /// \n public string ComponentName\n {\n get => _componentName;\n set => _componentName = value ?? throw new ArgumentNullException(nameof(ComponentName));\n }\n\n /// \n /// A delegate that returns the OpenTracing \"operation name\" for the given command.\n /// \n public Func OperationNameResolver\n {\n get\n {\n if (_operationNameResolver == null)\n {\n // Default value may not be set in the constructor because this would fail\n // if the target application does not reference SqlClient.\n _operationNameResolver = (cmd) =>\n {\n var commandType = cmd.CommandText?.Split(' ');\n return $\"{SqlClientPrefix}{commandType?.FirstOrDefault()}\";\n };\n }\n return _operationNameResolver;\n }\n set => _operationNameResolver = value ?? throw new ArgumentNullException(nameof(OperationNameResolver));\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/MicrosoftSqlClient/MicrosoftSqlClientDiagnostics.cs",
"content": "using System;\nusing System.Collections.Concurrent;\nusing System.Collections.Generic;\nusing Microsoft.Data.SqlClient;\nusing Microsoft.Extensions.Logging;\nusing Microsoft.Extensions.Options;\nusing OpenTracing.Contrib.NetCore.Configuration;\nusing OpenTracing.Contrib.NetCore.Internal;\nusing OpenTracing.Tag;\n\nnamespace OpenTracing.Contrib.NetCore.MicrosoftSqlClient\n{\n internal sealed class MicrosoftSqlClientDiagnostics : DiagnosticEventObserver\n {\n public const string DiagnosticListenerName = \"SqlClientDiagnosticListener\";\n\n private static readonly PropertyFetcher _activityCommand_RequestFetcher = new PropertyFetcher(\"Command\");\n private static readonly PropertyFetcher _exception_ExceptionFetcher = new PropertyFetcher(\"Exception\");\n\n private readonly MicrosoftSqlClientDiagnosticOptions _options;\n private readonly ConcurrentDictionary _spanStorage;\n\n public MicrosoftSqlClientDiagnostics(ILoggerFactory loggerFactory, ITracer tracer, IOptions options)\n : base(loggerFactory, tracer, options?.Value)\n {\n _options = options?.Value ?? throw new ArgumentNullException(nameof(options));\n _spanStorage = new ConcurrentDictionary();\n }\n\n protected override string GetListenerName() => DiagnosticListenerName;\n\n /// \n /// Both diagnostic listeners for System.Data.SqlClient and Microsoft.Data.SqlClient use the same listener name, \n /// so we need to make sure this observer gets the correct events.\n /// \n protected override bool IsSupportedEvent(string eventName) => eventName.StartsWith(\"Microsoft.\");\n\n protected override IEnumerable HandledEventNames()\n {\n yield return MicrosoftSqlClientDiagnosticOptions.EventNames.WriteCommandBefore;\n yield return MicrosoftSqlClientDiagnosticOptions.EventNames.WriteCommandError;\n yield return MicrosoftSqlClientDiagnosticOptions.EventNames.WriteCommandAfter;\n }\n\n protected override void HandleEvent(string eventName, object untypedArg)\n {\n switch (eventName)\n {\n case MicrosoftSqlClientDiagnosticOptions.EventNames.WriteCommandBefore:\n {\n var cmd = (SqlCommand)_activityCommand_RequestFetcher.Fetch(untypedArg);\n\n var activeSpan = Tracer.ActiveSpan;\n\n if (activeSpan == null && !_options.StartRootSpans)\n {\n if (IsLogLevelTraceEnabled)\n {\n Logger.LogTrace(\"Ignoring SQL command due to missing parent span\");\n }\n return;\n }\n\n if (IgnoreEvent(cmd))\n {\n if (IsLogLevelTraceEnabled)\n {\n Logger.LogTrace(\"Ignoring SQL command due to IgnorePatterns\");\n } \n return;\n }\n\n string operationName = _options.OperationNameResolver(cmd);\n\n var span = Tracer.BuildSpan(operationName)\n .AsChildOf(activeSpan)\n .WithTag(Tags.SpanKind, Tags.SpanKindClient)\n .WithTag(Tags.Component, _options.ComponentName)\n .WithTag(Tags.DbInstance, cmd.Connection.Database)\n .WithTag(Tags.DbStatement, cmd.CommandText)\n .Start();\n\n _spanStorage.TryAdd(cmd, span);\n }\n break;\n\n case MicrosoftSqlClientDiagnosticOptions.EventNames.WriteCommandError:\n {\n var cmd = (SqlCommand)_activityCommand_RequestFetcher.Fetch(untypedArg);\n var ex = (Exception)_exception_ExceptionFetcher.Fetch(untypedArg);\n\n if (_spanStorage.TryRemove(cmd, out var span))\n {\n span.SetException(ex);\n span.Finish();\n }\n }\n break;\n\n case MicrosoftSqlClientDiagnosticOptions.EventNames.WriteCommandAfter:\n {\n var cmd = (SqlCommand)_activityCommand_RequestFetcher.Fetch(untypedArg);\n\n if (_spanStorage.TryRemove(cmd, out var span))\n {\n span.Finish();\n }\n }\n break;\n\n default:\n HandleUnknownEvent(eventName, untypedArg);\n break;\n }\n }\n\n private bool IgnoreEvent(SqlCommand sqlCommand)\n {\n foreach (Func ignore in _options.IgnorePatterns)\n {\n if (ignore(sqlCommand))\n return true;\n }\n\n return false;\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/OpenTracing.Contrib.NetCore.csproj",
"content": "\n\n \n netcoreapp3.1;net6.0;net7.0\n Adds OpenTracing instrumentation for .NET Core apps that use the `Microsoft.Extensions.*` stack.\nInstrumented components: HttpClient calls, ASP.NET Core, Entity Framework Core and any other library that uses DiagnosticSource events.\n opentracing;distributed-tracing;tracing;netcore\n \n\n \n \n \n\n \n \n \n\n \n \n \n \n \n \n \n \n \n \n \n \n \n\n \n \n \n\n \n \n \n \n \n \n \n \n \n \n \n \n \n\n \n \n \n\n \n \n \n \n \n \n \n \n \n \n \n \n \n\n\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/Properties/AssemblyInfo.cs",
"content": "using System.Runtime.CompilerServices;\n\n[assembly: InternalsVisibleTo(\"OpenTracing.Contrib.NetCore.Tests, PublicKey=\" +\n\"00240000048000009400000006020000002400005253413100040000010001005d933a901e4687\" +\n\"a5fad68c57b11f12b925922968cb9b7c4ef5744b4e3c7d7535095fb13a87169a4dde0ef547a035\" +\n\"0089ae94e192dfeddfa8272b12a76cd42bd36d8b7003bfff49bc825c1a9d510b3c37549efac581\" +\n\"0194d6cfdac8c4dc2f465e4c893a32b65a2f2d55faa3d24d6e68d861a7ccdec74d80ccf03b4489\" +\n\"f51328ba\")]\n\n[assembly: InternalsVisibleTo(\"OpenTracing.Contrib.NetCore.Benchmarks, PublicKey=\" +\n\"00240000048000009400000006020000002400005253413100040000010001005d933a901e4687\" +\n\"a5fad68c57b11f12b925922968cb9b7c4ef5744b4e3c7d7535095fb13a87169a4dde0ef547a035\" +\n\"0089ae94e192dfeddfa8272b12a76cd42bd36d8b7003bfff49bc825c1a9d510b3c37549efac581\" +\n\"0194d6cfdac8c4dc2f465e4c893a32b65a2f2d55faa3d24d6e68d861a7ccdec74d80ccf03b4489\" +\n\"f51328ba\")]\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/SystemSqlClient/SqlClientDiagnosticOptions.cs",
"content": "using System;\nusing System.Collections.Generic;\nusing System.Data.SqlClient;\nusing System.Linq;\n\nnamespace OpenTracing.Contrib.NetCore.Configuration\n{\n public class SqlClientDiagnosticOptions : DiagnosticOptions\n {\n public const string DefaultComponent = \"SqlClient\";\n public const string SqlClientPrefix = \"sqlClient \";\n\n private string _componentName = DefaultComponent;\n private List> _ignorePatterns;\n private Func _operationNameResolver;\n\n /// \n /// A list of delegates that define whether or not a given SQL command should be ignored.\n /// \n /// If any delegate in the list returns true, the SQL command will be ignored.\n /// \n public List> IgnorePatterns => _ignorePatterns ??= new List>();\n\n /// \n /// Allows changing the \"component\" tag of created spans.\n /// \n public string ComponentName\n {\n get => _componentName;\n set => _componentName = value ?? throw new ArgumentNullException(nameof(ComponentName));\n }\n\n /// \n /// A delegate that returns the OpenTracing \"operation name\" for the given command.\n /// \n public Func OperationNameResolver\n {\n get\n {\n if (_operationNameResolver == null)\n {\n // Default value may not be set in the constructor because this would fail\n // if the target application does not reference SqlClient.\n _operationNameResolver = (cmd) =>\n {\n var commandType = cmd.CommandText?.Split(' ');\n return $\"{SqlClientPrefix}{commandType?.FirstOrDefault()}\";\n };\n }\n return _operationNameResolver;\n }\n set => _operationNameResolver = value ?? throw new ArgumentNullException(nameof(OperationNameResolver));\n }\n }\n}\n"
},
{
"path": "src/OpenTracing.Contrib.NetCore/SystemSqlClient/SqlClientDiagnostics.cs",
"content": "using System;\nusing System.Collections.Concurrent;\nusing System.Collections.Generic;\nusing System.Data.SqlClient;\nusing Microsoft.Extensions.Logging;\nusing Microsoft.Extensions.Options;\nusing OpenTracing.Contrib.NetCore.Configuration;\nusing OpenTracing.Contrib.NetCore.Internal;\nusing OpenTracing.Tag;\n\nnamespace OpenTracing.Contrib.NetCore.SystemSqlClient\n{\n internal sealed class SqlClientDiagnostics : DiagnosticEventObserver\n {\n public const string DiagnosticListenerName = \"SqlClientDiagnosticListener\";\n\n private static readonly PropertyFetcher _writeCommandBefore_CommandFetcher = new PropertyFetcher(\"Command\");\n private static readonly PropertyFetcher _writeCommandError_CommandFetcher = new PropertyFetcher(\"Command\");\n private static readonly PropertyFetcher _writeCommandAfter_CommandFetcher = new PropertyFetcher(\"Command\");\n private static readonly PropertyFetcher _exception_ExceptionFetcher = new PropertyFetcher(\"Exception\");\n\n private readonly SqlClientDiagnosticOptions _options;\n private readonly ConcurrentDictionary _spanStorage;\n\n public SqlClientDiagnostics(ILoggerFactory loggerFactory, ITracer tracer, IOptions options)\n : base(loggerFactory, tracer, options?.Value)\n {\n _options = options?.Value ?? throw new ArgumentNullException(nameof(options));\n _spanStorage = new ConcurrentDictionary();\n }\n\n protected override string GetListenerName() => DiagnosticListenerName;\n\n /// \n /// Both diagnostic listeners for System.Data.SqlClient and Microsoft.Data.SqlClient use the same listener name, \n /// so we need to make sure this observer gets the correct events.\n /// \n protected override bool IsSupportedEvent(string eventName) => eventName.StartsWith(\"System.\");\n\n protected override IEnumerable HandledEventNames()\n {\n yield return \"System.Data.SqlClient.WriteCommandBefore\";\n yield return \"System.Data.SqlClient.WriteCommandError\";\n yield return \"System.Data.SqlClient.WriteCommandAfter\";\n }\n\n protected override void HandleEvent(string eventName, object untypedArg)\n {\n switch (eventName)\n {\n case \"System.Data.SqlClient.WriteCommandBefore\":\n {\n var cmd = (SqlCommand)_writeCommandBefore_CommandFetcher.Fetch(untypedArg);\n\n var activeSpan = Tracer.ActiveSpan;\n\n if (activeSpan == null && !_options.StartRootSpans)\n {\n if (IsLogLevelTraceEnabled)\n {\n Logger.LogTrace(\"Ignoring SQL command due to missing parent span\");\n }\n return;\n }\n\n if (IgnoreEvent(cmd))\n {\n if (IsLogLevelTraceEnabled)\n {\n Logger.LogTrace(\"Ignoring SQL command due to IgnorePatterns\");\n }\n return;\n }\n\n string operationName = _options.OperationNameResolver(cmd);\n\n var span = Tracer.BuildSpan(operationName)\n .AsChildOf(activeSpan)\n .WithTag(Tags.SpanKind, Tags.SpanKindClient)\n .WithTag(Tags.Component, _options.ComponentName)\n .WithTag(Tags.DbInstance, cmd.Connection.Database)\n .WithTag(Tags.DbStatement, cmd.CommandText)\n .Start();\n\n _spanStorage.TryAdd(cmd, span);\n }\n break;\n\n case \"System.Data.SqlClient.WriteCommandError\":\n {\n var cmd = (SqlCommand)_writeCommandError_CommandFetcher.Fetch(untypedArg);\n var ex = (Exception)_exception_ExceptionFetcher.Fetch(untypedArg);\n\n if (_spanStorage.TryRemove(cmd, out var span))\n {\n span.SetException(ex);\n span.Finish();\n }\n }\n break;\n\n case \"System.Data.SqlClient.WriteCommandAfter\":\n {\n var cmd = (SqlCommand)_writeCommandAfter_CommandFetcher.Fetch(untypedArg);\n\n if (_spanStorage.TryRemove(cmd, out var span))\n {\n span.Finish();\n }\n }\n break;\n\n default:\n HandleUnknownEvent(eventName, untypedArg);\n break;\n }\n }\n\n private bool IgnoreEvent(SqlCommand sqlCommand)\n {\n foreach (Func ignore in _options.IgnorePatterns)\n {\n if (ignore(sqlCommand))\n return true;\n }\n\n return false;\n }\n }\n}\n"
},
{
"path": "test/OpenTracing.Contrib.NetCore.Tests/AspNetCore/HostingTest.cs",
"content": "using System;\nusing System.Linq;\nusing System.Net.Http;\nusing System.Threading.Tasks;\nusing Microsoft.AspNetCore;\nusing Microsoft.AspNetCore.Builder;\nusing Microsoft.AspNetCore.Hosting;\nusing Microsoft.AspNetCore.Http;\nusing Microsoft.AspNetCore.Mvc.Testing;\nusing Microsoft.Extensions.DependencyInjection;\nusing Microsoft.Extensions.Logging;\nusing Microsoft.Extensions.Options;\nusing OpenTracing.Contrib.NetCore.AspNetCore;\nusing OpenTracing.Contrib.NetCore.Configuration;\nusing OpenTracing.Mock;\nusing OpenTracing.Tag;\nusing Xunit;\nusing Xunit.Abstractions;\n\nnamespace OpenTracing.Contrib.NetCore.Tests.AspNetCore\n{\n public class TestProgramFactory : WebApplicationFactory\n {\n protected override IWebHostBuilder CreateWebHostBuilder()\n {\n var host = WebHost.CreateDefaultBuilder()\n // https://stackoverflow.com/a/69776251/5214796\n .UseSetting(\"TEST_CONTENTROOT_OPENTRACING_CONTRIB_NETCORE_TESTS\", \"\")\n .ConfigureServices(services =>\n {\n })\n .Configure(app =>\n {\n app.UseRouting();\n app.UseEndpoints(endpoints =>\n {\n endpoints.MapGet(\"/foo\", async context =>\n {\n await context.Response.WriteAsync(\"Hello\");\n });\n\n endpoints.MapGet(\"/exception\", _ =>\n {\n throw new InvalidOperationException(\"You shall not pass\");\n });\n });\n });\n\n return host;\n }\n }\n \n [Collection(\"DiagnosticSource\") /* All DiagnosticSource tests must be in the same collection to ensure they are NOT run in parallel. */]\n public class HostingTest : IClassFixture, IDisposable\n {\n private readonly WebApplicationFactory _factory;\n private readonly MockTracer _tracer;\n private readonly HostingOptions _options;\n\n public HostingTest(TestProgramFactory factory, ITestOutputHelper output)\n {\n _tracer = new MockTracer();\n\n AspNetCoreDiagnosticOptions aspNetCoreOptions = new();\n _options = aspNetCoreOptions.Hosting;\n\n _factory = factory\n .WithWebHostBuilder(x =>\n {\n x.ConfigureServices(services =>\n {\n services.AddLogging(logging =>\n {\n logging.AddXunit(output);\n logging.AddFilter(\"OpenTracing\", LogLevel.Trace);\n });\n services.AddOpenTracingCoreServices(builder =>\n {\n builder.AddAspNetCore();\n\n builder.Services.AddSingleton(_tracer);\n builder.Services.AddSingleton(Options.Create(aspNetCoreOptions));\n });\n });\n });\n }\n\n public void Dispose()\n {\n _factory.Dispose();\n }\n \n private HttpClient CreateClient()\n {\n var client = _factory.CreateClient();\n return client;\n }\n\n [Fact]\n public async Task Request_creates_span()\n {\n var client = CreateClient();\n\n await client.GetAsync(\"/foo\");\n \n var finishedSpans = _tracer.FinishedSpans();\n\n Assert.Single(finishedSpans);\n }\n\n [Fact]\n public async Task Span_has_correct_properties()\n {\n var client = CreateClient();\n\n await client.GetAsync(\"/foo\");\n\n var finishedSpans = _tracer.FinishedSpans();\n Assert.Single(finishedSpans);\n\n var span = finishedSpans[0];\n Assert.Empty(span.GeneratedErrors);\n \n Assert.Single(span.LogEntries);\n Assert.Equal(\"Microsoft.AspNetCore.Routing.EndpointMatched\", span.LogEntries[0].Fields[\"event\"]);\n \n Assert.Equal(\"HTTP GET\", span.OperationName);\n Assert.Null(span.ParentId);\n Assert.Empty(span.References);\n\n Assert.Equal(5, span.Tags.Count);\n Assert.Equal(Tags.SpanKindServer, span.Tags[Tags.SpanKind.Key]);\n Assert.Equal(\"HttpIn\", span.Tags[Tags.Component.Key]);\n Assert.Equal(\"GET\", span.Tags[Tags.HttpMethod.Key]);\n Assert.Equal(\"http://localhost/foo\", span.Tags[Tags.HttpUrl.Key]);\n Assert.Equal(200, span.Tags[Tags.HttpStatus.Key]);\n }\n\n [Fact]\n public async Task Span_has_status_404()\n {\n var client = CreateClient();\n\n await client.GetAsync(\"/not-found\");\n\n var finishedSpans = _tracer.FinishedSpans();\n Assert.Single(finishedSpans);\n\n var span = finishedSpans[0];\n Assert.Equal(404, span.Tags[Tags.HttpStatus.Key]);\n }\n\n [Fact]\n public async Task Extracts_trace_headers()\n {\n var client = CreateClient();\n\n await client.SendAsync(new HttpRequestMessage(HttpMethod.Get, \"/foo\")\n {\n Headers =\n {\n { \"traceid\", \"100\" },\n { \"spanid\", \"101\" },\n }\n });\n\n var finishedSpans = _tracer.FinishedSpans();\n Assert.Single(finishedSpans);\n\n var span = finishedSpans[0];\n Assert.Equal(\"100\", span.Context.TraceId);\n Assert.Single(span.References);\n\n var reference = span.References[0];\n Assert.Equal(References.ChildOf, reference.ReferenceType);\n Assert.Equal(\"100\", reference.Context.TraceId);\n Assert.Equal(\"101\", reference.Context.SpanId);\n }\n\n [Fact]\n public async Task Does_not_Extract_trace_headers_if_disabled_in_options()\n {\n var client = CreateClient();\n\n await client.SendAsync(new HttpRequestMessage(HttpMethod.Get, \"/foo\")\n {\n Headers =\n {\n { \"ignore\", \"1\" },\n }\n });\n\n _options.ExtractEnabled = context => !context.Request.Headers.ContainsKey(\"ignore\");\n\n var finishedSpans = _tracer.FinishedSpans();\n Assert.Single(finishedSpans);\n\n var span = finishedSpans[0];\n Assert.Empty(span.References);\n }\n\n [Fact]\n public async Task Ignores_requests_with_custom_rule()\n {\n _options.IgnorePatterns.Add(context => context.Request.Headers.ContainsKey(\"ignore\"));\n\n var client = CreateClient();\n\n await client.SendAsync(new HttpRequestMessage(HttpMethod.Get, \"/foo\")\n {\n Headers =\n {\n { \"ignore\", \"1\" },\n }\n });\n\n Assert.Empty(_tracer.FinishedSpans());\n }\n\n [Fact]\n public async Task Calls_Options_OperationNameResolver()\n {\n _options.OperationNameResolver = _ => \"test\";\n\n var client = CreateClient();\n await client.GetAsync(\"/foo\");\n\n var finishedSpans = _tracer.FinishedSpans();\n Assert.Single(finishedSpans);\n\n var span = finishedSpans[0];\n Assert.Equal(\"test\", span.OperationName);\n }\n\n [Fact]\n public async Task Calls_Options_OnRequest()\n {\n bool onRequestCalled = false;\n _options.OnRequest = (_, __) => onRequestCalled = true;\n\n var client = CreateClient();\n await client.GetAsync(\"/foo\");\n\n Assert.True(onRequestCalled);\n }\n\n [Fact]\n public async Task Calls_Options_OnError()\n {\n bool onErrorCalled = false;\n _options.OnError = (_, __, ___) => onErrorCalled = true;\n\n var client = CreateClient();\n try\n {\n await client.GetAsync(\"/exception\");\n }\n catch (InvalidOperationException)\n {\n // The OnError handler is invoked after the request has been finished,\n // so we need to wait a little bit to make sure this test isn't failing sometimes.\n await Task.Delay(TimeSpan.FromMilliseconds(100));\n }\n \n Assert.True(onErrorCalled);\n }\n\n [Fact]\n public async Task Creates_error_span_if_request_throws_exception()\n {\n var client = CreateClient();\n try\n {\n await client.GetAsync(\"/exception\");\n }\n catch (InvalidOperationException)\n {\n }\n\n var finishedSpans = _tracer.FinishedSpans();\n Assert.Single(finishedSpans);\n\n var span = finishedSpans[0];\n Assert.True(span.Tags[Tags.Error.Key] as bool?);\n\n var logs = span.LogEntries;\n var error = logs.FirstOrDefault(x => x.Fields.TryGetValue(LogFields.Event, out object val) && val is string and \"error\");\n Assert.NotNull(error);\n Assert.Equal(\"InvalidOperationException\", error.Fields[LogFields.ErrorKind]);\n }\n }\n}\n"
},
{
"path": "test/OpenTracing.Contrib.NetCore.Tests/CoreFx/HttpHandlerDiagnosticTest.cs",
"content": "using System;\nusing System.Collections.Generic;\nusing System.Diagnostics;\nusing System.Net;\nusing System.Net.Http;\nusing System.Reflection;\nusing System.Threading;\nusing System.Threading.Tasks;\nusing Microsoft.Extensions.DependencyInjection;\nusing Microsoft.Extensions.Logging;\nusing Microsoft.Extensions.Options;\nusing OpenTracing.Contrib.NetCore.Configuration;\nusing OpenTracing.Contrib.NetCore.Internal;\nusing OpenTracing.Mock;\nusing OpenTracing.Tag;\nusing Xunit;\nusing Xunit.Abstractions;\n\nnamespace OpenTracing.Contrib.NetCore.Tests.CoreFx\n{\n [Collection(\"DiagnosticSource\") /* All DiagnosticSource tests must be in the same collection to ensure they are NOT run in parallel. */]\n public class HttpHandlerDiagnosticTest : IDisposable\n {\n private readonly MockTracer _tracer;\n private readonly HttpHandlerDiagnosticOptions _options;\n private readonly DiagnosticManager _diagnosticsManager;\n private readonly MockHttpMessageHandler _httpHandler;\n private readonly HttpClient _httpClient;\n\n private class MockHttpMessageHandler : HttpMessageHandler\n {\n public Func OnSend = request =>\n {\n return new HttpResponseMessage(HttpStatusCode.OK)\n {\n RequestMessage = request,\n Content = new StringContent(\"Response\")\n };\n };\n\n protected override async Task SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)\n {\n // HACK: There MUST be an awaiter otherwise exceptions are not caught by the DiagnosticsHandler.\n // https://github.com/dotnet/corefx/pull/27472\n await Task.CompletedTask;\n\n return OnSend(request);\n }\n }\n\n public HttpHandlerDiagnosticTest(ITestOutputHelper output)\n {\n _tracer = new MockTracer();\n _options = new HttpHandlerDiagnosticOptions();\n\n IServiceProvider serviceProvider = new ServiceCollection()\n .AddLogging(logging =>\n {\n logging.AddXunit(output);\n })\n .AddOpenTracingCoreServices(builder =>\n {\n builder.AddHttpHandler();\n builder.Services.AddSingleton(_tracer);\n builder.Services.AddSingleton(Options.Create(_options));\n })\n .BuildServiceProvider();\n\n _diagnosticsManager = serviceProvider.GetRequiredService();\n _diagnosticsManager.Start();\n\n // Inner handler for mocking the result\n _httpHandler = new MockHttpMessageHandler();\n\n // Wrap with DiagnosticsHandler (which is internal :( )\n#if NETCOREAPP3_1\n Type type = typeof(HttpClientHandler).Assembly.GetType(\"System.Net.Http.DiagnosticsHandler\");\n ConstructorInfo constructor = type.GetConstructors(BindingFlags.Instance | BindingFlags.Public)[0];\n HttpMessageHandler diagnosticsHandler = (HttpMessageHandler)constructor.Invoke(new object[] { _httpHandler });\n#else\n DistributedContextPropagator propagator = DistributedContextPropagator.CreateDefaultPropagator();\n\n Type type = typeof(HttpClientHandler).Assembly.GetType(\"System.Net.Http.DiagnosticsHandler\");\n ConstructorInfo constructor = type.GetConstructors(BindingFlags.Instance | BindingFlags.Public)[0];\n HttpMessageHandler diagnosticsHandler = (HttpMessageHandler)constructor.Invoke(new object[] { _httpHandler, propagator, false /* autoRedirect */ });\n#endif\n\n _httpClient = new HttpClient(diagnosticsHandler);\n }\n\n public void Dispose()\n {\n _diagnosticsManager.Dispose();\n }\n\n [Fact]\n public async Task Creates_span()\n {\n await _httpClient.SendAsync(new HttpRequestMessage(HttpMethod.Get, new Uri(\"http://www.example.com/api/values\")));\n\n Assert.Single(_tracer.FinishedSpans());\n }\n\n [Fact]\n public async Task Span_is_child_of_parent()\n {\n // Create parent span\n using (var scope = _tracer.BuildSpan(\"parent\").StartActive())\n {\n await _httpClient.SendAsync(new HttpRequestMessage(HttpMethod.Get, new Uri(\"http://www.example.com/api/values\")));\n }\n\n var finishedSpans = _tracer.FinishedSpans();\n Assert.Equal(2, finishedSpans.Count);\n\n // Inner span is finished first\n var childSpan = finishedSpans[0];\n var parentSpan = finishedSpans[1];\n\n Assert.NotSame(parentSpan, childSpan);\n Assert.Single(childSpan.References);\n Assert.Equal(References.ChildOf, childSpan.References[0].ReferenceType);\n Assert.Same(parentSpan.Context, childSpan.References[0].Context);\n }\n\n [Fact]\n public async Task Span_has_correct_properties()\n {\n await _httpClient.SendAsync(new HttpRequestMessage(HttpMethod.Get, new Uri(\"http://www.example.com/api/values\")));\n\n var finishedSpans = _tracer.FinishedSpans();\n Assert.Single(finishedSpans);\n\n var span = finishedSpans[0];\n Assert.Empty(span.GeneratedErrors);\n Assert.Empty(span.LogEntries);\n Assert.Equal(\"HTTP GET\", span.OperationName);\n Assert.Null(span.ParentId);\n Assert.Empty(span.References);\n\n Assert.Equal(7, span.Tags.Count);\n Assert.Equal(Tags.SpanKindClient, span.Tags[Tags.SpanKind.Key]);\n Assert.Equal(\"HttpOut\", span.Tags[Tags.Component.Key]);\n Assert.Equal(\"GET\", span.Tags[Tags.HttpMethod.Key]);\n Assert.Equal(\"http://www.example.com/api/values\", span.Tags[Tags.HttpUrl.Key]);\n Assert.Equal(\"www.example.com\", span.Tags[Tags.PeerHostname.Key]);\n Assert.Equal(80, span.Tags[Tags.PeerPort.Key]);\n Assert.Equal(200, span.Tags[Tags.HttpStatus.Key]);\n }\n\n [Fact]\n public async Task Injects_trace_headers()\n {\n var request = new HttpRequestMessage(HttpMethod.Get, new Uri(\"http://www.example.com/api/values\"));\n\n await _httpClient.SendAsync(request);\n\n Assert.True(request.Headers.Contains(\"traceid\"));\n }\n\n [Fact]\n public async Task Does_not_inject_trace_headers_if_disabled_in_options()\n {\n#if NETCOREAPP3_1\n _options.InjectEnabled = req => !req.Properties.ContainsKey(\"ignore\");\n#else\n _options.InjectEnabled = req => !((IDictionary)req.Options).ContainsKey(\"ignore\");\n#endif\n\n var request = new HttpRequestMessage(HttpMethod.Get, new Uri(\"http://www.example.com/api/values\"));\n SetRequestOption(request, \"ignore\", true);\n\n await _httpClient.SendAsync(request);\n\n Assert.False(request.Headers.Contains(\"traceid\"));\n }\n\n [Fact]\n public async Task Ignores_requests_with_Ignore_property()\n {\n var request = new HttpRequestMessage(HttpMethod.Get, new Uri(\"http://www.example.com/api/values\"));\n SetRequestOption(request, HttpHandlerDiagnosticOptions.PropertyIgnore, true);\n\n await _httpClient.SendAsync(request);\n\n Assert.Empty(_tracer.FinishedSpans());\n }\n\n [Fact]\n public async Task Ignores_requests_with_custom_rule()\n {\n#if NETCOREAPP3_1\n _options.IgnorePatterns.Add(req => req.Properties.ContainsKey(\"foo\"));\n#else\n _options.IgnorePatterns.Add(req => ((IDictionary) req.Options).ContainsKey(\"foo\"));\n#endif\n\n var request = new HttpRequestMessage(HttpMethod.Get, new Uri(\"http://www.example.com/api/values\"));\n SetRequestOption(request, \"foo\", 1);\n\n await _httpClient.SendAsync(request);\n\n Assert.Empty(_tracer.FinishedSpans());\n }\n\n [Fact]\n public async Task Calls_Options_OperationNameResolver()\n {\n _options.OperationNameResolver = _ => \"foo\";\n\n await _httpClient.SendAsync(new HttpRequestMessage(HttpMethod.Get, new Uri(\"http://www.example.com/api/values\")));\n\n var finishedSpans = _tracer.FinishedSpans();\n Assert.Single(finishedSpans);\n\n var span = finishedSpans[0];\n Assert.Equal(\"foo\", span.OperationName);\n }\n\n [Fact]\n public async Task Calls_Options_OnRequest()\n {\n bool onRequestCalled = false;\n\n _options.OnRequest = (_, __) => onRequestCalled = true;\n\n await _httpClient.SendAsync(new HttpRequestMessage(HttpMethod.Get, new Uri(\"http://www.example.com/api/values\")));\n\n Assert.True(onRequestCalled);\n }\n\n [Fact]\n public async Task Calls_Options_OnError()\n {\n bool onErrorCalled = false;\n\n _options.OnError = (_, __, ___) => onErrorCalled = true;\n\n var request = new HttpRequestMessage(HttpMethod.Get, new Uri(\"http://www.example.com/api/values\"));\n\n _httpHandler.OnSend = _ => throw new InvalidOperationException();\n\n await Assert.ThrowsAsync(() => _httpClient.SendAsync(request));\n\n Assert.True(onErrorCalled);\n }\n\n [Fact]\n public async Task Creates_error_span_if_request_times_out()\n {\n var request = new HttpRequestMessage(HttpMethod.Get, new Uri(\"http://www.example.com/api/values\"));\n\n _httpHandler.OnSend = _ => throw new TaskCanceledException();\n\n await Assert.ThrowsAsync(() => _httpClient.SendAsync(request));\n\n var finishedSpans = _tracer.FinishedSpans();\n Assert.Single(finishedSpans);\n\n var span = finishedSpans[0];\n Assert.True(span.Tags[Tags.Error.Key] as bool?);\n }\n\n [Fact]\n public async Task Creates_error_span_if_request_fails()\n {\n var request = new HttpRequestMessage(HttpMethod.Get, new Uri(\"http://www.example.com/api/values\"));\n\n _httpHandler.OnSend = _ => throw new InvalidOperationException();\n\n await Assert.ThrowsAsync(() => _httpClient.SendAsync(request));\n\n var finishedSpans = _tracer.FinishedSpans();\n Assert.Single(finishedSpans);\n\n var span = finishedSpans[0];\n Assert.True(span.Tags[Tags.Error.Key] as bool?);\n }\n\n private void SetRequestOption(HttpRequestMessage request, string key, TValue value)\n {\n#if NETCOREAPP3_1\n request.Properties[key] = value;\n#else\n request.Options.Set(new HttpRequestOptionsKey(key), value);\n#endif\n }\n }\n}\n"
},
{
"path": "test/OpenTracing.Contrib.NetCore.Tests/Internal/DiagnosticManagerTest.cs",
"content": "using System.Collections.Generic;\nusing Microsoft.Extensions.Logging.Abstractions;\nusing Microsoft.Extensions.Options;\nusing OpenTracing.Contrib.NetCore.Internal;\nusing OpenTracing.Mock;\nusing OpenTracing.Noop;\nusing OpenTracing.Util;\nusing Xunit;\n\nnamespace OpenTracing.Contrib.NetCore.Tests.Internal\n{\n public class DiagnosticManagerTest\n {\n [Fact]\n public void Does_not_Start_if_Tracer_is_NoopTracer()\n {\n var loggerFactory = new NullLoggerFactory();\n var tracer = NoopTracerFactory.Create();\n var diagnosticSubscribers = new List();\n var options = Options.Create(new DiagnosticManagerOptions());\n\n using (DiagnosticManager diagnosticManager = new DiagnosticManager(loggerFactory, tracer, diagnosticSubscribers, options))\n {\n Assert.False(diagnosticManager.IsRunning);\n\n diagnosticManager.Start();\n\n Assert.False(diagnosticManager.IsRunning);\n }\n }\n\n [Fact]\n public void Does_not_Start_if_Tracer_is_GlobalTracer_with_NoopTracer()\n {\n var loggerFactory = new NullLoggerFactory();\n var tracer = GlobalTracer.Instance;\n var diagnosticSubscribers = new List();\n var options = Options.Create(new DiagnosticManagerOptions());\n\n using (DiagnosticManager diagnosticManager = new DiagnosticManager(loggerFactory, tracer, diagnosticSubscribers, options))\n {\n Assert.False(diagnosticManager.IsRunning);\n\n diagnosticManager.Start();\n\n Assert.False(diagnosticManager.IsRunning);\n }\n }\n\n [Fact]\n public void Start_if_valid_Tracer()\n {\n var loggerFactory = new NullLoggerFactory();\n var tracer = new MockTracer();\n var diagnosticSubscribers = new List();\n var options = Options.Create(new DiagnosticManagerOptions());\n\n using (DiagnosticManager diagnosticManager = new DiagnosticManager(loggerFactory, tracer, diagnosticSubscribers, options))\n {\n Assert.False(diagnosticManager.IsRunning);\n\n diagnosticManager.Start();\n\n Assert.True(diagnosticManager.IsRunning);\n }\n }\n }\n}\n"
},
{
"path": "test/OpenTracing.Contrib.NetCore.Tests/Internal/PropertyFetcherTest.cs",
"content": "using OpenTracing.Contrib.NetCore.Internal;\nusing Xunit;\n\nnamespace OpenTracing.Contrib.NetCore.Tests.Internal\n{\n public class PropertyFetcherTest\n {\n public class TestClass\n {\n public string TestProperty { get; set; }\n }\n\n [Fact]\n public void Fetch_NameNotFound_NullReturned()\n {\n var obj = new TestClass { TestProperty = \"TestValue\" };\n\n var sut = new PropertyFetcher(\"DifferentProperty\");\n\n var result = sut.Fetch(obj);\n\n Assert.Null(result);\n }\n\n [Fact]\n public void Fetch_NameFound_ValueReturned()\n {\n var obj = new TestClass { TestProperty = \"TestValue\" };\n\n var sut = new PropertyFetcher(\"TestProperty\");\n\n var result = sut.Fetch(obj);\n\n Assert.Equal(\"TestValue\", result);\n }\n\n [Fact]\n public void Fetch_NameFoundDifferentCasing_ValueReturned()\n {\n var obj = new TestClass { TestProperty = \"TestValue\" };\n\n var sut = new PropertyFetcher(\"testproperty\");\n\n var result = sut.Fetch(obj);\n\n Assert.Equal(\"TestValue\", result);\n }\n }\n}\n"
},
{
"path": "test/OpenTracing.Contrib.NetCore.Tests/Logging/LoggingDependencyInjectionTest.cs",
"content": "using System;\nusing Microsoft.Extensions.DependencyInjection;\nusing Microsoft.Extensions.Logging;\nusing OpenTracing.Propagation;\nusing Xunit;\n\nnamespace OpenTracing.Contrib.Netcore.Tests.Logging\n{\n public class LoggingDependencyInjectionTest\n {\n [Fact]\n public void Resolving_tracer_that_needs_ILoggerFactory_succeeds()\n {\n // https://github.com/opentracing-contrib/csharp-netcore/issues/14\n\n var serviceProvider = new ServiceCollection()\n .AddLogging()\n .AddOpenTracingCoreServices(ot =>\n {\n ot.AddLoggerProvider();\n ot.Services.AddSingleton