src/EditorFeatures/TestUtilities/Async/AsynchronousOperationBlocker.cs - main - githubmirror/roslyn

src/EditorFeatures/TestUtilities/Async/AsynchronousOperationBlocker.cs

85 строк2 KB

Tomáš Matoušek

More improvements to source packages (#78587)

16 май 2025, 01:46

16 май 2025, 01:46015a854

// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

#nullable disable

using System;
using System.Threading;
using Microsoft.CodeAnalysis.ErrorReporting;

namespace Roslyn.Test.Utilities;

public sealed class AsynchronousOperationBlocker : IDisposable
{
    private readonly ManualResetEvent _waitHandle;
    private readonly object _lockObj = new();
    private bool _blocking;
    private bool _disposed;

    public AsynchronousOperationBlocker()
    {
        _waitHandle = new ManualResetEvent(false);
        _blocking = true;
    }

    public bool IsBlockingOperations
    {
        get
        {
            lock (_lockObj)
            {
                return _blocking;
            }
        }

        private set
        {
            lock (_lockObj)
            {
                if (_blocking == value)
                {
                    return;
                }

                _blocking = value;
                if (!_disposed)
                {
                    if (_blocking)
                    {
                        _waitHandle.Reset();
                    }
                    else
                    {
                        _waitHandle.Set();
                    }
                }
            }
        }
    }

    public void BlockOperations()
        => this.IsBlockingOperations = true;

    public void UnblockOperations()
        => this.IsBlockingOperations = false;

    public bool WaitIfBlocked(TimeSpan timeout)
    {
        if (_disposed)
        {
            FailFast.Fail("Badness");
        }

        return _waitHandle.WaitOne(timeout);
    }

    public void Dispose()
    {
        if (!_disposed)
        {
            _disposed = true;
            _waitHandle.Dispose();
        }
    }
}