From 499548321fd17466cf033eac7cc5eed0dde1d70e Mon Sep 17 00:00:00 2001
From: Matthias Ahouansou <matthias@ahouansou.cz>
Date: Wed, 10 Jul 2024 11:17:38 +0100
Subject: [PATCH] enforce maximum capacity

---
 src/service/rate_limiting/mod.rs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/service/rate_limiting/mod.rs b/src/service/rate_limiting/mod.rs
index b33b2381..3900e843 100644
--- a/src/service/rate_limiting/mod.rs
+++ b/src/service/rate_limiting/mod.rs
@@ -89,8 +89,14 @@ impl Service {
                 if *capacity >= limit.burst_capacity.get() {
                     return Err(gap);
                 } else {
+                    let zero_capacity = *capacity == 0;
                     *capacity += 1;
-                    // TODO: update thing
+
+                    // Ensures that the update point is in the future
+                    if zero_capacity {
+                        *instant = Instant::now()
+                    }
+
                     *instant += gap;
                 }
             }