From 64ae41789fb3ad836126a21dfee6716c3ec3312c Mon Sep 17 00:00:00 2001
From: Gus Brodman <gbrodman@google.com>
Date: Fri, 26 Jun 2026 16:02:23 -0400
Subject: [PATCH] Remove entries from token store atomically

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 bf7c882d011..f2640ed2194 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 @@ void refresh() {
           if (Duration.between(availableTokens.timestamp(), clock.now())
                   .compareTo(config.getRefreshPeriod())
               >= 0) {
-            tokensMap.remove(user);
+            tokensMap.remove(user, availableTokens);
           }
         });
   }