redx177 / elastic-apm-extensions Goto Github PK

public static IHostBuilder CreateHostBuilder(string[] args) =>
    Host.CreateDefaultBuilder(args)
        .ConfigureWebHostDefaults(webBuilder =>
        {
            webBuilder.UseStartup<Startup>();
        })
        .UseElasticApm(
            // Add additional diagnostic subscribers e.g. AspNetCore, Http, etc
            new MassTransitDiagnosticsSubscriber());

new MassTransitDiagnosticsSubscriber(o => 
  o.ReceiveLabel = context => context.Host.AbsolutePath)

new MassTransitDiagnosticsSubscriber(o => 
  o.ReceiveLabel = context => 
    if (context is RabbitMqReceiveContext rabbitMqContext)
      ? rabbitMqContext.Exchange
      : default;)

public class Startup
{
    public void ConfigureServices(IServiceCollection services)
    {
        // Add your services

        // Add HotChocolate GraphQL Server
        services
            .AddGraphQLServer()
            .AddQueryType<Query>()
            .AddObservability(); // Register instrumentation for Elastic APM
    }
}

services
    .AddGraphQLServer()
    .AddObservability(o => 
        o.Enrich = (transaction, operationDetails) =>
        {
            transaction.SetLabel("GraphQLResult", operationDetails.HasFailed);
            transaction.SetLabel("Department", Environment.GetEnvironmentVariable("DEPARTMENT"));
        });