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;
        }
    }
}