public void Enqueue(System.Data.Common.DbConnection connection, System.Data.Common.DbTransaction transaction, string queue, string jobId)
{
var messageQueue = MemoryQueueBank.Get(_pathPattern, queue);
var message = new MemoryQueueMessage {
Label = jobId, IsTaking = false
};
messageQueue.AddOrUpdate(jobId, message, (key, oldValue) =>
{
oldValue.IsTaking = false;
return(oldValue);
});
}
public IFetchedJob Dequeue(string[] queues, CancellationToken cancellationToken)
{
var queueIndex = 0;
string jobId = null;
do
{
cancellationToken.ThrowIfCancellationRequested();
MemoryTransaction transaction = null;
try
{
var messageQueue = MemoryQueueBank.Get(_pathPattern, queues[queueIndex]);
transaction = new MemoryTransaction(messageQueue);
MemoryQueueMessage message = queueIndex == queues.Length - 1
? transaction.Receive(SyncReceiveTimeout)
: transaction.Receive(ReceiveTimeout);
if (message != null)
{
jobId = message.Label;
return(new MemoryFetchedJob(transaction, message.Label));
}
}
catch (Exception ex)
{
// Receive timeout occurred, we should just switch to the next queue
}
finally
{
if (jobId == null)
{
transaction?.Dispose();
}
}
queueIndex = (queueIndex + 1) % queues.Length;
Thread.Sleep(1);
} while (true);
}
