perf: Refine the Model Manager code (#3098)

This commit is contained in:
shaohuzhang1 2025-05-16 12:09:39 +08:00 committed by GitHub
parent b2bf69740c
commit b0630b3ddd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -40,15 +40,12 @@ def generate():
def get_key_pair():
rsa_value = rsa_cache.get(cache_key)
if rsa_value is None:
lock.acquire()
rsa_value = rsa_cache.get(cache_key)
if rsa_value is not None:
return rsa_value
try:
with lock:
rsa_value = rsa_cache.get(cache_key)
if rsa_value is not None:
return rsa_value
rsa_value = get_key_pair_by_sql()
rsa_cache.set(cache_key, rsa_value)
finally:
lock.release()
return rsa_value