osu-lazer/osu.Game/Performance/HighPerformanceSessionManager.cs

// Copyright (c) ppy Pty Ltd <contact@ppy.sh>. Licensed under the MIT Licence.
// See the LICENCE file in the repository root for full licence text.

using System;
using System.Runtime;
using osu.Framework.Allocation;
using osu.Framework.Graphics;

namespace osu.Game.Performance
{
    public partial class HighPerformanceSessionManager : Component
    {
        private GCLatencyMode originalGCMode;

        public IDisposable BeginSession()
        {
            EnableHighPerformanceSession();
            return new InvokeOnDisposal<HighPerformanceSessionManager>(this, static m => m.DisableHighPerformanceSession());
        }

        protected virtual void EnableHighPerformanceSession()
        {
            originalGCMode = GCSettings.LatencyMode;
            GCSettings.LatencyMode = GCLatencyMode.LowLatency;

            // Without doing this, the new GC mode won't kick in until the next GC, which could be at a more noticeable point in time.
            GC.Collect(0);
        }

        protected virtual void DisableHighPerformanceSession()
        {
            if (GCSettings.LatencyMode == GCLatencyMode.LowLatency)
                GCSettings.LatencyMode = originalGCMode;

            // No GC.Collect() as we were already collecting at a higher frequency in the old mode.
        }
    }
}