CommandStatus.cs

using System;
using System.Collections.Generic;
using System.Net;
using System.Threading.Tasks;
using Waher.Events;
 
namespace Waher.Networking.Cluster
{
    internal class CommandStatus<ResponseType> : CommandStatusBase
    {
        public Dictionary<IPEndPoint, KeyValuePair<ResponseType, Exception>> Responses = new Dictionary<IPEndPoint, KeyValuePair<ResponseType, Exception>>();
        public EventHandlerAsync<ClusterResponseEventArgs<ResponseType>> Callback;
 
        public override bool IsComplete(EndpointStatus[] Statuses)
        {
            lock (this.Responses)
            {
                foreach (EndpointStatus Status in Statuses)
                {
                    if (!this.Responses.ContainsKey(Status.Endpoint))
                        return false;
                }
            }
 
            return true;
        }
 
        public EndpointResponse<ResponseType>[] GetResponses(EndpointStatus[] Statuses)
        {
            EndpointResponse<ResponseType>[] Result;
            int i, c;
 
            lock (this.Responses)
            {
                foreach (EndpointStatus Status in Statuses)
                {
                    if (!this.Responses.ContainsKey(Status.Endpoint))
                        this.Responses[Status.Endpoint] = new KeyValuePair<ResponseType, Exception>(default, new TimeoutException("No response returned."));
                }
 
                Result = new EndpointResponse<ResponseType>[c = this.Responses.Count];
 
                i = 0;
                foreach (KeyValuePair<IPEndPoint, KeyValuePair<ResponseType, Exception>> P in this.Responses)
                    Result[i++] = new EndpointResponse<ResponseType>(P.Key, P.Value.Key, P.Value.Value);
            }
 
            return Result;
        }
 
        public override void AddResponse(IPEndPoint From, object Response)
        {
            lock (this.Responses)
            {
                if (Response is ResponseType Response2)
                    this.Responses[From] = new KeyValuePair<ResponseType, Exception>(Response2, null);
                else
                    this.Responses[From] = new KeyValuePair<ResponseType, Exception>(default, new Exception("Unexpected response returned."));
            }
        }
 
        public override void AddError(IPEndPoint From, Exception Error)
        {
            lock (this.Responses)
            {
                this.Responses[From] = new KeyValuePair<ResponseType, Exception>(default, Error);
            }
        }
 
        public override Task RaiseResponseEvent(EndpointStatus[] CurrentStatus)
        {
            return this.Callback.Raise(this, new ClusterResponseEventArgs<ResponseType>(
                this.Command, this.GetResponses(CurrentStatus), this.State));
        }
 
    }
}