Musify/backend/Core/Musify.Infrastructure/Observability/OpenTelemetryDependencyInjection.cs

using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Musify.Application.Configuration;
using Musify.Infrastructure.Configuration;
using OpenTelemetry.Exporter;
using OpenTelemetry.Logs;
using OpenTelemetry.Metrics;
using OpenTelemetry.Resources;
using OpenTelemetry.Trace;

namespace Musify.Infrastructure.Observability;

public static class OpenTelemetryDependencyInjection
{
    public static IServiceCollection AddObservability(this IServiceCollection services, IConfiguration configuration)
    {
        services.AddValidatedOptions<OpenTelemetryConfiguration>(configuration);

        services
            .AddOptions<OtlpExporterOptions>()
            .Configure<OpenTelemetryConfiguration>((exporterOptions, openTelemetryConfiguration) =>
            {
                exporterOptions.Endpoint = new Uri(openTelemetryConfiguration.OtlpEndpoint);
            });

        services
            .AddOpenTelemetry()
            .ConfigureResource(resource =>
            {
                resource.AddService("Musify");
            })
            .WithTracing(tracing =>
                tracing.AddSource("MassTransit")
                    .AddAspNetCoreInstrumentation()
                    .AddHttpClientInstrumentation()
                    .AddEntityFrameworkCoreInstrumentation()
                    .AddOtlpExporter()
            )
            .WithMetrics(metrics =>
                metrics.AddMeter("MassTransit")
                    .AddAspNetCoreInstrumentation()
                    .AddHttpClientInstrumentation()
                    .AddRuntimeInstrumentation()
                    .AddProcessInstrumentation()
                    .AddOtlpExporter()
            )
            .WithLogging(logging =>
                logging.AddOtlpExporter()
            );

        return services;
    }
}