Issue 54 make it possible to add Reset header when using the FixedWin…

…dowLimiter
cristipufu · Apr 21, 2023 · 9580775 · 9580775
1 parent 54fb56d
commit 9580775
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/RedisRateLimiting/FixedWindow/RedisFixedWindowRateLimiter.cs b/src/RedisRateLimiting/FixedWindow/RedisFixedWindowRateLimiter.cs
@@ -75,6 +75,7 @@ protected override RateLimitLease AttemptAcquireCore(int permitCount)
 
             leaseContext.Count = response.Count;
             leaseContext.RetryAfter = response.RetryAfter;
+            leaseContext.ExpiresAt = DateTimeOffset.FromUnixTimeSeconds(response.ExpiresAt);
 
             if (leaseContext.Count > _options.PermitLimit)
             {
@@ -114,6 +115,8 @@ private sealed class FixedWindowLeaseContext
             public TimeSpan Window { get; set; }
 
             public TimeSpan? RetryAfter { get; set; }
+
+            public DateTimeOffset? ExpiresAt { get; set; }
         }
 
         private sealed class FixedWindowLease : RateLimitLease
@@ -152,6 +155,12 @@ public override bool TryGetMetadata(string metadataName, out object? metadata)
                     return true;
                 }
 
+                if (metadataName == RateLimitMetadataName.Reset.Name && _context?.ExpiresAt is not null)
+                {
+                    metadata = _context.ExpiresAt.Value;
+                    return true;
+                }
+
                 metadata = default;
                 return false;
             }