Synnefo » Pithos MS Client

// -----------------------------------------------------------------------

// <copyright file="AsyncManualResetEvent.cs" company="Microsoft">

// TODO: Update copyright text.

// </copyright>

// -----------------------------------------------------------------------

using System.Threading;

using System.Threading.Tasks;

namespace Pithos.Core.Agents

{

    using System;

    using System.Collections.Generic;

    using System.Linq;

    using System.Text;

    /// <summary>

    /// TODO: Update summary.

    /// </summary>

    public class AsyncManualResetEvent

    {

        private volatile TaskCompletionSource<bool> _tcs;

        public AsyncManualResetEvent()

        {

            _tcs = new TaskCompletionSource<bool>();

        }

        public AsyncManualResetEvent(bool signalled)

        {            

            _tcs = new TaskCompletionSource<bool>();

            if (signalled)

                _tcs.SetResult(true);

        }

        public Task WaitAsync() { return _tcs.Task; }

        public void Wait()

        {            

            _tcs.Task.Wait();

        }

        public void Set() { _tcs.TrySetResult(true); }

        public void Reset()

        {

            while (true)

            {

                var tcs = _tcs;

                if (!tcs.Task.IsCompleted ||

                    Interlocked.CompareExchange(ref _tcs, new TaskCompletionSource<bool>(), tcs) == tcs)

                    return;

            }

        }

    }

}