WindowsTimerResolutionService.cs

이 파일의 문서화 페이지로 가기

using Dreamine.Threading.Interfaces;
using Dreamine.Threading.Windows.Native;
 
namespace Dreamine.Threading.Windows.Services;

public sealed class WindowsTimerResolutionService : ITimerResolutionService, IDisposable
{
    private readonly object _syncRoot = new();
    private readonly uint _period;
    private int _referenceCount;
    private bool _disposed;

    public WindowsTimerResolutionService()
        : this(1)
    {
    }

    public WindowsTimerResolutionService(uint period)
    {
        _period = period == 0 ? 1 : period;
    }

    public void Begin()
    {
        lock (_syncRoot)
        {
            ThrowIfDisposed();
 
            if (_referenceCount == 0)
            {
                WinMmNativeMethods.timeBeginPeriod(_period);
            }
 
            _referenceCount++;
        }
    }

    public void End()
    {
        lock (_syncRoot)
        {
            if (_disposed)
            {
                return;
            }
 
            if (_referenceCount <= 0)
            {
                return;
            }
 
            _referenceCount--;
 
            if (_referenceCount == 0)
            {
                WinMmNativeMethods.timeEndPeriod(_period);
            }
        }
    }

    public void Dispose()
    {
        lock (_syncRoot)
        {
            if (_disposed)
            {
                return;
            }
 
            while (_referenceCount > 0)
            {
                WinMmNativeMethods.timeEndPeriod(_period);
                _referenceCount--;
            }
 
            _disposed = true;
        }
    }

    private void ThrowIfDisposed()
    {
        if (_disposed)
        {
            throw new ObjectDisposedException(nameof(WindowsTimerResolutionService));
        }
    }
}