fix(login): fix lock during redis login

eddrit/utils/oauth.py

@@ -71,7 +71,8 @@
 ]
 
 _valkey_connection_pool = valkey.ConnectionPool.from_url(url=VALKEY_URL)
-_valkey_lock_key = "oauth_login_lock"
+_valkey_login_lock_key = "oauth_login_lock"
+_valkey_check_headers_lock_key = "oauth_check_headers_lock"
 _valkey_headers_key = "oauth_headers"
 
 
@@ -119,8 +120,11 @@ def _get_login_headers_from_cache() -> dict[str, Any]:
     Get the login headers from cache
     """
     valkey_client = valkey.Valkey(connection_pool=_valkey_connection_pool)
-    with valkey_client.lock(_valkey_lock_key, timeout=20, blocking_timeout=5):
-        # Check if we have login headers in cache
+    # Check if we have login headers in cache, with lock if multiple workers
+    # try to login at same time
+    with valkey_client.lock(
+        _valkey_check_headers_lock_key, timeout=20, blocking_timeout=5
+    ):
         if not valkey_client.exists(_valkey_headers_key):
             oauth_login()
 
@@ -136,7 +140,7 @@ def oauth_login() -> None:
     logger.debug("Performing OAuth login")
     valkey_client = valkey.Valkey(connection_pool=_valkey_connection_pool)
 
-    with valkey_client.lock(_valkey_lock_key, timeout=20, blocking_timeout=5):
+    with valkey_client.lock(_valkey_login_lock_key, timeout=20, blocking_timeout=5):
         # Generate identity: unique ID + random user-agent
         unique_uuid = str(uuid4())
         android_app_version = random.choice(OFFICIAL_ANDROID_APP_VERSIONS)  # noqa: S311