using System;
using System.IO;
using System.Runtime.Serialization;
using System.Threading;
using RabbitMQ.Client;
using RabbitMQ.Client.Events;
using RabbitMQ.Client.Exceptions;
using RabbitMQ.Util;
using System.Configuration;
namespace RabbitMQClient
{
public class MessageQueueConsumer : IHealthVerifiable
{
public class TimeoutException : Exception { }
private class BufferQueue : SharedQueue<BasicDeliverEventArgs>
{
public int Count()
{
return this.m_queue.Count;
}
}
private const int DEFAULT_ACK_COUNT = 1000;
private String connString;
private EventingBasicConsumer consumer;
private IConnection conn;
private IModel channel;
private String queueName;
private BufferQueue buffer;
private Object locker = new Object();
private ushort prefetchCount;
private ushort ackCount;
private bool stopRequested;
private BasicDeliverEventArgs lastMsgDequeued;
public MessageQueueConsumer(String queueName, String connString, ushort? ackCount = null)
{
this.queueName = queueName;
this.connString = connString;
if (ackCount != null)
this.ackCount = ackCount.Value;
else
{
if (!string.IsNullOrEmpty(ConfigurationManager.AppSettings["DefultAckCount"]))
this.ackCount = ushort.Parse(ConfigurationManager.AppSettings["DefultAckCount"].ToString());
else
ackCount = DEFAULT_ACK_COUNT;
}
this.prefetchCount = (ushort)(this.ackCount * 2);
stopRequested = false;
InitConsumer();
}
~MessageQueueConsumer()
{
Close();
}
public void StopRequested()
{
stopRequested = true;
}
public void Close()
{
}
private void InitConsumer()
{
ConnectionFactory factory = new ConnectionFactory();
factory.Uri = connString;
conn = factory.CreateConnection();
channel = conn.CreateModel();
channel.BasicQos(0, prefetchCount, false);
buffer = new BufferQueue();
consumer = new EventingBasicConsumer(channel);
channel.BasicConsume(queueName, false, consumer);
// when message is recieved do following
consumer.Received += (model, message) =>
{
//if reached max buffer size sleep for 3 seconds to give dequeing thread some time to catch up
if (buffer.Count() > DEFAULT_ACK_COUNT)
Thread.Sleep(3000);
buffer.Enqueue(message);
//reached max buffer ack messages in queue
if (buffer.Count() > DEFAULT_ACK_COUNT)
channel.BasicAck(message.DeliveryTag, true);
};
}
/// <summary>
/// Get the next event from the queue
/// </summary>
/// <returns>Event</returns>
public byte[] Dequeue(int? timeout = null)
{
lock (locker)
{
try
{
return AttemptDequeue(timeout);
}
catch (EndOfStreamException)
{
// Network interruption while reading the input stream
InitConsumer();
return AttemptDequeue(timeout);
}
catch (OperationInterruptedException)
{
// The consumer was removed, either through channel or connection closure, or through the
// action of IModel.BasicCancel().
// Attempt to reopen and try again
InitConsumer();
return AttemptDequeue(timeout);
}
catch (ConnectFailureException)
{
//Problems connecting to the queue, wait 10sec, then try again.
Thread.Sleep(10000);
InitConsumer();
return AttemptDequeue(timeout);
}
}
}
private byte[] AttemptDequeue(int? tomeout)
{
BasicDeliverEventArgs message;
while (true)
{
if (stopRequested)
{
//Ack last message in buffer and send that
if (lastMsgDequeued != null)
{
channel.BasicAck(lastMsgDequeued.DeliveryTag, true);
lastMsgDequeued = null;
return lastMsgDequeued.Body;
}
}
else //while buffer has no events
{
if (buffer.Count() == 0)
{
if (lastMsgDequeued != null)
{
channel.BasicAck(lastMsgDequeued.DeliveryTag, true);
lastMsgDequeued = null;
}
Thread.Sleep(3000);
}
else
{
message = buffer.Dequeue();
lastMsgDequeued = message;
break;
}
}
}
try
{
return message.Body;
}
catch (Exception e)
{
throw new SerializationException("Error deserializing queued message:", e);
}
}
/// <summary>
/// Attempt to connect to queue to see if it is available
/// </summary>
/// <returns>true if queue is available</returns>
public bool IsHealthy()
{
try
{
if (channel.IsOpen)
return true;
else
{
InitConsumer();
return true;
}
}
catch
{
return false;
}
}
}
}