Umbraco-CMS/src/Umbraco.Web.Common/Profiler/InitializeWebProfiling.cs

// Copyright (c) Umbraco.
// See LICENSE for more details.

using Microsoft.Extensions.Logging;
using Umbraco.Cms.Core;
using Umbraco.Cms.Core.Events;
using Umbraco.Cms.Core.Logging;
using Umbraco.Cms.Core.Notifications;

namespace Umbraco.Cms.Web.Common.Profiler;

/// <summary>
///     Initialized the web profiling. Ensures the boot process profiling is stopped.
/// </summary>
public class InitializeWebProfiling : INotificationHandler<UmbracoApplicationStartingNotification>
{
    private readonly bool _profile;
    private readonly WebProfiler? _profiler;

    /// <summary>
    ///     Initializes a new instance of the <see cref="InitializeWebProfiling" /> class.
    /// </summary>
    public InitializeWebProfiling(IProfiler profiler, ILogger<InitializeWebProfiling> logger)
    {
        _profile = true;

        // although registered in UmbracoBuilderExtensions.AddUmbraco, ensure that we have not
        // been replaced by another component, and we are still "the" profiler
        _profiler = profiler as WebProfiler;
        if (_profiler != null)
        {
            return;
        }

        // if VoidProfiler was registered, let it be known
        if (profiler is NoopProfiler)
        {
            logger.LogInformation(
                "Profiler is VoidProfiler, not profiling (must run debug mode to profile).");
        }

        _profile = false;
    }

    /// <inheritdoc />
    public void Handle(UmbracoApplicationStartingNotification notification)
    {
        if (_profile && notification.RuntimeLevel == RuntimeLevel.Run)
        {
            // Stop the profiling of the booting process
            _profiler?.StopBoot();
        }
    }
}