public override async Task <Response> RequestAsync(Command command, TimeSpan timeout)
{
FutureResponse future = AsyncRequest(command);
if (timeout.TotalMilliseconds > 0)
{
using (CancellationTokenSource ct = new CancellationTokenSource(timeout))
{
using (ct.Token.Register(() =>
{
if (future.TrySetException(new RequestTimedOutException(timeout)))
{
requestMap.Remove(command.CommandId);
}
}, false))
{
return(await future.Task);
}
}
}
else
{
return(await future.Task);
}
}
