private async Task ProcessingTaskAsync(byte[] data)
{
var oldContext = Core.ContextGroupName;
try
{
(var body, var correlationId) = NATSQueueClient.GetFromMessageBody(data);
Counters.IncrementTotalReceivingBytes(body.Count);
Core.Log.LibVerbose("Received {0} bytes from the Queue '{1}/{2}'", body.Count, Connection.Route, Connection.Name);
var messageBody = ReceiverSerializer.Deserialize(body, _messageType);
switch (messageBody)
{
case RequestMessage request when request.Header != null:
if (!string.IsNullOrEmpty(request.Header.ContextGroupName))
{
Core.ContextGroupName = request.Header.ContextGroupName;
}
request.Header.ApplicationReceivedTime = Core.Now;
Counters.IncrementReceivingTime(request.Header.TotalTime);
if (request.Header.ClientName != Config.Name)
{
Core.Log.Warning("The Message Client Name '{0}' is different from the Server Name '{1}'", request.Header.ClientName, Config.Name);
}
var evArgs = new RequestReceivedEventArgs(_name, Connection, request, body.Count, SenderSerializer, Config.RequestOptions.ServerReceiverOptions.CancellationBeforeClientResponseTimeoutInSec);
if (request.Header.ResponseQueue != null)
{
evArgs.ResponseQueues.Add(request.Header.ResponseQueue);
}
await OnRequestReceivedAsync(evArgs).ConfigureAwait(false);
break;
case ResponseMessage response when response.Header != null:
if (!string.IsNullOrEmpty(response.Header.Request.Header.ContextGroupName))
{
Core.ContextGroupName = response.Header.Request.Header.ContextGroupName;
}
response.Header.Response.ApplicationReceivedTime = Core.Now;
Counters.IncrementReceivingTime(response.Header.Response.TotalTime);
var evArgs2 = new ResponseReceivedEventArgs(_name, response, body.Count);
await OnResponseReceivedAsync(evArgs2).ConfigureAwait(false);
break;
}
Counters.IncrementTotalMessagesProccesed();
}
catch (Exception ex)
{
Counters.IncrementTotalExceptions();
Core.Log.Write(ex);
Interlocked.Exchange(ref _exceptionSleep, 1);
}
finally
{
Core.ContextGroupName = oldContext;
}
}
private void MessageHandler(object sender, MsgHandlerEventArgs e)
{
Core.Log.LibVerbose("Message received");
try
{
(var body, var correlationId) = NATSQueueClient.GetFromMessageBody(e.Message.Data);
var rMsg = new NATSQMessage
{
CorrelationId = correlationId,
Body = body
};
Task.Run(() => EnqueueMessageToProcessAsync(ProcessingTaskAsync, rMsg));
}
catch (Exception ex)
{
Core.Log.Write(ex);
}
}
/// <inheritdoc />
/// <summary>
/// On Send message data
/// </summary>
/// <param name="message">Response message instance</param>
/// <param name="e">Event Args</param>
protected override Task <int> OnSendAsync(ResponseMessage message, RequestReceivedEventArgs e)
{
if (e.ResponseQueues?.Any() != true)
{
return(TaskHelper.CompleteValueMinus1);
}
var senderOptions = Config.ResponseOptions.ServerSenderOptions;
if (senderOptions == null)
{
throw new ArgumentNullException("ServerSenderOptions");
}
var data = SenderSerializer.Serialize(message);
var body = NATSQueueClient.CreateMessageBody(data, message.CorrelationId);
var response = true;
foreach (var queue in e.ResponseQueues)
{
try
{
var producerPool = _rQueue.GetOrAdd(queue.Route, q => new ObjectPool <IConnection>(pool =>
{
Core.Log.LibVerbose("New Producer from QueueServer");
return(_factory.CreateConnection(queue.Route));
}, null, 1));
Core.Log.LibVerbose("Sending {0} bytes to the Queue '{1}' with CorrelationId={2}", data.Count, queue.Route + "/" + queue.Name, message.CorrelationId);
var producer = producerPool.New();
producer.Publish(queue.Name, body);
producerPool.Store(producer);
}
catch (Exception ex)
{
response = false;
Core.Log.Write(ex);
}
}
return(response ? Task.FromResult(data.Count) : TaskHelper.CompleteValueMinus1);
}
/// <inheritdoc />
/// <summary>
/// On Send message data
/// </summary>
/// <param name="message">Response message instance</param>
/// <param name="e">Event Args</param>
protected override Task <int> OnSendAsync(ResponseMessage message, RequestReceivedEventArgs e)
{
if (e.ResponseQueues?.Any() != true)
{
return(TaskHelper.CompleteValueMinus1);
}
var senderOptions = Config.ResponseOptions.ServerSenderOptions;
if (senderOptions is null)
{
throw new NullReferenceException("ServerSenderOptions is null.");
}
var data = SenderSerializer.Serialize(message);
var body = NATSQueueClient.CreateMessageBody(data, message.CorrelationId);
var response = true;
foreach (var queue in e.ResponseQueues)
{
try
{
var producer = _rQueue.GetOrAdd(queue.Route, qRoute =>
{
Core.Log.LibVerbose("New Producer from QueueServer");
return(Extensions.InvokeWithRetry(() => _factory.CreateConnection(qRoute), 5000, int.MaxValue).WaitAsync());
});
Core.Log.LibVerbose("Sending {0} bytes to the Queue '{1}/{2}' with CorrelationId={3}", data.Count, queue.Route, queue.Name, message.CorrelationId);
producer.Publish(queue.Name, body);
}
catch (Exception ex)
{
response = false;
Core.Log.Write(ex);
}
}
return(response ? Task.FromResult(data.Count) : TaskHelper.CompleteValueMinus1);
}
