From c7a27061d81537eec8e2d9fd22eaf438547c30d3 Mon Sep 17 00:00:00 2001
From: gbrodman <gbrodman@google.com>
Date: Wed, 1 Jul 2026 14:42:59 -0400
Subject: [PATCH] Remove entries from token store atomically (#3112)

just in case some other thread comes in while we're iterating
---
 proxy/src/main/java/google/registry/proxy/quota/TokenStore.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/proxy/src/main/java/google/registry/proxy/quota/TokenStore.java b/proxy/src/main/java/google/registry/proxy/quota/TokenStore.java
index bf7c882d0..f2640ed21 100644
--- a/proxy/src/main/java/google/registry/proxy/quota/TokenStore.java
+++ b/proxy/src/main/java/google/registry/proxy/quota/TokenStore.java
@@ -185,7 +185,7 @@ public class TokenStore {
           if (Duration.between(availableTokens.timestamp(), clock.now())
                   .compareTo(config.getRefreshPeriod())
               >= 0) {
-            tokensMap.remove(user);
+            tokensMap.remove(user, availableTokens);
           }
         });
   }