Cache the result of normalize_repo for 10 minutes.

vdbergh · ppigazzini · commit d7bb5eddedf7 · 2025-08-31T13:38:28.000+02:00
diff --git a/server/fishtest/github_api.py b/server/fishtest/github_api.py
@@ -22,6 +22,8 @@
 TIMEOUT = 3
 INITIAL_RATELIMIT = 5000
 LRU_CACHE_SIZE = 6000
+NORMALIZE_REPO_CACHE_SIZE = 128
+NORMALIZE_REPO_CACHE_EXPIRY_SECONDS = 600
 
 _api_initialized = False
 
@@ -379,7 +381,15 @@ def get_master_repo(
             r = r["parent"]
 
 
+normalize_repo_cache = LRUCache(NORMALIZE_REPO_CACHE_SIZE)
+
+
 def normalize_repo(repo):
+    global normalize_repo_cache
+    if repo in normalize_repo_cache:
+        cache = normalize_repo_cache[repo]
+        if time.time() - cache[0] < NORMALIZE_REPO_CACHE_EXPIRY_SECONDS:
+            return cache[1]
     r = call(
         repo,
         _method="HEAD",
@@ -388,6 +398,7 @@ def normalize_repo(repo):
         _ignore_rate_limit=True,
     )
     r.raise_for_status()
+    normalize_repo_cache[repo] = (time.time(), r.url)
     return r.url
 
 
diff --git a/server/fishtest/views.py b/server/fishtest/views.py
@@ -1727,7 +1727,11 @@ def allow_github_api_calls():
             return False
         return True
 
-    user, repo = gh.parse_repo(gh.normalize_repo(tests_repo(run)))
+    try:
+        user, repo = gh.parse_repo(gh.normalize_repo(tests_repo(run)))
+    except Exception as e:
+        user, repo = gh.parse_repo(tests_repo(run))
+        print(f"Unable to normalize_repo: {str(e)}")
 
     anchor_url = gh.compare_branches_url(
         user1="official-stockfish",
