CryptoExchange.Net/CryptoExchange.Net/Sockets/PendingRequest.cs

using CryptoExchange.Net.Objects;
using Newtonsoft.Json.Linq;
using System;
using System.Threading;

namespace CryptoExchange.Net.Sockets
{
    internal class PendingRequest
    {
        public Func<JToken, bool> Handler { get; }
        public JToken? Result { get; private set; }
        public bool Completed { get; private set; }
        public AsyncResetEvent Event { get; }
        public DateTime RequestTimestamp { get; set; }
        public TimeSpan Timeout { get; }
        public SocketSubscription? Subscription { get; }

        private CancellationTokenSource cts;

        public PendingRequest(Func<JToken, bool> handler, TimeSpan timeout, SocketSubscription? subscription)
        {
            Handler = handler;
            Event = new AsyncResetEvent(false, false);
            Timeout = timeout;
            RequestTimestamp = DateTime.UtcNow;
            Subscription = subscription;

            cts = new CancellationTokenSource(timeout);
            cts.Token.Register(Fail, false);
        }

        public bool CheckData(JToken data)
        {
            return Handler(data);
        }

        public bool Succeed(JToken data)
        {
            Result = data;
            Completed = true;
            Event.Set();
            return true;
        }

        public void Fail()
        {
            Completed = true;
            Event.Set();
        }
    }
}