73 lines
2.2 KiB
C#
73 lines
2.2 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace StellaOps.Scanner.Queue.Redis;
|
|
|
|
internal sealed class RedisScanQueueLease : IScanQueueLease
|
|
{
|
|
private readonly RedisScanQueue _queue;
|
|
private int _completed;
|
|
|
|
internal RedisScanQueueLease(
|
|
RedisScanQueue queue,
|
|
string messageId,
|
|
string jobId,
|
|
byte[] payload,
|
|
int attempt,
|
|
DateTimeOffset enqueuedAt,
|
|
DateTimeOffset leaseExpiresAt,
|
|
string consumer,
|
|
string? idempotencyKey,
|
|
IReadOnlyDictionary<string, string> attributes)
|
|
{
|
|
_queue = queue;
|
|
MessageId = messageId;
|
|
JobId = jobId;
|
|
Payload = payload;
|
|
Attempt = attempt;
|
|
EnqueuedAt = enqueuedAt;
|
|
LeaseExpiresAt = leaseExpiresAt;
|
|
Consumer = consumer;
|
|
IdempotencyKey = idempotencyKey;
|
|
Attributes = attributes;
|
|
}
|
|
|
|
public string MessageId { get; }
|
|
|
|
public string JobId { get; }
|
|
|
|
public ReadOnlyMemory<byte> Payload { get; }
|
|
|
|
public int Attempt { get; }
|
|
|
|
public DateTimeOffset EnqueuedAt { get; }
|
|
|
|
public DateTimeOffset LeaseExpiresAt { get; private set; }
|
|
|
|
public string Consumer { get; }
|
|
|
|
public string? IdempotencyKey { get; }
|
|
|
|
public IReadOnlyDictionary<string, string> Attributes { get; }
|
|
|
|
public Task AcknowledgeAsync(CancellationToken cancellationToken = default)
|
|
=> _queue.AcknowledgeAsync(this, cancellationToken);
|
|
|
|
public Task RenewAsync(TimeSpan leaseDuration, CancellationToken cancellationToken = default)
|
|
=> _queue.RenewLeaseAsync(this, leaseDuration, cancellationToken);
|
|
|
|
public Task ReleaseAsync(QueueReleaseDisposition disposition, CancellationToken cancellationToken = default)
|
|
=> _queue.ReleaseAsync(this, disposition, cancellationToken);
|
|
|
|
public Task DeadLetterAsync(string reason, CancellationToken cancellationToken = default)
|
|
=> _queue.DeadLetterAsync(this, reason, cancellationToken);
|
|
|
|
internal bool TryBeginCompletion()
|
|
=> Interlocked.CompareExchange(ref _completed, 1, 0) == 0;
|
|
|
|
internal void RefreshLease(DateTimeOffset expiresAt)
|
|
=> LeaseExpiresAt = expiresAt;
|
|
}
|