From b0630b3dddc30f32ed6a3c0aa0d603d2fbe61a37 Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Fri, 16 May 2025 12:09:39 +0800
Subject: [PATCH] perf: Refine the Model Manager code (#3098)

---
 apps/common/util/rsa_util.py | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/apps/common/util/rsa_util.py b/apps/common/util/rsa_util.py
index 003018672..452ca678d 100644
--- a/apps/common/util/rsa_util.py
+++ b/apps/common/util/rsa_util.py
@@ -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