protected async Task <IEnumerable <T> > GetMessagesAsync <T>(string request, Func <byte[], int, MessageContainer <T> > messageHandler)
{
var client = await _lookupDispatcher.TakeAsync();
var messages = new List <T>();
var invalidMessages = new List <InvalidMessage <T> >();
var ct = new CancellationTokenSource(_timeout);
var res = new TaskCompletionSource <IEnumerable <T> >();
ct.Token.Register(() => res.TrySetCanceled(), false);
void SocketClientOnMessageReceived(object sender, SocketMessageEventArgs args)
{
var container = messageHandler(args.Message, args.Count);
// exception must be throw at the very end when all messages have been received and parsed to avoid
// continuation in the next request since we don't use request id
if (container.ErrorMessage != null)
{
res.TrySetException(_exceptionFactory.CreateNew(request, container.ErrorMessage, container.MessageTrace));
return;
}
messages.AddRange(container.Messages);
invalidMessages.AddRange(container.InvalidMessages);
if (!container.End)
{
return;
}
if (invalidMessages.Count > 0)
{
res.TrySetException(_exceptionFactory.CreateNew(request, invalidMessages, messages));
return;
}
res.TrySetResult(messages);
}
client.MessageReceived += SocketClientOnMessageReceived;
await _lookupRateLimiter.WaitAsync().ConfigureAwait(false);
client.Send(request);
await res.Task.ContinueWith(x =>
{
client.MessageReceived -= SocketClientOnMessageReceived;
_lookupDispatcher.Add(client);
ct.Dispose();
}, TaskContinuationOptions.None).ConfigureAwait(false);
return(await res.Task.ConfigureAwait(false));
}
// TODO(mathip): add support for requestId parsing.
public async Task <string> GetFilenameAsync(string request)
{
var client = await _lookupDispatcher.TakeAsync();
var filename = Path.GetRandomFileName();
var binaryWriter = new BinaryWriter(File.Open(filename, FileMode.OpenOrCreate));
var ct = new CancellationTokenSource(_timeout);
var res = new TaskCompletionSource <string>();
ct.Token.Register(() => res.TrySetCanceled(), false);
void SocketClientOnMessageReceived(object sender, SocketMessageEventArgs args)
{
// check for errors
if (args.Message[0] == IQFeedDefault.PrototolErrorCharacter && args.Message[1] == IQFeedDefault.ProtocolDelimiterCharacter)
{
// at this level, we might have true negative, further checks needed
var received = Encoding.ASCII.GetString(args.Message, 0, args.Count);
var messages = received.SplitFeedLine();
var errorMessage = ParseErrorMessage(messages);
if (!string.IsNullOrEmpty(errorMessage))
{
// error has been confirmed
res.TrySetException(_exceptionFactory.CreateNew(request, errorMessage, received));
return;
}
}
binaryWriter.Write(args.Message, 0, args.Count);
// check if the message end
if (args.Message.EndsWith(args.Count, _endOfMsgBytes))
{
res.TrySetResult(filename);
}
}
client.MessageReceived += SocketClientOnMessageReceived;
await _lookupRateLimiter.WaitAsync().ConfigureAwait(false);
client.Send(request);
await res.Task.ContinueWith(x =>
{
binaryWriter.Close();
client.MessageReceived -= SocketClientOnMessageReceived;
_lookupDispatcher.Add(client);
ct.Dispose();
if (res.Task.IsFaulted)
{
File.Delete(filename);
}
}, TaskContinuationOptions.None).ConfigureAwait(false);
return(await res.Task.ConfigureAwait(false));
}
public void Should_Return_InvalidData_Exception()
{
// Arrange
var request = "request";
var messageTrace = "message trace";
// Act
var exception = _exceptionFactory.CreateNew(request, InvalidDataIQFeedException.InvalidData, messageTrace);
// Assert
var expectedException = new InvalidDataIQFeedException(request, InvalidDataIQFeedException.InvalidData, messageTrace);
Assert.AreEqual(exception, expectedException);
}
protected async Task <IEnumerable <T> > GetMessagesAsync <T>(string request, Func <byte[], int, MessageContainer <T> > messageHandler)
{
var client = await _lookupDispatcher.TakeAsync();
var messages = new List <T>();
var ct = new CancellationTokenSource(_timeout);
var res = new TaskCompletionSource <IEnumerable <T> >();
ct.Token.Register(() => res.TrySetCanceled(), false);
void SocketClientOnMessageReceived(object sender, SocketMessageEventArgs args)
{
var container = messageHandler(args.Message, args.Count);
if (container.ErrorMessage != null)
{
res.TrySetException(_exceptionFactory.CreateNew(request, container.ErrorMessage, container.MessageTrace));
return;
}
messages.AddRange(container.Messages);
if (container.End)
{
res.TrySetResult(messages);
}
}
client.MessageReceived += SocketClientOnMessageReceived;
client.Send(request);
await res.Task.ContinueWith(x =>
{
client.MessageReceived -= SocketClientOnMessageReceived;
_lookupDispatcher.Add(client);
ct.Dispose();
}, TaskContinuationOptions.None).ConfigureAwait(false);
return(await res.Task.ConfigureAwait(false));
}
public void Should_Return_InvalidData_Exception()
{
// Arrange
var request = "request";
var intervalMessage = new IntervalMessage(new DateTime(2000, 01, 01, 9, 30, 00), 1, 1, 1, 1, 1, 1, 0);
var invalidMessages = new List <InvalidMessage <IntervalMessage> >()
{
new InvalidMessage <IntervalMessage>(intervalMessage, intervalMessage.ToCsv())
};
var messages = new List <IntervalMessage>()
{
intervalMessage
};
// Act
var exception = _exceptionFactory.CreateNew(request, invalidMessages, messages);
// Assert
var expectedException = new InvalidDataIQFeedException <IntervalMessage>(request, invalidMessages, messages);
Assert.AreEqual(exception, expectedException);
}