internal Task DeleteDeadletterAsync(RedisDeadletter <T> deadletter)
{
var deadletterQueueName = RedisQueueConventions.GetDeadLetterQueueName(_queueName);
var hash = RedisQueueConventions.GetMessageHashKey(_queueName, deadletter.Message.Id);
return(Task.WhenAll(_db.KeyDeleteAsync(hash), _db.ListRemoveAsync(deadletterQueueName, _serializer.Serialize(deadletter.Message), -1)));
}
private async Task <RedisMessage <T> > GetMessageAsync()
{
try
{
var listItem = await _db.ListRightPopLeftPushAsync(_queueName, RedisQueueConventions.GetProcessingQueueName(_queueName)).ConfigureAwait(false);
if (listItem.IsNullOrEmpty)
{
return(null);
}
var message = _redisConfiguration.MessageSerializer.Deserialize <T>(listItem);
var hashKey = RedisQueueConventions.GetMessageHashKey(_queueName, message.Id);
var tasks = new Task[]
{
_db.StringSetAsync(RedisQueueConventions.GetMessageExpirationKey(_queueName, message.Id), DateTimeOffset.Now.ToUnixTimeMilliseconds()),
_db.HashIncrementAsync(hashKey, RedisHashKeys.DeliveryCount, 1),
};
await Task.WhenAll(tasks).ConfigureAwait(false);
var hash = await _db.HashGetAllAsync(hashKey).ConfigureAwait(false);
return(new RedisMessage <T>(listItem, message, hash, _queueName));
}
catch (RedisTimeoutException e)
{
_hostConfiguration.Log.Error(e, "Error retrieving redis message");
return(null);
}
catch (RedisException e)
{
_hostConfiguration.Log.Error(e, "Error retrieving redis message");
return(null);
}
}
public RedisMessage(RedisValue redisValue, string id, T message, HashEntry[] hashEntries, string queueName)
{
RedisValue = redisValue;
HashKey = RedisQueueConventions.GetMessageHashKey(queueName, id);
ExpirationKey = RedisQueueConventions.GetMessageExpirationKey(queueName, id);
Message = message;
HashEntries = hashEntries.ToStringDictionary();
}
private async Task UploadAttachment <T>(RedisListItem <T> message, string queueName, IDatabase db) where T : IRedisMessage
{
if (message.Body is ICommandWithAttachment attachmentMessage)
{
if (_attachmentProvider == null)
{
throw new AttachmentProviderMissingException();
}
if (attachmentMessage.Attachment != null)
{
var attachmentId = Guid.NewGuid().ToString("N");
await db.HashSetAsync(RedisQueueConventions.GetMessageHashKey(queueName, message.Id), AttachmentUtility.AttachmentKey, attachmentId).ConfigureAwait(false);
await _attachmentProvider.UploadAttachmentAsync(queueName, attachmentId, attachmentMessage.Attachment).ConfigureAwait(false);
}
}
}
private async Task <RedisMessage <T> > GetMessageAsync(CancellationTokenSource cancellationsSource)
{
if (cancellationsSource.IsCancellationRequested)
{
return(null);
}
try
{
byte[] listItem = await _db.ListRightPopLeftPushAsync(_queueName, RedisQueueConventions.GetProcessingQueueName(_queueName)).ConfigureAwait(false);
if (listItem == null)
{
cancellationsSource.Cancel();
return(null);
}
var message = _serializer.Deserialize <RedisListItem <T> >(listItem.AsSpan());
var hashKey = RedisQueueConventions.GetMessageHashKey(_queueName, message.Id);
var tasks = new Task[]
{
_db.StringSetAsync(RedisQueueConventions.GetMessageExpirationKey(_queueName, message.Id), DateTimeOffset.Now.ToUnixTimeMilliseconds()),
_db.HashIncrementAsync(hashKey, RedisHashKeys.DeliveryCount)
};
await Task.WhenAll(tasks).ConfigureAwait(false);
var hash = await _db.HashGetAllAsync(hashKey).ConfigureAwait(false);
return(new RedisMessage <T>(listItem, message.Id, message.Body, hash, _queueName));
}
catch (RedisTimeoutException e)
{
_log.Error(e, "Error retrieving redis message");
return(null);
}
catch (RedisException e)
{
_log.Error(e, "Error retrieving redis message");
return(null);
}
}
internal async IAsyncEnumerable <RedisDeadletter <T> > PeekDeadlettersAsync(int limit)
{
if (limit >= 1)
{
limit--; //0 is the first element of the list, thus 0 will return 1
}
var values = await _db.ListRangeAsync(RedisQueueConventions.GetDeadLetterQueueName(_queueName), 0, limit).ConfigureAwait(false);
foreach (byte[] value in values)
{
var deadletter = new RedisDeadletter <T> {
Message = _serializer.Deserialize <RedisListItem <T> >(value.AsSpan())
};
var hash = RedisQueueConventions.GetMessageHashKey(_queueName, deadletter.Message.Id);
var hashes = await _db.HashGetAllAsync(hash).ConfigureAwait(false);
deadletter.HashEntries = hashes.ToStringDictionary();
yield return(deadletter);
}
}
internal async Task RequeueDeadletterAsync()
{
var deadLetterQueueName = RedisQueueConventions.GetDeadLetterQueueName(_queueName);
var deadLetterProcessingQueueName = RedisQueueConventions.GetProcessingQueueName(deadLetterQueueName);
byte[] listItem = await _db.ListRightPopLeftPushAsync(deadLetterQueueName, deadLetterProcessingQueueName).ConfigureAwait(false);
if (listItem == null)
{
return;
}
var message = _serializer.Deserialize <RedisListItem <T> >(listItem.AsSpan());
var hashKey = RedisQueueConventions.GetMessageHashKey(_queueName, message.Id);
var expirationKey = RedisQueueConventions.GetMessageExpirationKey(_queueName, message.Id);
await _db.HashDeleteAsync(hashKey, new RedisValue[] { expirationKey, RedisHashKeys.DeliveryCount }).ConfigureAwait(false);
await _db.ListRightPopLeftPushAsync(deadLetterProcessingQueueName, _queueName).ConfigureAwait(false);
await _db.PublishAsync(_queueName, 0, CommandFlags.FireAndForget).ConfigureAwait(false);
}
